Codeforces - 1213F - Unstable String Sort（建图 + 拓扑序 dp）

题目链接：https://codeforces.com/problemset/problem/1213/F

9.1 题意

给你两个长度为 $n(1\le n\le 2\cdot 10^5)$ 的排列 $p$ 和 $q$，现在需要构造一个至少有 $k(1\le k\le 26)$ 个不同字母的字符串 $s$，并且满足对于任意的 $i\in [1,n-1]$，都有 $s[p_i]\le s[p_{i+1}]$ 并且 $s[q_i]\le s[q_{i+1}]$。

如果无解，输出 NO。

9.2 解题过程

场上一下子就想到了建边！

根据排列，建立有向边 $<p_i,p_{i+1}>$ 和 $<q_i,q_{i+1}>$，有向边 $<a,b>$ 的含义为 $a$ 位置的字符要小于等于 $b$ 位置的字符。

很显然，一个强连通分量中的所有点都必须是同一种字母。

因此考虑 Tarjan 缩点，缩点之后形成一张有向无环图，这个时候每个缩出来的点都填上同一种字母。

因为字母种类要尽可能多，所以考虑在拓扑序上面跑最长路，设 $dp[i]$ 表示截止到 $i$ 的最长路长度。在拓扑排序过程中，如果当前在处理边 $<u,v>$，则进行转移 $dp[v]=\max (dp[v],dp[u]+1)$。

这时发现，如果最大的 $dp$ 值仍然小于 $k-1$ ，证明无解。

否则，贪心地构造解即可。

时间复杂度：$O(n)$。

9.3 错误点

忘记判最大的 $dp$ 值与 $k-1$ 的关系，导致 WA3。

9.4 代码

int head[maxn], n, m, cnt, tot, top, num;
int dfn[maxn], low[maxn], st[maxn], col[maxn];
bool instack[maxn];
struct edge
{
    int v, nxt;
} Edge[2 * maxn];
void init()
{
    for(int i = 0; i <= n; i++) head[i] = -1;
    cnt = 0;
}
void addedge(int u, int v)
{
    Edge[cnt].v = v;
    Edge[cnt].nxt = head[u];
    head[u] = cnt++;
}
void tarjan(int id)
{
    dfn[id] = low[id] = ++tot;
    st[++top] = id;
    instack[id] = true;
    for(int i = head[id]; i != -1; i = Edge[i].nxt)
    {
        int v = Edge[i].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[id] = min(low[id], low[v]);
        }
        else if(instack[v])
            low[id] = min(low[id], dfn[v]);
    }
    if(dfn[id] == low[id])
    {
        int v;
        num++;
        do
        {
            v = st[top--];
            instack[v] = false;
            col[v] = num;
        } while(v != id);
    }
}
int p[maxn], q[maxn];
int indeg[maxn];
int dp[maxn];
vector<pii> ve;
bool toposort() {
    queue<int> que;
    memset(dp, 0, sizeof(dp));
    for (int i = 1; i <= num; i++) {
        if (!indeg[i]) {
            que.push(i);
            dp[i] = 0;
        }
    }
    int cnt2 = 0;
    while (!que.empty()) {
        int now = que.front();
        que.pop();
        cnt2++;
        for (int i = head[now]; i != -1; i = Edge[i].nxt) {
            int v = Edge[i].v;
            indeg[v]--;
            dp[v] = max(dp[v], dp[now] + 1);
            if (!indeg[v]) {
                que.push(v);
            }
        }
    }
    return cnt2 == num;
}
char ans[maxn];
int main()
{
    int k;
    scanf("%d%d", &n, &k);
    init();
    for (int i = 1; i <= n; i++) {
        scanf("%d", &p[i]);
    }
    for (int i = 1; i <= n; i++) {
        scanf("%d", &q[i]);
    }
    for (int i = 1; i < n; i++) {
        addedge(p[i], p[i + 1]);
        addedge(q[i], q[i + 1]);
        ve.pb({p[i], p[i + 1]});
        ve.pb({q[i], q[i + 1]});
    }
    for (int i = 1; i <= n; i++) {
        if (!dfn[i]) tarjan(i);
    }
    init();
    for (auto ed: ve) {
        int u = ed.Fi;
        int v = ed.Se;
        if (col[u] == col[v]) continue;
        addedge(col[u], col[v]);
        indeg[col[v]]++;
    }
    if (!toposort()) {
        puts("NO");
        return 0;
    }
    int maxvalue = 0;
    for (int i = 1; i <= num; i++) maxvalue = max(maxvalue, dp[i]);
    if (maxvalue < k - 1) {
        puts("NO");
        return 0;
    }
    puts("YES");
    for (int i = 1; i <= n; i++) {
        if (dp[col[i]] >= k) {
            ans[i] = char('a' + (k - 1));
        } else {
            ans[i] = char('a' + dp[col[i]]);
        }
    }
    printf("%s\n", ans + 1);
    return 0;
}