刚学OI,#9MLE,求大佬改题
查看原帖
刚学OI,#9MLE,求大佬改题
639483
FiresonZ楼主2022/5/21 21:51
#include<iostream>
#include<array>
#include<algorithm>
#include<cmath>
#include<bitset>
#include<vector>
#define reg register
using namespace std;
using gg=long long;
struct node{
	gg from,to,nxt;
};
array<gg,200001> x,y;
array<gg,200010> fa,mt;
array<gg,200010> res;
array<gg,200010> head;
array<node,400010> plt;
array<bool,200010> ned;
gg top=0;
gg fafind(const gg &x)
{
	if(fa[x]==x)
	{
		return x;
	}
	return fa[x]=fafind(fa[x]);
}
void add(const gg &from,const gg &to)
{
	top++;
	plt[top].from=from;
	plt[top].to=to;
	plt[top].nxt=head[from];
	head[from]=top;
	return;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	reg gg n,m,k,ans;
	cin>>n>>m;
	for(reg gg i=0;i<n;i++) fa[i]=i;
	for(reg gg i=1;i<=m;i++)
	{
		cin>>x[i]>>y[i];
	}
	cin>>k;
	ans=n-k;
	for(reg gg i=1;i<=k;i++)
	{
		cin>>mt[i];
		ned[mt[i]]=true;
	}
	for(reg gg i=1;i<=m;i++)
	{
		if(ned[x[i]]==false&&ned[y[i]]==false)
		{
			gg fidx=fafind(x[i]),fidy=fafind(y[i]);
			if(fidx!=fidy)
			{
				ans--;
				fa[fidx]=fidy;
			}
		}
		else
		{
			add(x[i],y[i]);
			add(y[i],x[i]);
		}
	}
	for(reg gg i=k;i>=0;i--)
	{
		res[i]=ans;
		ned[mt[i]]=false;
		ans++;
		gg fidmt=fafind(mt[i]);
		for(reg gg j=head[mt[i]];j;j=plt[j].nxt)
		{
			gg fidplt=fafind(plt[j].to);
			if(ned[plt[j].to]==false&&fidmt!=fidplt)
			{
				ans--;
				if(ans<1)
				{
					ans=1;
				}
				fa[fidplt]=fidmt;
			}
		}
	}
	for(reg gg i=0;i<=k;i++) cout<<res[i]<<'\n';
	return 0;
}
2022/5/21 21:51
加载中...