LeetCode-1617-统计子树中城市之间最大距离

1、位运算+枚举

我们可以使用16位的二进制串表示每个节点保留与否，从而表示每一个可能的子串。我们首先将当前的二维矩阵改写位每个节点的邻接矩阵，而后开始进行遍历。

考虑到一个集合中至少要两个节点，我们从011开始进行循环枚举每一个可能的子串。我们根据每一个可能的子串来计算当前子串对用树的直径即最大距离：我们首先利用二进制串 vis 表示每一个节点是否被访问过。而后我们遍历当前节点的每一个邻居节点，若其没有被访问过而且在当前的子串中，我们则将调用函数计算出以该邻居节点为根节点时，对应子树的最大距离。显然，以当前节点为子树根节点的最大距离用当比以邻居节点为子树根节点的最大距离加一。我们不断探索新节点并更新当前子树中的直径和最大距离。最终当子树中的所有节点都被遍历时，我们将当前子树对应直径的个数加一。

class Solution {
public:
    vector<int> countSubgraphsForEachDiameter(int n, vector<vector<int>> &edges) {
        vector<vector<int>> g(n);
        for (auto &e : edges) {
            int x = e[0] - 1, y = e[1] - 1; // 编号改为从 0 开始
            g[x].push_back(y);
            g[y].push_back(x); // 建树
        }

        vector<int> ans(n - 1);
        // 二进制枚举
        for (int mask = 3; mask < 1 << n; ++mask) {
            if ((mask & (mask - 1)) == 0) continue; // 需要至少两个点
            // 求树的直径
            int vis = 0, diameter = 0;
            function<int(int)> dfs = [&](int x) -> int {
                vis |= 1 << x; // 标记 x 访问过
                int max_len = 0;
                for (int y : g[x])
                    if ((vis >> y & 1) == 0 && mask >> y & 1) { // y 没有访问过且在 mask 中
                        int ml = dfs(y) + 1;
                        diameter = max(diameter, max_len + ml);
                        max_len = max(max_len, ml);
                    }
                return max_len;
            };
            dfs(__builtin_ctz(mask)); // 从一个在 mask 中的点开始递归
            if (vis == mask)
                ++ans[diameter - 1];
        }
        return ans;
    }
};