标签：AtCoder Beginner Contest int 264 fi find se define

比赛链接

AtCoder Beginner Contest 264

E.Blackout 2

给出很多点(\(n+m\leq 2\times 10^5\))，有发电站和城市，以及很多边（\(e\leq 5\times 10^5\)），有 \(q\) 次删边操作，求每次操作后有电的城市数量

解题思路

并查集

离线逆序处理，即只考虑建边操作，用并查集维护有电城市的信息即可

• 时间复杂度：\(O(n+m+e+q)\)

代码

// Problem: E - Blackout 2
// Contest: AtCoder - freee Programming Contest 2022（AtCoder Beginner Contest 264）
// URL: https://atcoder.jp/contests/abc264/tasks/abc264_e
// Memory Limit: 1024 MB
// Time Limit: 3000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

// %%%Skyqwq
#include <bits/stdc++.h>
 
// #define int long long
#define help {cin.tie(NULL); cout.tie(NULL);}
#define pb push_back
#define fi first
#define se second
#define mkp make_pair
using namespace std;
 
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
 
template <typename T> bool chkMax(T &x, T y) { return (y > x) ? x = y, 1 : 0; }
template <typename T> bool chkMin(T &x, T y) { return (y < x) ? x = y, 1 : 0; }
 
template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

const int N=5e5+5;
int n,m,e,q,Q[N],fa[N],sz[N],res[N];
bool v[N],st[N];
PII a[N];
int find(int x)
{
	return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main()
{
    cin>>n>>m>>e;
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=n;i++)sz[i]=1;
    for(int i=1;i<=e;i++)
    {
    	cin>>a[i].fi>>a[i].se;
    	if(a[i].fi>a[i].se)swap(a[i].fi,a[i].se);
    }
    cin>>q;
    for(int i=1;i<=q;i++)cin>>Q[i],st[Q[i]]=true;
    for(int i=1;i<=e;i++)
    {
    	if(!st[i])
    	{
    		if(a[i].fi<=n&&a[i].se>n)
    		{
    			int x=find(a[i].fi);
    			if(!v[x])
    			{
    				v[x]=true;
    				res[q]+=sz[x];
    			}
    		}
    		else if(a[i].se<=n)
    		{
    			int x=find(a[i].fi),y=find(a[i].se);
    			if(x!=y)
    			{
    				if(!v[x]&&v[y])
    				{
    					res[q]+=sz[x];
    					fa[x]=y,sz[y]+=sz[x];
    				}
    				else if(!v[y]&&v[x])
    				{
    					res[q]+=sz[y];
    					fa[y]=x,sz[x]+=sz[y];
    				}
    				else
    					fa[y]=x,sz[x]+=sz[y];
    			}
    		}
    	}
    }
    for(int i=q;i;i--)
    {
    	int id=Q[i];
    	int t=res[i];
    	if(a[id].fi<=n&&a[id].se>n)
		{
			int x=find(a[id].fi);
			if(!v[x])
			{
				v[x]=true;
				t+=sz[x];
			}
		}
		else if(a[id].se<=n)
		{
			int x=find(a[id].fi),y=find(a[id].se);
			if(x!=y)
			{
				if(!v[x]&&v[y])
				{
					t+=sz[x];
					fa[x]=y,sz[y]+=sz[x];
				}
				else if(!v[y]&&v[x])
				{
					t+=sz[y];
					fa[y]=x,sz[x]+=sz[y];
				}
				else
    				fa[y]=x,sz[x]+=sz[y];
			}
		}
    	res[i-1]=t;
    }
    for(int i=1;i<=q;i++)cout<<res[i]<<'\n';
    return 0;
}

标签：AtCoder,Beginner,Contest,int,264,fi,find,se,define
来源： https://www.cnblogs.com/zyyun/p/16585066.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。