HDU 5416(思路题目)

题意:

给定一棵树,要求统计所有的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;
}


你可能感兴趣的:(HDU 5416(思路题目))