1. 程式人生 > >Leetcode-952 Largest Component Size by Common Factor(按公因數計算最大元件大小)

Leetcode-952 Largest Component Size by Common Factor(按公因數計算最大元件大小)

一開始想法單純,試了試並查集+GCD,果斷TLE 73

實在想不出比O(n^2)更好的演算法,所以偷瞄了大神的程式碼,如下

 1 #include <iostream>
 2 #include <set>
 3 #include <vector>
 4 #define _for(i,a,b) for(int i = (a);i < (b);i ++)
 5 
 6 using namespace std;
 7 
 8 const int maxn = 100039;
 9 set<int> g[maxn];
10 vector<int
> f[maxn]; 11 bool fvis[maxn] {0},gvis[maxn] {0}; 12 class Solution 13 { 14 public: 15 void demo(int x,int id) 16 { 17 for(int i = 2;i*i <= x;i ++) 18 { 19 if(!(x%i)) 20 { 21 f[id].push_back(i);//
factor in vector 22 g[i].insert(id);//index in set 23 while(!(x%i)) x /= i; 24 } 25 } 26 if(x>1) 27 { 28 f[id].push_back(x);//factor in vector 29 g[x].insert(id);//index in set
30 } 31 } 32 void dfs(int id,set<int> &S) 33 { 34 fvis[id] = true; 35 S.insert(id); 36 for(auto fac:f[id]) 37 { 38 if(gvis[fac]) continue; 39 gvis[fac] = true; 40 for(auto fidx:g[fac]) 41 { 42 if(fvis[fidx]) continue; 43 S.insert(fidx); 44 dfs(fidx,S); 45 } 46 } 47 } 48 int largestComponentSize(vector<int>& A) 49 { 50 _for(i,0,A.size()) 51 demo(A[i],i); 52 int rnt = 1; 53 _for(i,0,A.size()) 54 { 55 set<int> tmp; 56 if(fvis[i]) continue; 57 dfs(i,tmp); 58 rnt = max(rnt,(int)tmp.size()); 59 } 60 return rnt; 61 } 62 }; 63 int main() 64 { 65 Solution obj; 66 vector<int> A {2,3,6,7,4,12,21,39}; 67 cout << obj.largestComponentSize(A) << endl; 68 return 0; 69 }

稍微解釋一下,大神搞了一個雙射,f裡裝的都是A[i]的因子,g裡裝的都是因子的下標,比如g[2]裡裝的就都是偶數的A的元素的下標,然後dfs vector f,走一遍因子,再根據因子從set g裡找下標,每找到一個下標就裝到set裡,這其實就是一個"集",最後看看集的大小就是元件的大小,最大的集就是最大元件的大小