1. 程式人生 > >樹上dfs+思維

樹上dfs+思維

dfs clas amp com n-1 ... += n) add

技術分享圖片

技術分享圖片

 1 #include<cstdio>
 2 const int N=1000010;
 3 int cnt,head[N],n;
 4 int size[N],num[N];
 5 void init()
 6 {
 7     cnt  = 0;
 8     for(int i =0;i<N;i++) head[i]=-1;
 9 }
10 struct Node{
11     int fr,to,nex;
12 }e[N<<1];//乘2
13 void add(int x,int y){
14     e[cnt].fr=x;e[cnt].to=y;
15 e[cnt].nex=head[x];head[x]=cnt++; 16 } 17 void dfs(int x,int rt){ 18 size[x] =1; 19 for(int i =head[x];i!=-1;i=e[i].nex){ 20 int v= e[i].to; 21 if(v!=rt){ 22 dfs(v,x); 23 size[x]+=size[v]; 24 } 25 } 26 num[size[x]]++;
27 } 28 int x,y; 29 bool check(int i) 30 { 31 i++;//子樹數目 32 if(n%i) return false;//無法均分 33 int tmp = n/i;//每個子樹大小 34 int sum=0; 35 for(int i=tmp;i<=n;i+=tmp) sum+=num[i];//找子樹大小為tmp的子樹數目 36 //tmp 2*tmp 3*tmp 其實都是再找tmp 37 return sum==i; 38 } 39 int main() 40 { 41 //
freopen("1.txt","r",stdin); 42 init(); 43 scanf("%d",&n); 44 for(int i =0;i<n-1;i++) { 45 scanf("%d%d",&x,&y); 46 add(x,y);add(y,x);//無向邊 47 } 48 dfs(1,0);//1不一定是根節點dfs(n-1,0)都可以。 49 for(int i=1;i<n;i++){ 50 //printf("%d......",i); 51 if(check(i)) printf("%d ",i); 52 } 53 return 0; 54 }

樹上dfs+思維