http://codeforces.com/contest/1060/problem/E
题意:
给出一棵树,现对一个节点相连的两个节点再连一条边,问所有点对的距离和
自己做的时候还一直想着树上dp瞎搞,搞了半天弄不出来,发现这道题思路还是比较巧妙
思路:
先把问题转化一下,求树上所有点对的边距离和,那么针对每一条边他的贡献就是 一端点数*另一端点数 这是所有的要使用她的点对。 那么问题就被简化了,针对这道题目,可以得到距离为偶数的点 x/2,距离为奇数的为 (x+1)/2 ,那么就是奇偶层计算距离时需要加1,那么我们统计出这个再和之前每条边的贡献求和除2就是答案
#include<bits/stdc++.h>
#define fi first
#define se second
#define log2(a) log(n)/log(2)
#define show(a) cout<<a<<endl;
#define show2(a,b) cout<<a<<" "<<b<<endl;
#define show3(a,b,c) cout<<a<<" "<<b<<" "<<c<<endl;
#define tim printf("Time cost : %lf s\n",(double)clock()/CLOCKS_PER_SEC);
using namespace std;
typedef long long ll;
typedef long long LL;
typedef pair<int, int> P;
typedef pair<P, ll> LP;
const ll inf = 1e18;
const int N = 1e6 + 10;
const ll mod = 1e9 + 7;
const int base = 131;
const double pi = acos ( -1 );
const double eps = 1e-8;
inline ll ksm(ll a,ll b){ll ans=1;while(b){if(b&1)ans=ans*a%mod;a=a*a%mod,b>>=1;}return ans;}
#define a(i,j) a[(i-1)*m+(j)]
#define b(i,j) b[(i-1)*m+(j)]
ll vis[N],num[N];
ll n, m, k, x, y, z;
ll a[N], b[N],c[N];
ll cx, cy, cnt, ans, sum, flag, t, ff,q;
struct node
{
ll to,k;
};
vector<int> v[N],res;
unordered_map<int, int> mp;
set<string> st;
ll dp[N];//该点下方的点数
void dfs(int x,int fa,int de)
{
dp[x]=1;
cnt+=de;
for(int to:v[x])
{
if(to==fa) continue;
dfs(to,x,de^1);
dp[x]+=dp[to];
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n;
for(int i=2;i<=n;i++)
{
cin>>x>>y;
v[x].push_back(y);
v[y].push_back(x);
}
dfs(1,-1,0);
for(int i=1;i<=n;i++)
{
ans+=dp[i]*(n-dp[i]);
}
ans+=cnt*(n-cnt);//奇偶层的贡献
cout<<ans/2;
}