NowCoder--牛客练习赛30 C_小K的疑惑

题目链接 ：牛客练习赛30 C_小K的疑惑

i j k 可以相同 而且 距离%2 只有 0 1两种情况 我们考虑 因为要 d(i j)=d(i k)=d(j k) 所以我们只能找 要么三个点 任意两个点之间的距离都是 1  要么都是 0

所以我们先吧每个点到根节点的距离表示出来 找到有 x个 距离为1 y个距离为0 的点    当 选择距离是1 的时候 我们假设任选一个数 首先确定了 i 这个值  j 可以选x个数  k也可以选x个数 （因为包含了他自己本身的距离为 0） 然后又有 x 个数 所以 当选择 距离是1 的时候  答案是x^3         0 的情况同理

#include<bits/stdc++.h>
using namespace std;
#define maxn 10010
#define pii pair<int,int>
int dis[maxn];
vector<pii>q[maxn];
void dfs(int u,int fa,int z){
 dis[u]=z;
 for(int j=0;j<q[u].size();j++){
     int v=q[u][j].first;
     int va=q[u][j].second;
     if(v!=fa){
         dfs(v,u,va+z);
     }
 }
}
int main(){
   int n;
   cin>>n;
   memset(dis,0,sizeof(dis));
   for(int j=0;j<n-1;j++){
      int x,y,z;
      cin>>x>>y>>z;
      q[x].push_back(pii(y,z));
      q[y].push_back(pii(x,z));
   }
   dfs(1,0,0);
   int x=0,y=0;
   for(int j=1;j<=n;j++){
      if(dis[j]%2) x++;
      else y++;
   }
   //cout<<x<<" "<<y<<endl;
   long long ans=1LL*x*x*x+1LL*y*y*y;
   cout<<ans<<endl;

   return 0;
}