CF1444C Team-Building(可持久化并查集)(二分图)
生活随笔
收集整理的這篇文章主要介紹了
CF1444C Team-Building(可持久化并查集)(二分图)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
解析
容易想到補集思想,尋找那些組之間不能形成二分圖
二分圖一般的兩個判定方法:
這里考慮并查集(看題解似乎染色也可做)
先把所有組內的邊合并并查集
如果某個組自己內部就有奇環,顯然不能和任何點配對,直接刪去并減去其貢獻
對于其他的邊,注意到,由于總邊數只有5e5級別,所以圖會非常稀疏
可以使用一些暴力點思路
我們sort一下,對于每個集合對(u,v),把其對應的邊全部加上,判斷完畢之后,再把邊刪去
由于并查集需要撤銷,不路徑壓縮,只按秩合并
代碼
#include<bits/stdc++.h> const int N=2e6+100; const int mod=1e9+7; #define ll long long using namespace std; inline ll read() {ll x(0),f(1);char c=getchar();while(!isdigit(c)) {if(c=='-')f=-1;c=getchar();}while(isdigit(c)) {x=(x<<1)+(x<<3)+c-'0';c=getchar();}return x*f; }int n,m,k;ll ans; int vis[N],col[N];#define oth(u) (u>n?u-n:u+n) int fa[N],siz[N],dis[N],top,u[N],v[N],ori; int find(int x){return x==fa[x]?x:find(fa[x]);} inline void merge(int x,int y){x=find(x);y=find(y);if(x==y) return;if(siz[x]>siz[y]) swap(x,y);fa[x]=y;siz[y]+=siz[x];++top;u[top]=x;v[top]=y;return; }struct edge{int x,y; }e[N]; int num; bool cmp(edge a,edge b){if(col[a.x]!=col[b.x]) return col[a.x]<col[b.x];else return col[a.y]<col[b.y]; }int main(){#ifndef ONLINE_JUDGE//freopen("a.in","r",stdin);//freopen("a.out","w",stdout);#endifn=read();m=read();k=read();ans=1ll*k*(k-1)/2;for(int i=1;i<=n*2;i++) fa[i]=i,siz[i]=1;for(int i=1;i<=n;i++) col[i]=read();for(int i=1;i<=m;i++){int x=read(),y=read();if(col[x]==col[y]){if(find(x)==find(y)){if(!vis[col[x]]) ans-=--k;vis[col[x]]=1;}else{merge(x,oth(y));merge(y,oth(x));}}else{if(col[x]>col[y]) swap(x,y);e[++num]=(edge){x,y};}}ori=top;sort(e+1,e+1+num,cmp);for(int i=1;i<=num;){int x=e[i].x,y=e[i].y,a=col[x],b=col[y],flag=0;if(vis[a]||vis[b]){++i;continue;}if(find(x)==find(y)){flag=1;}else{merge(x,oth(y));merge(y,oth(x));}i++;while(i<=num&&col[e[i].x]==a&&col[e[i].y]==b){x=e[i].x,y=e[i].y;if(find(x)==find(y)){flag=1;}else{merge(x,oth(y));merge(y,oth(x));}i++;}while(top!=ori){int x=u[top],y=v[top];top--;fa[x]=x;siz[y]-=siz[x];}ans-=flag;}printf("%lld\n",ans); }總結
以上是生活随笔為你收集整理的CF1444C Team-Building(可持久化并查集)(二分图)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 2K+180Hz+Fast-IPS:泰坦
- 下一篇: 机兵题材战棋游戏《钢岚》宣布 12 月