G. Is Topo Logical?（贪心+拓扑排序理解）

G. Is Topo Logical?

题目链接

题目大意：给定 $n$ 个点，再给两个有 $n$ 个元素的序列，第一个序列表示初始 $n$ 个点的入度，第二个序列表示由这 $n$ 个点构成的图经过基于 $BFS$ 的拓扑排序后的入度，输出一种合法的构图，若不可能构成图，就输出 $-1$ 。

解：

（1）首先我们可以构建出最后经过拓扑后的图，也就是我们要对最后每个入度不为0的点建边，显然这些边的起点肯定只能来源于同样入度最后不为0的那些点，而且只要不连接自己，就随便连，直到满足自己的最终入度。
（2）随后我们可以考虑一个贪心的方法：拓扑排序初始删除的点肯定就是一开始入度就为0的点，那么此时需要建立被删边的那些点（也就是最终入度小于初始入度的点）肯定要尽量连接初始入度就为0的点，因为此时有机会建立被删边当然就要建立，题目规定了两个点的一个方向最多只能连接一条边，所以有机会连的话，不连白不连，不然后面没机会连，越早多连才能尽可能避免后面可能没有新的零入度点去连接的情况。那么这样连接还能保证一个最优：因为我们刚才建的边也就是第一步会删除的边，我们肯定期望删除这些边之后，新的零入度点尽可能多，这样才有更多的资源去分配给需要建立被删边的点。因为我们每次都会使得被删边尽可能多的建立（反过来说也就是对于某个入度会减少的点来说，拓扑一次会使得这个点的入度下降的最快），那么这样新的入度为零的点就会尽快出来，这样就能保证资源尽可能充足。所以我们就只要每次遍历所有的要建立被删边的点，对于每个点，用当前的零入度点连接它（能连几个是几个），如果这个点已经连接完所有被删边就删除这个节点，然后如果这个节点被删边都建立完成且最终入度为 $0$ ，那么它就可以作为新入度为 $0$ 的使用去进行下一次拓扑排序，连接其它要建立被删边的点。

因为在遍历时得动态删除要建立被删边的节点，所以可以用 $multiset$ 或者手写链表实现（STL库的链表的删除貌似是 $O(n)$的， 手写可以做到遍历的时候 $O(1)$ 删除）

O(nlogn)做法：

#include <bits/stdc++.h>
#define p pair<int, int>

using namespace std;

const int N = 2e5 + 2;

int n, sum;
int pre[N], last[N];
vector<int> de0;
vector<int> temp;
vector<p> ans;
vector<int> fl;
vector<int> shan;
multiset<int> del;
int deg[N];

bool solved() {
    while(!del.empty()) {
        if (de0.empty()) return true;
        temp.clear();
        shan.clear();
        for (auto i : del) {
            int limit = min(deg[i], (int)de0.size());
            for (int j = 0; j < limit; ++j) {
                int u = de0[j];
                ans.push_back({u, i});
            }
            if (deg[i] > de0.size()) {
                deg[i] -= (int)de0.size();
            }
            else {
                shan.push_back(i);
                if (last[i] == 0) {
                    temp.push_back(i);
                }
            }
        }
        de0.clear();
        for (int i = 0; i < (int)temp.size(); ++i) {
            de0.push_back(temp[i]);
        }
        for (int i = 0; i < (int)shan.size(); ++i) {
            del.erase(shan[i]);
        }
    }
    return false;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    freopen("topo.in", "r", stdin);
    int T;
    scanf("%d", &T);
    while(T--) {
        sum = 0;
        ans.clear();
        de0.clear();
        fl.clear();
        del.clear();
        bool bad = false;
        scanf("%d", &n);
        for (int i = 1; i <= n; ++i) {
            scanf("%d", &pre[i]);
            if (pre[i] == 0) {
                de0.push_back(i);
            }
            sum += pre[i];
        }
        for (int i = 1; i <= n; ++i) {
            scanf("%d", &last[i]);
            if (last[i] ^ 0) fl.push_back(i);
            if (last[i] ^ pre[i]) {
                del.insert(i);
                deg[i] = pre[i] - last[i];
            }
        }
        for (int i = 0; i < (int)fl.size(); ++i) {
            int u = fl[i], num = 0; 
            for (int j = 0; j < (int)fl.size(); ++j) {
                int v = fl[j];
                if (u ^ v) {
                    ans.push_back({v, u});
                    num++;
                }
                if (num == last[u]) break;
            }
            if (num < last[u]) {
                bad = true;
                break;
            }
        }
        if (bad) {
            printf("-1\n");
            continue;
        }
        bad = solved();
        if (bad) {
            printf("-1\n");
            continue;
        }
        printf("%d\n", sum);
        for (int i = 0; i < (int)ans.size(); ++i) {
            printf("%d %d\n", ans[i].first, ans[i].second);
        }
    }
}

O(n) （不知道为啥没快多少。。。）

#include <bits/stdc++.h>
#define P pair<int, int>
 
using namespace std;
 
const int N = 2e5 + 2;
 
struct List {
    int data;
    List *next;
};
int n, sum;
int pre[N], last[N];
vector<int> de0;
vector<int> temp;
vector<P> ans;
vector<int> fl;
int deg[N];
List *del;
List zu[N];
 
bool solved() {
    while(del->next) {
        if (de0.empty()) return true;
        temp.clear();
        List *pre = del, *p = del->next;
        while(p) {
            int v = p->data;
            int limit = min(deg[v], (int)de0.size());
            for (int j = 0; j < limit; ++j) {
                int u = de0[j];
                ans.push_back({u, v});
            }
            if (deg[v] > de0.size()) {
                deg[v] -= (int)de0.size();
            }
            else {
                pre->next = p->next;
                if (last[v] == 0) {
                    temp.push_back(v);
                }
                pre->next = p->next;
                p = pre->next;
                continue;
            }
            pre = pre->next;
            p = p->next;
        }
        de0.clear();
        for (int i = 0; i < (int)temp.size(); ++i) {
            de0.push_back(temp[i]);
        }
    }
    return false;
}
 
int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    freopen("topo.in", "r", stdin);
    del = &zu[0];
    int T;
    scanf("%d", &T);
    while(T--) {
        sum = 0;
        ans.clear();
        de0.clear();
        fl.clear();
        del->next = NULL; List *tail = del;
        bool bad = false;
        scanf("%d", &n);
        for (int i = 1; i <= n; ++i) {
            scanf("%d", &pre[i]);
            if (pre[i] == 0) {
                de0.push_back(i);
            }
            sum += pre[i];
        }
        for (int i = 1; i <= n; ++i) {
            scanf("%d", &last[i]);
            if (last[i] ^ 0) fl.push_back(i);
            if (last[i] ^ pre[i]) {
                List *p = &zu[i]; p->data = i, p->next = NULL;
                tail->next = p; tail = p;
                deg[i] = pre[i] - last[i];
            }
        }
        for (int i = 0; i < (int)fl.size(); ++i) {
            int u = fl[i], num = 0; 
            for (int j = 0; j < (int)fl.size(); ++j) {
                int v = fl[j];
                if (u ^ v) {
                    ans.push_back({v, u});
                    num++;
                }
                if (num == last[u]) break;
            }
            if (num < last[u]) {
                bad = true;
                break;
            }
        }
        if (bad) {
            printf("-1\n");
            continue;
        }
        bad = solved();
        if (bad) {
            printf("-1\n");
            continue;
        }
        printf("%d\n", sum);
        for (int i = 0; i < (int)ans.size(); ++i) {
            printf("%d %d\n", ans[i].first, ans[i].second);
        }
    }
}