#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
const int mod=1e9+7;
struct node{
int to,cost;
};
int n,sum,dis[N],ans[N];
bool st[N];
vector<node>p[N];
void dfs(int u){
for(int i=0;i<p[u].size();i++){
node now=p[u][i];
if(!st[now.to]){
st[now.to]=1;
dis[now.to]=dis[u]^now.cost;
dfs(now.to);
st[now.to]=0;
}
}
}
signed main(){
ios::sync_with_stdio(false);
cin>>n;
for(int i=1;i<n;i++){
int x,y,w;
cin>>x>>y>>w;
p[x].push_back((node){y,w});
p[y].push_back((node){x,w});
}
dfs(1);
for(int i=1;i<=n;i++){
for(int j=0;j<=60;j++){
if(dis[i]&1ll<<j){
ans[j]=(ans[j]+1)%mod;
}
}
}
for(int i=0;i<=60;i++){
sum=sum+((1ll<<i)%mod)*(ans[i]%mod)*(n-ans[i]%mod)%mod;
}
cout<<sum%mod;
return 0;
}
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
const int mod=1e9+7;
struct node{
int to,cost;
};
int n,sum,dis[N],ans[N];
vector<node>p[N];
void dfs(int u,int fa){
for(int i=0;i<p[u].size();i++){
node now=p[u][i];
if(now.to==fa)continue;
dis[now.to]=dis[u]^now.cost;
dfs(now.to,u);
}
}
signed main(){
ios::sync_with_stdio(false);
cin>>n;
for(int i=1;i<n;i++){
int x,y,w;
cin>>x>>y>>w;
p[x].push_back((node){y,w});
p[y].push_back((node){x,w});
}
dfs(1,0);
for(int i=1;i<=n;i++){
for(int j=0;j<=60;j++){
if((dis[i]&(1ll<<j))){
++ans[j]%=mod;
}
}
}
for(int i=0;i<=60;i++){
sum+=((1ll<<i)%mod)*((ans[i]*(n-ans[i]))%mod)%mod;
sum%=mod;
}
cout<<sum%mod;
return 0;
}
这两份代码唯一的区别在于对每一位统计答案时第一份代码为:
sum=sum+((1ll<<i)%mod)*(ans[i]%mod)*(n-ans[i]%mod)%mod;
第二份代码为:
sum+=((1ll<<i)%mod)*((ans[i]*(n-ans[i]))%mod)%mod;
sum%=mod;
ans[i]表示二进制下该位1的个数,
这两种统计方式有何区别?为什么第二种可以AC而第一种不能?求解答。