题目:http://acm.hdu.edu.cn/showproblem.php?pid=4705
题目大意:给你一棵树,让你求出不在同一条简单链的三个点的个数。
思路:枚举这三个点的中点,我们默认让1号节点做树根,设以某个节点u的子节点为根节点的子树大小为 v1、v2、v3。。,则另外的点有 (n - size[ u ]),然后把依次过来把这些点乘一下就好了。
这道题目我竟然写了2个小时,中间爆 int 还错了几次。。。 不想说什么了,挫啊。。。 = =
代码如下:
#pragma comment(linker, "/STACK:16777216")
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef __int64 lld;
const int MAXN = 100011 ;
int n;
struct Edge
{
int t,next;
} edge[MAXN<<1];
int tot,head[MAXN];
void add_edge(int s,int t)
{
edge[tot].t = t;
edge[tot].next = head[s];
head[s] = tot++;
}
int size[MAXN];
lld ans;
void dfs(int u,int fa)
{
size[u] = 1;
for(int e = head[u];e!=-1;e = edge[e].next)
{
int v = edge[e].t;
if(v==fa) continue;
dfs(v,u);
ans += (lld)(size[u] - 1)*size[v];
size[u] += size[v];
}
ans += (lld)(size[u] - 1)*(n - size[u]);
}
int main()
{
while(~scanf("%d",&n))
{
tot=0;
memset(head,-1,sizeof(head));
int a,b;
for(int i= 1;i<n;i++)
{
scanf("%d%d",&a,&b);
add_edge(a,b);
add_edge(b,a);
}
ans = 0;
dfs(1,0);
ans = (lld)n*(n - 1)*(n - 2)/(lld)6 - ans;
printf("%I64d\n",ans);
}
return 0;
}
/*
5
1 2
1 3
2 4
2 5
6
1 2
1 3
2 4
2 5
3 6
7
1 2
1 3
2 4
2 5
3 6
3 7
*/