VOJ 威虎山上的分配 题解 拓扑排序

威虎山上的分配

题目描述

每年过年的时候，座山雕都会给兄弟们分银子。分银子之前，座山雕允许大伙儿发表意见，因为要是没法满足所有人的意见，指不定谁要搞出什么大新闻。不过每个人在提意见的时候只能说：“我认为 A 分的银子应该比 B 多！”。座山雕决定要找出一种分配方案，满足所有人的意见，同时使得所有人分得的银子总数最少，并且每个人分得的银子最少为 100 两。

输入描述

第一行两个整数 n , m ( 0 < n ≤ 10000 , 0 < m ≤ 20000 ) ，表示总人数和总意见数；
接下来 m 行，每行两个整数 a 、b ，之间用一个空格隔开，表示某个意见认为第 a 号小弟所分得的银两应该比第 b 号小弟多，所有小弟的编号由 1 开始。

输出描述

若无法找到合法方案，则输出 “Unhappy!” (不包含引号)，否则输出一个数表示最少总银两数。

样例 #1

样例输入 #1

3 2
1 2
2 3

样例输出 #1

303

代码

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

int n;       // 节点个数
i64 sum = 0; // 统计最少总银两数

int Toposort(vector<int> &a, vector<int> &b) // a和b分别记录了所有有向边的起点和终点
{
    vector<vector<int>> access(n + 1); // 对于每个点所指向的点，例如对于access[i][j]表示对于第i个点所指向的第j个点
    vector<int> d(n + 1, 0);           // 每个点的入度
    for (int i = 0; i < a.size(); i++)
    {
        int x = a[i];
        int y = b[i];
        d[y]++;                 // 对每个点的入度进行记录
        access[x].push_back(y); // 把y放入x的点中
    }
    int cnt = 0;
    queue<int> q;                // 拓扑排序入度为0的收集队列
    for (int i = 1; i <= n; i++)
    {
        if (d[i] == 0)
        {
            cnt++;
            q.push(i); // 把所有入度为0的点放入q中
        }
    }
    int level = 0; // 统计层数
    while (!q.empty())
    {
        int size = q.size();
        sum += size * (100 + level); // 计算银子总数
        level++;                     // 每层银子数需要+1
        while (size--)
        {
            int cur = q.front();
            q.pop();
            for (auto &next : access[cur]) // 遍历第i个点所指向的所有点
            {
                d[next]--;
                if (!d[next]) // 入度为0
                {
                    cnt++;
                    q.push(next); // q需要不断放入入度为0的点
                }
            }
        }
    }
    if (cnt < d.size() - 1) // 入队的数量若小于当前读入的总数量，说明有环
        return -1;
    else
        return 1;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    int m;
    cin >> n >> m;
    vector<int> a(m), b(m);
    for (int i = 0; i < m; i++)
        cin >> b[i] >> a[i]; // 读入时先读入银子少的
    int res = Toposort(a, b);
    if (res == -1) // 有环的话则不存在合法方案
        cout << "Unhappy!" << '\n';
    else
        cout << sum << '\n';

    return 0;
}