Intel Code Challenge Final Round (Div. 1 + Div. 2, Combined) G - Xor-matic Number of the Graph 线性基好题

G - Xor-matic Number of the Graph

上一道题的加强版本,对于每个联通块需要按位算贡献。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PII pair<int, int>
#define PLI pair<LL, int>
#define ull unsigned long long
using namespace std; const int N = 1e5 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + ; int n, m;
LL d[N], bin[N], num[];
bool vis[N];
vector<PLI> edge[N];
vector<int> id;
struct Base {
LL a[];
int cnt;
void init() {
memset(a, , sizeof(a));
cnt = ;
}
void add(LL x) {
for(int j = ; ~j; j--) {
if((x >> j) & ) {
if(!a[j]) {a[j]=x; cnt++; break;}
else x ^= a[j];
}
}
}
} base; void dfs(int u, int fa) {
vis[u] = true;
id.push_back(u);
for(int i = ; i < edge[u].size(); i++) {
int v = edge[u][i].se; LL w = edge[u][i].fi;
if(v == fa) continue;
if(vis[v]) {
base.add(d[u]^d[v]^w);
} else {
d[v] = d[u] ^ w;
dfs(v, u);
}
}
} int main() {
base.pirnt();
scanf("%d%d", &n, &m);
for(int i=bin[]=; i <= n; i++)
bin[i] = bin[i-] * % mod;
for(int i = ; i <= m; i++) {
int u, v; LL w;
scanf("%d%d%lld", &u, &v, &w);
edge[u].push_back(mk(w, v));
edge[v].push_back(mk(w, u));
} LL ans = ;
for(int i = ; i <= n; i++) {
if(!vis[i]) {
base.init();
id.clear();
dfs(i, );
for(int j = ; j <= ; j++) {
num[] = , num[] = ;
for(auto &x : id) {
num[(d[x]>>j)&]++;
}
bool flag = false;
for(int k = ; k < ; k++) {
if((base.a[k]>>j)&) {
flag = true;
break;
}
} LL tmp = num[]*(num[]-)/ + num[]*(num[]-)/;
tmp %= mod;
if(flag) ans = (ans + bin[j]*bin[base.cnt-]%mod*tmp%mod) % mod;
tmp = num[] * num[] % mod;
if(flag) ans = (ans + bin[j]*bin[base.cnt-]%mod*tmp%mod) % mod;
else ans = (ans + bin[j]*bin[base.cnt]%mod*tmp%mod) % mod;
}
}
}
printf("%lld\n", ans);
return ;
} /*
*/
上一篇:rest api load test


下一篇:python3 字典常见用法总结