帶有限期的作業排序改進實現

算法思想:
三個數組分別是作業的value值p[i],時間d[i], 最近的前一個可插入的作業f[i], 以及初始值均爲-1的parent[i]
將輸出結果保存入std::vector< int> res中
首先根據p[i]排序,每次使用貪心儘可能找出最大的p[i]值即可
從1開始遍歷數組判斷d[i]是否滿足條件
判斷方法:
通過find方法找出根j,也就是最接近的可能插入的點, 然後判斷f[j]是否可以插入,如果不可以插入則捨棄,否則 將i保存入res中然後再將Union前面最近的根和這個值

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int parent[1005], d[1005] = {0, 2, 2, 1, 3, 3}, f[1005];

void unionArray(int i, int j) {
    int x = parent[i] + parent[j];
    if(parent[i] > parent[j]) { // i的節點少
        parent[i] = j;
        parent[j] = x;
    } else {
        parent[j] = i;
        parent[i] = x;
    }
}

int find(int i) {
    int j = i;
    while (parent[j] > 0) {
        j = parent[j];
    }
    int k = i;
    while (parent[k] > 0) {
        int temp = parent[k];
        parent[k] = j;
        k = temp;
    }
    return j;
}

int main() {
    int size = 5;
    vector<int> res;
    for (int i = 0; i <= size; ++i) {
        parent[i] = -1;
        f[i] = i;
    }
    for (int i = 1; i <= size; ++i) {
        int j = find(min(size, d[i]));
        if(f[j]!= 0) {
            res.push_back(i);
            int l = find(f[j]-1);
            unionArray(l, j);
            f[j] = f[l];
        }
        for (int k = 0; k <= size; ++k) {
            cout << f[k] << ends;
        }
        cout << endl;
    }
    for (int re : res) {
        cout << re << ends;
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章