CodeForces - 1369E DeadLee

Solution

题面真的鬼畜。

设 $w$ 为菜品的数量，$deg$ 为需求量。你会发现，当所有 $w<deg$ 时是一定没有解的。

为什么呢？可以思考我们臆想出来这种方案可以有解的思路：如果一个人只选了一种，就减小了其他菜品的负担。但是，虽然这种情况会发生，它仍不足以使其合法。

可以这样想，假设某人选择某样菜，那么那样菜排队序列中那个人后面一定有人（即一定有人没选到）。这样是矛盾的，因为排队序列对于每个菜品都是相同的。

我们可以先选择 $w<deg$ 的，这样反正有 $w>=deg$ 的来兜底，但又会尽量增加只选一种菜品的人数。

把一个人看做连接两个菜的边。做一个拓扑排序，将 $w>=deg$ 的菜先塞进队列，然后将其连边的点的 $deg--$（那个人已经有兜底的了）。然后用个随便玩意存答案。

注意第一次插入的时候就可以存人了。

Code

#include <queue>
#include <cstdio>
#include <vector>
#include <iostream>
using namespace std;

const int N = 1e5 + 5, M = 2e5 + 5;

int n, m, w[N], head[N], nxt[M << 1], to[M << 1], cnt, deg[N], id[M << 1];
queue <int> q;
bool vis[N], ok[M];
vector <int> ans;

int read() {
	int x = 0, f = 1; char s;
	while((s = getchar()) > '9' || s < '0') if(s == '-') f = -1;
	while(s >= '0' && s <= '9') x = (x << 1) + (x << 3) + (s ^ 48), s = getchar();
	return x * f;
}

void addEdge(const int u, const int v, const int Id) {
    to[++ cnt] = v, nxt[cnt] = head[u], head[u] = cnt, id[cnt] = Id;
}

void topol() {
    for(int i = 1; i <= n; ++ i) if(w[i] >= deg[i]) q.push(i), vis[i] = 1;
    while(! q.empty()) {
        int u = q.front(); q.pop();
        for(int i = head[u]; i; i = nxt[i])
        	if(! ok[id[i]]) ans.push_back(id[i]), ok[id[i]] = 1;
        for(int i = head[u]; i; i = nxt[i]) {
            int v = to[i];
            -- deg[v];
            if(w[v] >= deg[v])
				if(! vis[v]) q.push(v), vis[v] = 1;
        }
    }
}

int main() {
    int u, v;
    n = read(), m = read();
    for(int i = 1; i <= n; ++ i) w[i] = read();
    for(int i = 1; i <= m; ++ i) {
        u = read(), v = read(); ++ deg[u], ++ deg[v];
        addEdge(v, u, i), addEdge(u, v, i);
    }
    topol();
    if(ans.size() == m) {
        puts("ALIVE");
        for(int i = ans.size() - 1; ~i; -- i) printf("%d ", ans[i]); puts("");
    }
    else puts("DEAD");
    return 0;
}