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