Leetcode-952 Largest Component Size by Common Factor(按公因数计算最大组件大小)

一开始想法单纯，试了试并查集+GCD，果断TLE 73

实在想不出比O(n^2)更好的算法，所以偷瞄了大神的代码，如下

#include <iostream>
#include <set>
#include <vector>
#define _for(i,a,b) for(int i = (a);i < (b);i ++)

using namespace std;

const int maxn = 100039;
set<int> g[maxn];
vector<int> f[maxn];
bool fvis[maxn] {0},gvis[maxn] {0};
class Solution
{
    public:
        void demo(int x,int id)
        {
            for(int i = 2;i*i <= x;i ++)
            {
                if(!(x%i))
                {
                    f[id].push_back(i);//factor in vector
                    g[i].insert(id);//index in set
                    while(!(x%i)) x /= i;
                }
            }
            if(x>1)
            {
                f[id].push_back(x);//factor in vector
                g[x].insert(id);//index in set
            }
        }
        void dfs(int id,set<int> &S)
        {
            fvis[id] = true;
            S.insert(id);
            for(auto fac:f[id])
            {
                if(gvis[fac]) continue;
                gvis[fac] = true;
                for(auto fidx:g[fac])
                {
                    if(fvis[fidx]) continue;
                    S.insert(fidx);
                    dfs(fidx,S);
                }
            }
        }
        int largestComponentSize(vector<int>& A)
        {
            _for(i,0,A.size())
                demo(A[i],i);
            int rnt = 1;
            _for(i,0,A.size())
            {
                set<int> tmp;
                if(fvis[i]) continue;
                dfs(i,tmp);
                rnt = max(rnt,(int)tmp.size());
            }
            return rnt;
        }
};
int main()
{
    Solution obj;
    vector<int> A {2,3,6,7,4,12,21,39};
    cout << obj.largestComponentSize(A) << endl;
    return 0;
}

稍微解释一下，大神搞了一个双射，f里装的都是A[i]的因子，g里装的都是因子的下标，比如g[2]里装的就都是偶数的A的元素的下标，然后dfs vector f，走一遍因子，再根据因子从set g里找下标，每找到一个下标就装到set里，这其实就是一个"集"，最后看看集的大小就是组件的大小，最大的集就是最大组件的大小