hdu 4705(树形DP)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4705

思路：反面考虑，用总的方案数减去A，B，C三点在同一路径上的方案数。于是我们可以确定中间点B，在当前以B为根求得的son中任选一个，在剩下的节点n-tmp-1（tmp为已经求得的B的儿子的个数）中任选一个,产生tmp*(n-tmp-1)中组合。

 1 #include<iostream>

 2 #include<cstdio>

 3 #include<cstring>

 4 #include<algorithm>

 5 using namespace std;

 6 #define MAXN 100100

 7 #pragma comment(linker, "/STACK:16777216")

 8 

 9 struct Edge{

10     int v,next;

11 }edge[MAXN*4];

12 

13 int NE;

14 long long n;

15 int head[MAXN];

16 

17 void Insert(int u,int v)

18 {

19     edge[NE].v=v;

20     edge[NE].next=head[u];

21     head[u]=NE++;

22 }

23 

24 

25 bool mark[MAXN];

26 long long sum,ans;

27 

28 int dfs(int u)

29 {

30     mark[u]=true;

31     int son,tmp=0;

32     for(int i=head[u];i!=-1;i=edge[i].next){

33         int v=edge[i].v;

34         if(mark[v])continue;

35         son=dfs(v);//当前分支儿子的个数

36         tmp+=son;//已经求出的儿子的个数

37         ans+=(long long )(n-1-tmp)*son;

38     }

39     return tmp+1;

40 }

41 

42 

43 int main()

44 {

45     int u,v;

46     while(~scanf("%I64d",&n)){

47         NE=0;

48         memset(head,-1,sizeof(head));

49         for(int i=1;i<n;i++){

50             scanf("%d%d",&u,&v);

51             Insert(u,v);

52             Insert(v,u);

53         }

54         memset(mark,false,sizeof(mark));

55         ans=0;

56         dfs(1);

57         sum=n*(n-1)*(n-2)/6;

58         printf("%I64d\n",sum-ans);

59     }

60     return 0;

61 }

View Code