Source Code
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
using namespace __gnu_pbds;
#define ordered_set tree<pair<lng,lng>,null_type,less<pair<lng,lng>>,rb_tree_tag,tree_order_statistics_node_update>
#define ordered_multiset tree<double,null_type,less<double>,rb_tree_tag,tree_order_statistics_node_update>
#define rtr return 0
#define lng long long
#define double long double
#define endl "\n"
#define cmbntrcs for(int i=1;i<INF;i++){fact[i]=mul(i,fact[i-1]);}
//#pragma GCC optimize
const int MOD=1e9+7,INF=2e6+7,SZ=5e5+7,MSZ=1e5+3,MAXA=6e5+7;
int add(int x,int y){int z=x+y;if(z>=MOD){z-=MOD;}return z;}
int sub(int x,int y){int z=x-y;if(z<0){z+=MOD;}return z;}
int mul(int x,int y){return(x*1ll*y)%MOD;}
int pwr(int a,lng b)
{
 if(!b){return 1;}
 int res=pwr(a,b/2);res=mul(res,res);if(b%2){res=mul(res,a);}
 return res;
}
int mod_inv(int a){return pwr(a,MOD-2);}
vector<int>fact(INF,1),inv(INF,1);
int nCr(int n,int r){return mul(fact[n],mul(inv[r],inv[n-r]));}
int nPr(int n,vector<int>r)
{
 int den=1;
 for(int i=0;i<r.size();i++){den=mul(den,fact[r[i]]);}
 return mul(fact[n],mod_inv(den));
}
int sb(int s,int b){return nCr(s+b-1,s);}
int prfx(int i,int j){if(i<0||j<0){return 0;}return nCr(i+j,i);}
lng lcm(lng x,lng y){return(x*y/__gcd(x,y));}
lng trig(lng x){return x*(x+1)/2;}
lng sigma(lng x,lng y){return trig(y)-trig(x-1);}
lng fp(lng a,lng b)
{
 if(!b){return 1;}
 lng res=fp(a,b/2);res*=res;if(b%2){res*=a;}
 return res;
}
string num_bin(lng n){if(n==0){return"0";}string x="";while(n){char c=(n%2)+'0';x+=c;n/=2;}reverse(x.begin(),x.end());return x;}
lng bin_num(string x){reverse(x.begin(),x.end());lng n=0;for(int i=0;i<x.size();i++){if(x[i]=='1'){n+=pow(2,i);}}return n;}
/**struct mtrx
{
 lng nx,mx;vector<vector<lng>>bd;
 void bld(vector<vector<lng>>obd){nx=obd.size(),mx=obd[0].size();bd=obd;}
 void outm(){for(lng i=0;i<nx;i++){for(lng j=0;j<mx;j++){cout << bd[i][j] << ' ';}cout << endl;}}
};
mtrx idnt(lng n)
{
 vector<vector<lng>>v;v.resize(n);
 for(lng i=0;i<n;i++){v[i].assign(n,0);v[i][i]=1;}
 return{n,n,v};
}
mtrx modn(mtrx a,lng x)
{
 for(lng i=0;i<a.nx;i++){for(lng j=0;j<a.mx;j++){a.bd[i][j]%=x;}}
 return a;
}
mtrx mulm(mtrx a,mtrx b)
{
 if(a.mx!=b.nx){return{1,1,{{0}}};}
 vector<vector<lng>>v;v.resize(a.nx);
 for(lng i=0;i<a.nx;i++)
 {
  v[i].resize(b.mx);
  for(lng j=0;j<b.mx;j++)
  {
   lng s=0;
   for(lng x=0;x<a.mx;x++)
   {
    s+=a.bd[i][x]*b.bd[x][j];
    s%=MOD;
   }
   v[i][j]=s;
  }
 }
 mtrx ret;ret.bld(v);
 //MOD FACTOR ret=modn(ret,MOD);
 return ret;
}
mtrx fpm(mtrx a,lng b)
{
 if(b==0){return idnt(a.nx);}
 mtrx ret=fpm(a,b/2);ret=mulm(ret,ret);
 if(b%2){ret=mulm(ret,a);}
 return ret;
}**/
lng rnd_num()
{
 srand(time(0));
 lng x=0;for(lng i=0;i<60;i++){x+=(1ll<<i)*(rand()%2);}
 return x;
}
lng rndm(lng a,lng b){return a+rnd_num()%((b+1)-a);}
map<pair<int,int>,pair<int,int>>f1,f2;
map<pair<int,int>,int>w1o,w2o,w1,w2;
vector<int>a1[MSZ],a2[MSZ];
vector<pair<int,int>>e1,e2;
pair<lng,lng>s1(lng x,lng pr)
{
 if(f1[{x,pr}]!=make_pair(-1,-1)){return f1[{x,pr}];}
 lng z=1,o=0;
 for(lng i=0;i<a1[x].size();i++)
 {
  lng y=a1[x][i];if(y==pr){continue;}
  pair<lng,lng>r=s1(y,x);
  if(w1[{x,y}]){o+=r.first;z+=r.second;}
  else{z+=r.first;o+=r.second;}
 }
 return f1[{x,pr}]={z,o};
}
pair<lng,lng>s2(lng x,lng pr)
{
 if(f2[{x,pr}]!=make_pair(-1,-1)){return f2[{x,pr}];}
 lng z=1,o=0;
 for(lng i=0;i<a2[x].size();i++)
 {
  lng y=a2[x][i];if(y==pr){continue;}
  pair<lng,lng>r=s2(y,x);
  if(w2[{x,y}]){o+=r.first;z+=r.second;}
  else{z+=r.first;o+=r.second;}
 }
 return f2[{x,pr}]={z,o};
}
int main()
{
 ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
 int n;cin >> n;
 for(int i=0;i<n-1;i++)
 {
  int u,v,w;cin >> u >> v >> w;
  a1[u].push_back(v);a1[v].push_back(u);
  w1o[{v,u}]=w;w1o[{u,v}]=w;
  e1.push_back({u,v});e1.push_back({v,u});
 }
 for(int i=0;i<n-1;i++)
 {
  int u,v,w;cin >> u >> v >> w;
  a2[u].push_back(v);a2[v].push_back(u);
  w2o[{v,u}]=w;w2o[{u,v}]=w;
  e2.push_back({u,v});e2.push_back({v,u});
 }
 int ans=0;
 for(int p=0;p<30;p++)
 {
  int pp=1<<p,c=0;
  for(int i=0;i<e1.size();i++)
  {
   w1[e1[i]]=0;if((w1o[e1[i]]&pp)!=0){w1[e1[i]]=1;}
   f1[e1[i]]={-1,-1};
  }
  for(int i=0;i<e2.size();i++)
  {
   w2[e2[i]]=0;if((w2o[e2[i]]&pp)!=0){w2[e2[i]]=1;}
   f2[e2[i]]={-1,-1};
  }
  for(lng x=1;x<=n;x++)
  {
   pair<lng,lng>p1={0,0},p2={0,0},t;
   t=s1(x,a1[x][0]);
   p1.first+=t.first;
   p1.second+=t.second;
   t=s1(a1[x][0],x);
   if(w1[{a1[x][0],x}]){swap(t.first,t.second);}
   p1.first+=t.first;
   p1.second+=t.second;
   t=s2(x,a2[x][0]);
   p2.first+=t.first;
   p2.second+=t.second;
   t=s2(a2[x][0],x);
   if(w2[{a2[x][0],x}]){swap(t.first,t.second);}
   p2.first+=t.first;
   p2.second+=t.second;
   c=add(c,add(mul(p1.first,p2.second),mul(p2.first,p1.second)));
  }
  ans=add(ans,mul(pp,c));
 }
 cout << ans;
 return 0;
}
Copy
Trees xOr Yahia_Emara
GNU G++17
7043 ms
37.0 MB
Time Limit Exceeded