Codeforces Round #652 (Div. 2) E. DeadLee 贪心

题意：

派会上有n种食物，每种食物有wi份。有m个朋友，每一个朋友有两种他喜欢吃的食物xi，yi。你需要判断他的朋友是否都能吃到食物。如果都能吃到食物，那么要输出朋友来的顺序，不能的话输出“DEAD”。

如果一个朋友来的时候发现两种他喜欢的食物都有，那么他会两种食物都吃一份，如果仅有一种食物还有，那么就只吃一份这种食物。

 

题解：

首先先预处理一下每一种食物有多少人要吃，我们设某种食物的需求量为needi份，如果某种食物的needi<=wi，那么只要喜欢第i种食物的人永远不可能没有食物吃。那么我们可以使用贪心策略，既然这些人什么时候都有吃的东西，那么我们就最后再通知他来派会，同时这些人对其他食物的需求我们可以消除掉，因为另一种食物有没有都不重要了。就一直这样判断看最后还有没有人没有加入到通知到派会，如果没有的话就输出“ALIVE”并输出序列

 

其次就是这道题的数据较大，所以暴力方法就不可行，我们就可以采用建图方式来解决问题，具体方法见代码

 

代码：

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<string>
#include<queue>
#include<string.h>
#include<map>
#include <iostream>
#include <math.h>
using namespace std;
typedef long long ll;
#define maxn 300005
#include<vector>
struct node
{
    int id,x;
};
vector<node>e[maxn];
queue<int>q;
int d[maxn],w[maxn],ans[maxn],vis[maxn];
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        e[x].push_back({i,y});
        e[y].push_back({i,x});
        d[x]++;d[y]++;
    }
    for(int i=1;i<=n;i++)
    {
        if(d[i]<=w[i])
        {
            q.push(i);
        }
    }
    int f=m;
    while(!q.empty())
    {
        int i=q.front();
        q.pop();
        //printf("i=%d\n",i);
        for(int k=0;k<e[i].size();++k)
        {
            node j=e[i][k];
            if(vis[j.id]==1)
                continue;
            vis[j.id]=1;
            d[j.x]--;
            //printf("j.id=%d j.x=%d\n",j.id,j.x);
            if(d[j.x]<=w[j.x]) q.push(j.x);
            ans[f--]=j.id;

        }
    }
    if(f)
    {
        printf("DEAD\n");
        return 0;
    }
    printf("ALIVE\n");
    for(int i=1;i<=m;i++)
    {
        printf("%d",ans[i]);
        printf("%c",i==n?'\n':' ');
    }
}