2017国家集训队作业[agc006f]Blackout
2017国家集训队作业[agc006f]Blackout
题意:
有一个\(N*N\)的网格,一开始有\(M\)个格子被涂黑,给出这\(M\)个格子,和染色操作:如果有坐标为\((x,y),(y,z)\)的格子已被染黑,那么就可以染黑坐标为\((y,z)\)的格子。问操作到不能再操作的时候,网格里有多少个黑格子?(\(1\le N,M\le 10^5\),最开始给出的\(M\)个坐标互不相同)
题解:
在场上签到签了两个小时,看到这题一脸懵逼。= =!大概想到是把形如\((x,y),(y,z)\)格子之间连一条边,然后会产生个新点之类的。现在想想,这样的空间复杂度完全不对了,做法不可能是这个。(然而我在场上想到这的时候,决定去打暴力了)
每个\(OIer\)肯定做过一题,就是在网格上求某种合法方案,使得同一行、同一列只能有一个点。二分图匹配,对吧。那里选一个点等于行号同列之间连一条边。这题也是这样的模型。
把所有初始涂黑的点的行号同列号之间连一条单向边,那么所有点组成了若干个弱联通块,如果有边\(x\rightarrow y,y\rightarrow z\),就会产生\(z\rightarrow x\)的边,这样,我们发现一次操作一定生成了一个三元环,三染色。。。。。。。。不妨设染色顺序为\(0,1,2\)。
三种情况:
\(1.\)如果三种颜色没法全部用到,那么说明当前弱联通块无法进行操作。那么它对答案的贡献就是弱联通块的边数。
\(2.\)如果三种颜色全部用到了,但是存在某两个同颜色的点之间有边,那么说明这个弱联通块出现了环,稍微画画就能得出这个弱联通块最后一定会成为一个完全图,贡献为点数的平方。
\(3.\)其它情况,贡献为:颜色为颜色为\(0\)的点数\(*\)颜色为\(1\)的点数\(+\)颜色为\(1\)的点数\(*\)颜色为\(2\)的点数\(+\)颜色为\(2\)的点数\(*\)颜色为\(0\)的点数。
一名签到和暴力选手的啰嗦题解。。。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define fo(i,l,r) for(ll i=l;i<=r;i++)
#define of(i,l,r) for(ll i=l;i>=r;i--)
#define fe(i,u) for(ll i=head[u];i;i=e[i].next)
using namespace std;
typedef long long ll;
inline ll rd()
{
static ll x,f;
x=0,f=1;
char ch=getchar();
for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
return x*f;
}
const ll N=100010;
struct edge{
ll v,ty,next;
edge(ll v=0,ll ty=0,ll next=0):v(v),ty(ty),next(next){}
}e[N<<1];
ll n,m,col[N],cnt[3],siz;
ll tot=0,head[N];
bool flag;
inline void add(ll u,ll v,ll ty){e[++tot]=edge(v,ty,head[u]);head[u]=tot;}
void dfs(ll u)
{
cnt[col[u]]++;
fe(i,u){
ll v=e[i].v,ty=e[i].ty;
siz+=(ty==1);
if(~col[v])flag|=(col[v]!=(col[u]+ty)%3);
else col[v]=(col[u]+ty)%3,dfs(v);
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("inb.txt","r",stdin);
#endif
n=rd();m=rd();
fo(i,1,m){
ll x=rd(),y=rd();
add(x,y,1);add(y,x,2);
}
memset(col,-1,sizeof col);
ll ans=0;
fo(i,1,n){
if(~col[i])continue;
cnt[col[i]=0]=cnt[1]=cnt[2]=siz=flag=0;
dfs(i);
if(flag)ans+=(cnt[0]+cnt[1]+cnt[2])*(cnt[0]+cnt[1]+cnt[2]);
else if(cnt[0]&&cnt[1]&&cnt[2])ans+=cnt[0]*cnt[1]+cnt[1]*cnt[2]+cnt[2]*cnt[0];
else ans+=siz;
}
printf("%lld\n",ans);
return 0;
}