题目链接:点我啊╭(╯^╰)╮
题目大意:
n
n
n 个点,
m
m
m 条边的无向图,每个点都属于一个组
选取两个组的所有点,这些点组成的图无奇环
问有多少组对满足要求
解题思路:
考虑求有多少非法组对
是否存在奇环的判断方法可以用并查集:
对于边
(
u
,
v
)
(u,v)
(u,v),将
u
u
u 与
v
+
n
v+n
v+n 合并,
v
v
v 与
u
+
n
u+n
u+n合并
若存在边
(
u
,
v
)
(u,v)
(u,v) 已经在同一个集合里了,则说明存在奇环
对于同一组的所有点,直接按上述方法判断
对于不同组,一条边若连接了组
u
u
u 与组
v
v
v,则将所有连接了组
u
u
u 与组
v
v
v 的边放到一起
然后将这些边一起加上,用上述方法判断是否存在奇环
判断完后再撤销这些并查集,判断另外两组
这样保证判断不会重复
#include<bits/stdc++.h>
#define rint register int
#define deb(x) cerr<<#x<<" = "<<(x)<<'\n';
using namespace std;
typedef long long ll;
typedef pair <int,int> pii;
const int maxn = 1e6 + 5;
int n, m, k, a[maxn], b[maxn];
int ji[maxn], c[maxn], ans;
map <pii, int> mp;
vector <int> g[maxn];
struct USF {
struct Stack {
int u, v, rk;
} st[maxn];
int fa[maxn], rk[maxn], top;
inline void init() {
top = 0;
for(int i=1; i<=n*2; i++) fa[i] = i, rk[i] = 0;
}
inline int getf(int x) {
return x == fa[x] ? x : getf(fa[x]);
}
inline bool merge(int u, int v) {
int fu = getf(u), fv = getf(v);
if(fu == fv) return false;
if(rk[fu] > rk[fv]) swap(u, v), swap(fu, fv);
st[++top] = {fu, fv, rk[fv]};
fa[fu] = fv, rk[fv] += rk[fu] == rk[fv];
return true;
}
inline void undo(int k) {
while(k--) {
Stack &now = st[top--];
int u = now.u, v = now.v;
fa[u] = u, rk[v] = now.rk;
}
}
} t;
signed main() {
scanf("%d%d%d", &n, &m, &k);
for(int i=1; i<=n; i++) scanf("%d", c+i);
t.init();
for(int i=1; i<=m; i++) {
scanf("%d%d", a+i, b+i);
if(c[a[i]] != c[b[i]]) continue;
int fu = t.getf(a[i]), fv = t.getf(b[i]);
if(fu == fv) ji[c[a[i]]] = 1;
else t.merge(a[i], b[i]+n), t.merge(b[i], a[i]+n);
}
int tot = 0;
for(int i=1; i<=m; i++) {
if(c[a[i]] == c[b[i]]) continue;
if(ji[c[a[i]]] || ji[c[b[i]]]) continue;
int u = c[a[i]], v = c[b[i]];
if(u > v) swap(u, v);
if(mp.count({u, v})) g[mp[{u, v}]].push_back(i);
else g[mp[{u, v}]=++tot].push_back(i);
}
for(auto i : mp) {
int now = t.top, id = i.second;
for(auto j : g[id]) {
int u = a[j], v = b[j];
int fu = t.getf(u), fv = t.getf(v);
if(fu == fv) { ans--; break; }
else t.merge(u, v+n), t.merge(v, u+n);
}
t.undo(t.top - now);
}
ll cnt = 0;
for(int i=1; i<=k; i++) cnt += !ji[i];
printf("%lld\n", cnt*(cnt-1)/2+ans);
}