题目连接:http://acm.hdu.edu.cn/showproblem.php?pid=3887
题意:给出一棵树,对于每一个节点,问他的子孙节点中有多少个节点小于该节点。
思路:首先找出这棵树的DFS序列,每一个节点出现在两个位置,这两个位置之间的节点就是该节点的子孙节点。
然后用树状数组求出这两个位置之间有多少个节点小于该节点。
hdu这题出的有点龊 ,,用dfs搜索会爆栈,要手动模拟先序或者后序遍历!
code:
View Code
1 # include<stdio.h>
2 # include<string.h>
3 # define N 100050
4 struct node{
5 int from,to,next;
6 }edge[2*N],edge1[2*N];
7 int head[N],tol,head1[N],tol1,S[N],s[N],k,dp[N],count[N],visit[N],sig[N];
8 void add(int a,int b)
9 {
10 edge[tol].from=a;edge[tol].to=b;edge[tol].next=head[a];head[a]=tol++;
11 }
12 void add1(int a,int b)
13 {
14 edge1[tol1].from=a;edge1[tol1].to=b;edge1[tol1].next=head1[a];head1[a]=tol1++;
15 }
16 void dfs2(int root)
17 {
18 int j,top,u,v;
19 top=0;
20 k=0;
21 S[++top]=root;
22 while(top>0)
23 {
24 u=S[top];
25 if(!visit[u])
26 {
27 visit[u]=1;
28 s[++k]=u;
29 sig[u]=k;
30 }
31 for(j=head[u];j!=-1;j=edge[j].next)
32 {
33 v=edge[j].to;
34 if(visit[v]) continue;
35 S[++top]=v;
36 break;
37 }
38 if(j==-1)
39 {
40 top--;
41 dp[u]=k-sig[u];
42 }
43 }
44 }
45 void insert(int i)
46 {
47 while(i<=k)
48 {
49 count[i]++;
50 i+=i&(-i);
51 }
52 }
53 int query(int i)
54 {
55 int sum=0;
56 while(i>0)
57 {
58 sum+=count[i];
59 i-=i&(-i);
60 }
61 return sum;
62 }
63 int main()
64 {
65 int i,j,v,n,root,num[N],ans,node,a,b;
66 while(scanf("%d%d",&n,&root)!=EOF)
67 {
68 if(!n && !root) break;
69 tol=0;
70 memset(head,-1,sizeof(head));
71 for(i=1;i<n;i++)
72 {
73 scanf("%d%d",&a,&b);
74 add(a,b);
75 add(b,a);
76 }
77 memset(dp,0,sizeof(dp));
78 //dfs1(root,0);//树形DP求节点子孙的个数
79 memset(visit,0,sizeof(visit));
80 dfs2(root);//模拟栈,先序遍历
81 tol1=0;
82 memset(head1,-1,sizeof(head1));
83 for(i=1;i<=n;i++)
84 {
85 node=s[i];
86 ans=dp[node]+i;
87 add1(ans,node);
88 }
89 memset(count,0,sizeof(count));
90 for(i=1;i<=n;i++)
91 {
92 node=s[i];
93 num[node]=query(node-1);
94 insert(node);
95 for(j=head1[i];j!=-1;j=edge1[j].next)
96 {
97 v=edge1[j].to;
98 num[v]=query(v-1)-num[v];
99 }
100 }
101 printf("%d",num[1]);
102 for(i=2;i<=n;i++)
103 printf(" %d",num[i]);
104 printf("\n");
105 }
106 return 0;
107 }