指数做法过二十万?
查看原帖
指数做法过二十万?
378346
expnoi楼主2023/7/13 15:22

求证正确性/Hack

#include<bits/stdc++.h>
using namespace std;
inline int read()
{
	int s=0,w=1;
	char c=getchar();
	while(c<'0'||c>'9')
	{
		if(c=='-')w=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')s=(s<<3)+(s<<1)+(c^48),c=getchar();
	return s*w;
}
inline void print(int x)
{
	if(x<0)x=-x,putchar('-');
	if(x>=10)print(x/10);
	putchar(x%10+48);
}
int n,m,k;
struct node{
	int v,next;
}e[1000010];
int eid=1,head[1000010],p[1000010],h[1000010],dis[1000010],vis[1000010];
inline void insert(int u,int v)
{
	e[eid].v=v;
	e[eid].next=head[u];
	head[u]=eid++;
}
stack<int> q;
pair<int,int> P[1000010];
int main()
{
	n=read();
	m=read();
	k=read();
	for(int i=1;i<=m;i++)
	{
		int u=read(),v=read();
		insert(u,v);
		insert(v,u);
	}
	memset(dis,-1,sizeof(dis)); 
	for(int i=1;i<=k;i++)
	{
		p[i]=read();
		h[i]=read();
		dis[p[i]]=max(dis[p[i]],h[i]);
	}
	int tot=0;
	for(int i=1;i<=n;i++){
		if(dis[i])
		{
			P[++tot]={dis[i],i};
			vis[i]=1;
		}
	}
	sort(P+1,P+tot+1);
	for(int i=1;i<=tot;i++)
	{
		q.push(P[i].second);
		vis[P[i].second]=1;
	}
	while(q.size())
	{
		int u=q.top();
		q.pop();
		vis[u]=0;
		for(int i=head[u];i;i=e[i].next)
		{
			int v=e[i].v;
			if(dis[v]<dis[u]-1)
			{
				dis[v]=dis[u]-1;
				if(!vis[v])
				{
					q.push(v);
					vis[v]=1;
				}
			}
		}
	}
	int cnt=0;
	for(int i=1;i<=n;i++)
	{
		if(dis[i]>=0)
		{
			cnt++;
		}
	}
	print(cnt);
	puts("");
	for(int i=1;i<=n;i++)
	{
		if(dis[i]>=0)
		{
			print(i);
			putchar(' ');
		}
	}
	
}
2023/7/13 15:22
加载中...