621. Task Scheduler

[kazuki-official]

https://leetcode.com/problems/task-scheduler/description/

[nodchip]

こちらのコメントをご参照ください。
hemispherium/LeetCode_Arai60#10 (comment)

[kazuki-official]

ありがとうございます。今回は値渡しの方がよさそうですね

[nodchip]

書いてみたところ、延々とバグってひどいコードになってしまいました。供養のために掲載します。

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        map<char, int> task_to_count;
        for (char task : tasks) {
            ++task_to_count[task];
        }

        map<int, vector<char>> remaining_tasks;
        for (auto [task, count] : task_to_count) {
            remaining_tasks[count].push_back(task);
        }

        vector<int> last_clocks(128, std::numeric_limits<int>::min() / 2);
        int clock = 0;
        while (!remaining_tasks.empty()) {
            ++clock;

            // for (const auto& [count, tasks] : remaining_tasks) {
            //     std::cout << count;
            //     for (char task : tasks) {
            //         std::cout << " " << task;
            //     }
            //     std::cout << std::endl;
            // }

            ComsumeTask(remaining_tasks, last_clocks, clock, n);
        }

        return clock;
    }

    void ComsumeTask(
        map<int, vector<char>>& remaining_tasks, vector<int>& last_clocks,
        int clock, int n) {
        for (auto rit = remaining_tasks.rbegin(); rit != remaining_tasks.rend(); ++rit) {
            int count = rit->first;
            for (auto it = rit->second.begin(); it != rit->second.end(); ++it) {
                char task = *it;
                // std::cout << task << "?" << std::endl;
                if (clock - last_clocks[task] <= n) {
                    continue;
                }

                // std::cout << task << std::endl;

                swap(*it, rit->second.back());
                rit->second.pop_back();
                if (rit->second.empty()) {
                    remaining_tasks.erase(count);
                }
                if (count > 1) {
                    remaining_tasks[count - 1].push_back(task);
                }
                last_clocks[task] = clock;
                return;
            }
        }

        // std::cout << "idle" << std::endl;
    }
};

[kazuki-official]

書いてみたところ、延々とバグってひどいコードになってしまいました。供養のために掲載します。

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        map<char, int> task_to_count;
        for (char task : tasks) {
            ++task_to_count[task];
        }

        map<int, vector<char>> remaining_tasks;
        for (auto [task, count] : task_to_count) {
            remaining_tasks[count].push_back(task);
        }

        vector<int> last_clocks(128, std::numeric_limits<int>::min() / 2);
        int clock = 0;
        while (!remaining_tasks.empty()) {
            ++clock;

            // for (const auto& [count, tasks] : remaining_tasks) {
            //     std::cout << count;
            //     for (char task : tasks) {
            //         std::cout << " " << task;
            //     }
            //     std::cout << std::endl;
            // }

            ComsumeTask(remaining_tasks, last_clocks, clock, n);
        }

        return clock;
    }

    void ComsumeTask(
        map<int, vector<char>>& remaining_tasks, vector<int>& last_clocks,
        int clock, int n) {
        for (auto rit = remaining_tasks.rbegin(); rit != remaining_tasks.rend(); ++rit) {
            int count = rit->first;
            for (auto it = rit->second.begin(); it != rit->second.end(); ++it) {
                char task = *it;
                // std::cout << task << "?" << std::endl;
                if (clock - last_clocks[task] <= n) {
                    continue;
                }

                // std::cout << task << std::endl;

                swap(*it, rit->second.back());
                rit->second.pop_back();
                if (rit->second.empty()) {
                    remaining_tasks.erase(count);
                }
                if (count > 1) {
                    remaining_tasks[count - 1].push_back(task);
                }
                last_clocks[task] = clock;
                return;
            }
        }

        // std::cout << "idle" << std::endl;
    }
};

clockをインクリメントしながら実際の動作に近い形で実装したのですね。ぱっと見変なところはないのですが、バグっているのですね...

[nodchip]

clockをインクリメントしながら実際の動作に近い形で実装したのですね。ぱっと見変なところはないのですが、バグっているのですね...

説明足らずで申し訳ありません。上記のコードは何度か修正して AC をもらったものです。途中、

if (rit->second.empty()) {
    remaining_tasks.erase(count);
}

が抜けており、無限ループしたりしていました。

あと、 rit と it の変数名がひどいと感じています。

[kazuki-official]

ああ、そういうことですね。
自分では思い付いてなかった解法なので勉強になりました。