版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/C_13579/article/details/83901373
地址:https://ac.nowcoder.com/acm/contest/216/C
思路:容易得出,(i,j,k)中三个相等i=j=k时一定满足条件,当(i,j,k)中有两个相等i=j时,那么长度L(i,k)一定为偶数,当(i,j,k)三个都不相同时,可以肯定两两之间长度一定为偶数(路径唯一,且L(i,j)+L(j,k)=L(i,k)) ,那么就相当于找两点间距离为偶数的个数了,对于树结构,可以求DFS求出节点到根节点的距离,那么所有节点就分为距离为偶数s0和奇数s1两种。那么两点间距离为偶数的个数就是L(s0,s0)和L(s1,s1)两种。
Code :
#include<iostream>
#include<vector>
using namespace std;
typedef pair<int,int> pr;
typedef long long LL;
const int MAX_N=1e5+5;
LL n,s0,s1;
vector<pr> e[MAX_N];
void DFS(int u,int pre,int Sum);
int main()
{
ios::sync_with_stdio(false);
while(cin>>n){
s0=1; s1=0;
for(int i=0;i<=n;++i)
e[i].clear();
int u,v,w;
for(int i=1;i<n;++i)
{
cin>>u>>v>>w;
e[u].push_back(pr{v,w});
e[v].push_back(pr{u,w});
}
DFS(1,0,0);
LL ans=n+s0*(s0-1)*3+s1*(s1-1)*3+s0*(s0-1)*(s0-2)+s1*(s1-1)*(s1-2);
cout<<ans<<endl;
}
return 0;
}
void DFS(int u,int pre,int Sum)
{
for(auto t:e[u])
if(t.first!=pre){
if((Sum+t.second)%2) ++s1;
else ++s0;
DFS(t.first,u,Sum+t.second);
}
}