c++ - 从#define 到函数

Question

我在一个函数中有这段代码，但我无法理解它的作用。

....
#define ascend(i) do {\
            int h = nodes[i].heavyindex;\
            int p = nodes[i].heavypos;\
            m##i = max(m##i + paths[h].ftree.sum(p), paths[h].stree.max_(0, p));\
            i = paths[h].parent;\
        } while (0)
    while (nodes[a].heavyindex != nodes[b].heavyindex) {
        if (nodes[a].heavyindex > nodes[b].heavyindex) {
            ascend(a);
        } else {
            ascend(b);
        }
    }
#undef ascend
...

我认为#define的代码是:

#define ascend(i) do {\
            int h = nodes[i].heavyindex;\
            int p = nodes[i].heavypos;\
            m##i = max(m##i + paths[h].ftree.sum(p), paths[h].stree.max_(0, p));\
            i = paths[h].parent;\
        } while (0)

所以函数内部的真正代码只有这样:

while (nodes[a].heavyindex != nodes[b].heavyindex) {
        if (nodes[a].heavyindex > nodes[b].heavyindex) {
            ascend(a);
        } else {
            ascend(b);
        }
    }

1)是吗？
2) 我想将 #define 的代码移到一个函数中以更好地理解它的作用，但我如何翻译以下行？

m##i = max(m##i + paths[h].ftree.sum(p), paths[h].stree.max_(0, p));\ 

Answer 1

1. 是的。

2. 如 Ben Voigt 所述在注释中，## 是标记粘贴运算符。因此，在定义了 #define f(i) m##i 后，f(a) 将扩展为 ma，f(b ) 将扩展为 mb 等。

   由于这只能通过预处理器实现，因此您必须想出其他办法将其实现为一个函数。通过引用传递 ma 和 mb 是个好主意。它可能看起来像这样:

   ascend(T& mi) {
       ...
       mi = max(mi + paths[h].ftree.sum(p), paths[h].stree.max_(0, p)); 
       ...
   }
   

   其中 T 是 ma 和 mb 的类型。如果它们是不同的类型，则需要将其设为函数模板。