codeforces1369E DeadLee

https://codeforces.com/problemset/problem/1369/E

ned[x]为当前这个食物要被多少人吃

如果所有ned[x]>w[x],那么无论怎么安排,最后还是不够

如果某个ned[x]<=w[x],说明我们可以把吃x的人放到最后,那么这样他们最终至少会有一个人吃,在放人的同时,我们ned进行减操作,如果某个时刻ned[i]<=w[i],说吃第i种食物的人也可以放在当前已经放了的人除外的最后了,那么加入队列,用个队列维护吃哪些食物的人已经可以放最后了,然后维护ned[i]的大小,顺便放人。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=3e5+10;

int n,m,cas,k,cnt,tot,ans;
int a[maxl][2],w[maxl],ned[maxl];
vector<int> b[maxl],c;
char s[maxl];
bool in[maxl],vis[maxl]; 
queue<int> q;

inline void prework()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&w[i]),vis[i]=false;
	int x,y;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		a[i][0]=x;a[i][1]=y;
		ned[x]++;ned[y]++;
		b[x].push_back(i);
		b[y].push_back(i);
		in[i]=false;
	}
} 

inline void mainwork()
{
	ans=0;
	for(int i=1;i<=n;i++)
	if(ned[i]<=w[i])
		ans=1,q.push(i),vis[i]=true;
	if(!ans) 
		return;
	int x,y,tot=0,u;
	while(!q.empty())
	{
		u=q.front();q.pop();tot++;
		for(int v:b[u])
		if(!in[v])
		{
			c.push_back(v);in[v]=true;
			x=a[v][0];y=a[v][1];
			--ned[x];--ned[y];
			if(ned[x]<=w[x] && !vis[x])
			{
				q.push(x);
				vis[x]=true;
			}
			if(ned[y]<=w[y] && !vis[y])
			{
				q.push(y);
				vis[y]=true;
			}
		}
	}
	int len=c.size();
	if(tot!=n || len!=m)
		ans=0;
}

inline void print()
{
	if(!ans)
		puts("DEAD");
	else
	{ 
		puts("ALIVE");
		int len=c.size();
		for(int i=len-1;i>=0;i--)
			printf("%d%c",c[i]," \n"[i==0]);
	}
}

int main()
{
	int t=1;
	//scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/liufengwei1/article/details/107237317