正題

題目鏈接:https://www.luogu.com.cn/problem/AT2339

---

題目大意

給出$n$個(gè)點(diǎn)$m$條邊的一張無向圖，然后有一張$n\times n$的圖，每個(gè)點(diǎn)是一個(gè)二元組$(a,b)$。$(a,b)$和$(c,d)$連邊當(dāng)且僅當(dāng)$a$和$c$有連邊，$b$和$d$有連邊。

求新圖的連通塊數(shù)量

$1\le n\le 10^5,1\le m\le 2\times 10^5$

---

解題思路

計(jì)數(shù)問題，我們考慮固定一個(gè)基準(zhǔn)，以原圖中的連通塊為基準(zhǔn)。

對于一個(gè)點(diǎn)$(x,y)$，它能走到的點(diǎn)，發(fā)現(xiàn)如果$x$和$y$所在的連通塊都可以黑白染色，那么$x$和$y$的黑白順序是固定的，否則無論如何$x$整個(gè)都可以變?yōu)檎麄€(gè)連通塊或者$y$可以變?yōu)檎麄€(gè)連通塊。

然后這樣一些會發(fā)現(xiàn)樣例都過不了，因?yàn)橛幸环N很特殊的點(diǎn)，就是沒有任何邊連接的點(diǎn)，這一部分的點(diǎn)我們需要特判。

記大小不是$1$的連通塊數(shù)為$A$，其中能奇偶染色的為$B$，大小為$1$的連通塊數(shù)為$C$，那么答案就是：
$$A\times A+B\times B+C\times n\times 2?C\times C$$

時(shí)間復(fù)雜度：$O(n+m)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=2e5+10; struct node{ll to,next; }a[N<<1]; ll n,m,tot,siz,A,B,C,ls[N],v[N]; bool flag; void addl(ll x,ll y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;return; } void dfs(ll x,ll c){if(v[x]==(c^1))flag=1;if(v[x]>=0)return;v[x]=c;siz++;for(ll i=ls[x];i;i=a[i].next)dfs(a[i].to,c^1);return; } signed main() {scanf("%lld%lld",&n,&m);for(ll i=1,x,y;i<=m;i++){scanf("%lld%lld",&x,&y);addl(x,y);addl(y,x);}memset(v,-1,sizeof(v));for(ll i=1;i<=n;i++)if(v[i]<0){flag=siz=0;dfs(i,0);if(siz==1)C++;else A++,B+=!flag;}printf("%lld\n",1ll*A*A+1ll*B*B+2ll*C*n-C*C);return 0; }

總結(jié)

以上是生活随笔為你收集整理的AT2339-[AGC011C]Squared Graph【黑白染色】的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。