RC-u5 树与二分图 题解

#include<bits/stdc++.h>
using namespace std;
const int N=100010;int n;
int h[N],ne[N],e[N],dist;
int sum=0;
int st[N];//防止重复!!!!
void add(int a,int b){
	e[dist]=b;ne[dist]=h[a];h[a]=dist++;
}
void dfs(int p,int color){
	st[p]=1;if(color)sum++;
	for(int i=h[p];i!=-1;i=ne[i]){
		int j=e[i];if(!st[j])dfs(j,1-color);
	}
}
int main()
{
	cin>>n;memset(h,-1,sizeof h);
	for(int i=1;i<=n-1;i++){
		int a,b;cin>>a>>b;add(a,b);add(b,a);
	}
	dfs(1,1);
	cout<<sum*(n-sum)-n+1;
	return 0;
}