Leetcode 135. 分发糖果 (拓扑排序)

最新推荐文章于 2022-02-21 10:06:31 发布

chr1st0pher

最新推荐文章于 2022-02-21 10:06:31 发布

阅读量131

点赞数 1

分类专栏：图论 LeetCode

本文链接：https://blog.csdn.net/qq_43521140/article/details/111608846

版权

LeetCode 同时被 2 个专栏收录

29 篇文章 0 订阅

订阅专栏

图论

12 篇文章 0 订阅

订阅专栏

Description
在这里插入图片描述
Solution
以孩子间的关系建单向边，每个点在拓扑排序中的层数之和即为答案

Hint
评分相同的孩子需要建双向边

Code

class Solution {
public:
    #define maxn 200005
    int d[maxn];
    vector<int>edge[maxn];
    inline void add(int u,int v) {
        edge[u].push_back(v);
        d[v]++;
    }
    int candy(vector<int>& ratings) {
        int n =  ratings.size();
        for(int i = 0;i < n;++i) edge[i].clear(), d[i] = 0;
        for(int i = 0;i < n;++i) {
            if(i != n-1) {
                if(ratings[i] > ratings[i+1]) add(i+1,i);
                if(ratings[i] < ratings[i+1]) add(i,i+1);
            }
            if(i != 0) {
                if(ratings[i] > ratings[i-1]) add(i-1,i);
                if(ratings[i] < ratings[i-1]) add(i,i-1);
            }
        }
        queue<int>q;
        int cnt = 1, res = 0;
        for(int i = 0;i < n;++i) if(d[i]==0) q.push(i),q.push(cnt);
        while(!q.empty()) {
            int now = q.front();q.pop();
            cnt = q.front();q.pop();
            res += cnt;
            for(auto v : edge[now]) {
                if(--d[v] == 0) q.push(v), q.push(cnt+1);
            }
        }return res;
    }
};