码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-06-24 22:07:53      阅读:71      评论:0      收藏:0      [点我收藏+]

标签:最优   style   com   push   tps   bool   顺序   cout   color   

题目链接:https://codeforces.com/contest/1369/problem/E

题意

Lee 有 $n$ 种不同种类的食物和 $m$ 个朋友,每种食物有 $w_i$ 个,每个朋友喜欢吃两种食物 $x_i, y_i$,如果一个朋友被叫去厨房时一个喜欢吃的食物也没有,他就会吃掉 Lee,否则对还有的他喜欢吃的食物一种吃掉一个,问是否存在一个叫朋友去厨房的顺序保证 Lee 不会被吃掉。

题解

设 $s_i$ 为第 $i$ 种食物共需要多少个,若 $w_i \ge s_i$,则喜欢第 $i$ 种食物的朋友一定不会吃 Lee,最优起见,可以最后再叫他们,同时他们喜欢的另一种食物也可以省下来了。

代码

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n, m; cin >> n >> m;
    int w[n] = {};
    for (int i = 0; i < n; i++)
        cin >> w[i];
    vector<pair<int, int>> e[n];
    int s[n] = {};
    for (int i = 0; i < m; i++) {
        int u, v; cin >> u >> v;
        --u, --v;
        e[u].emplace_back(v, i);
        e[v].emplace_back(u, i);
        ++s[u], ++s[v];
    }
    queue<int> que;
    for (int i = 0; i < n; i++)
        if (w[i] >= s[i])
            que.push(i);
    vector<int> ans;
    bool vis[m] = {};
    while (!que.empty()) {
        int u = que.front();
        que.pop(); 
        for (auto [v, i] : e[u]) {
            if (vis[i])
                continue;
            vis[i] = 1;
            ans.push_back(i);
            if (--s[v] == w[v])
                que.push(v);
        }
    }
    if (int(ans.size()) < m) {
        cout << "DEAD" << "\n";
        return 0;
    }
    cout << "ALIVE" << "\n";
    reverse(ans.begin(), ans.end());
    for (int i = 0; i < m; i++)
        cout << ans[i] + 1 << " \n"[i == m - 1];
}

 

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

标签:最优   style   com   push   tps   bool   顺序   cout   color   

原文地址:https://www.cnblogs.com/Kanoon/p/13189658.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!