HDU 5416(思路题目)

来源:互联网 发布:ubuntu ftp服务器搭建 编辑:程序博客网 时间:2024/05/19 18:43

题意:

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


0 0