Codeforces 919D. Substring (拓扑序)

传送门: http://codeforces.com/contest/919/problem/D

题意:

给出一个长为 n 的字符串, 每一个字符是一个点, 给出 m 条连接两个字符的单向边
定义路径的价值为路径上字符最多的字符的数量.
求图的路径价值的最大值.

分析:

在不存在环的情况下, 最大价值路径一定是拓扑序中从首至尾的所有路径中的一个
按照拓扑序遍历, 更新下一个点的所有字母的数量即可求出答案.

代码:

#include <stdio.h>
#include <cstring>
using namespace std;
const int MAX = 3e5+10;
int n, m, u, v, ecnt, head[MAX], ind[MAX], qit, que[MAX], mmax, tmp, node[MAX][26];
char s[MAX];
struct Edge {
    int v, nxt;
} edge[MAX];
int max(int a, int b) {
    return a > b ? a : b;
}
void addedge(int u, int v) {
    edge[ecnt].v = v;
    edge[ecnt].nxt = head[u];
    head[u] = ecnt++;
    ++ind[v];
}
void getin() {
    for (int i = 0; i < m; ++i) {
        scanf("%d%d", &u, &v);
        addedge(u-1, v-1);
    }
}
void init() {
    memset (head, -1, sizeof (head[0]) * n);
    memset (ind, 0, sizeof (ind[0]) * n);
    memset (node, 0, sizeof (node[0]) * n);
    qit = 0, mmax = -1, ecnt = 0;
    getin();
    for (int i = 0; i < n; ++i)
        if (ind[i] == 0) {
            que[qit++] = i;
            node[i][s[i]-'a']++;
        }
}
void judge() {
    for (int i = 0; i < qit; ++i) {
        for (int j = head[que[i]]; j != -1; j = edge[j].nxt) {
            u = que[i], v = edge[j].v;
            for (int k = 0; k < 26; ++k) {
                tmp = node[u][k] + (k == (s[v] - 'a'));
                if (tmp > node[v][k]) {
                    node[v][k] = tmp;
                    mmax = max(mmax, node[v][k]);
                }
            }
            if (!--ind[v])
                que[qit++] = v;
        }
    }
}
int main() {
    scanf("%d%d%s", &n, &m, s);
    init();         
    judge();
    printf("%d\n", (qit < n ? -1 : mmax));
    return 0;
}

ps: cin 改为 scanf, string 改成 char[] 之后时间 1000ms -> 150ms…..

2018-02-02