HDU 5416（思路题目）

最新推荐文章于 2020-07-07 00:12:22 发布

原创最新推荐文章于 2020-07-07 00:12:22 发布 · 410 阅读

0 ·

CC 4.0 BY-SA版权

思路题专栏收录该内容

31 篇文章

订阅专栏

题意：

给定一棵树，要求统计所有的F(U,V)，使得路径上的权值异或之和为一个查询值，魅族数据查询数至多为10

解析：

从 f[u->v] = f(u->1 )^(1->v);这样先预处理f（1->i）然后统计即可，注意统计的时候 cnt数组要开到（1<<17）

因为路径权值不超过1e5，至多为17位二进制数。异或最大值不超过（1<<17）;

#include<cstring>
#include<cstdio>
#include<algorithm>
#include <iostream>
#include <vector>
using namespace std;
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define rep1(i,x,y) for(int i=x;i<=(int)y;i++)
const int N = (1e5)+1100;
int cnt[1<<17],d[N],n;
struct node{ int to,v; node(int x=0,int y=0):to(x),v(y){} };
vector<node> G[N];
void dfs(int u,int val,int fa){
   cnt[d[u]=val]++;
   rep(i,G[u].size())if(G[u][i].to!=fa){
       int v=G[u][i].to;
       dfs(v,val^G[u][i].v,u);
   }
}
int main()
{
    memset(cnt,0,sizeof(cnt));
    int T;
    scanf("%d",&T);
    while(T--){
       scanf("%d",&n);
       rep1(i,1,n) G[i].clear();
       rep(i,n-1){
           int x,y,z;
           scanf("%d %d %d",&x,&y,&z);
           G[x].push_back(node(y,z));
           G[y].push_back(node(x,z));
       }
       dfs(1,0,-1);
       int Q,nee; scanf("%d",&Q);
       for(int i=1;i<=Q;i++){
           scanf("%d",&nee);
           long long ans = 0;
           rep1(i,1,n) ans=ans+cnt[nee^d[i]];
           if(nee==0) ans=((ans+n)>>1);
           else ans = (ans>>1);
           printf("%I64d\n",ans);
       }
       for(int i=1;i<=n;i++) cnt[d[i]]=0;
    }
    return 0;
}