codeforces 919D 记忆化+判环

题意:在有向图中找出一条路径上边出现的相同颜色最多的,输出这个相同颜色,有环输出-1.
思路:对于每一个点进行记忆化搜索,这里的vis数组的妙用,vis == -1表示未访问过,vis == 0表示已经访问过,vis == 1表示这个节点所有子树已经访问过,所以可以返回值。判断环的话出现了vis == 0那么表示有环。

#include<iostream>
#include<cstring>
#include<queue>
#include<cstdio>
using namespace std;
#define clr(x,y) memset(x,y,sizeof x)
typedef long long ll;
const int maxn = 3e5 + 10;
#define INF 0x3f3f3f3f
const ll Mod = 1e9 + 7;

struct Edge{int to,next;}edge[maxn << 1];
int head[maxn],vis[maxn],edge_num;//vis == -1表示未访问过,vis == 0表示这个点已经被访问了,vis == 1表示这个点的所有子节点全部被访问了
int dp[maxn][26];
char s[maxn];
void Init()
{
    edge_num = 0;clr(head,-1);clr(dp,0);clr(vis,-1);
}
void add_edge(int x,int y)
{
    edge[edge_num] = (Edge){y,head[x]};head[x] = edge_num ++;
}

bool dfs(int u)
{
    if(vis[u] == 0)return false;
    if(vis[u] == 1)return true;
    vis[u] = 0;
    for(int i = head[u];i != -1;i = edge[i].next)
    {
        int v = edge[i].to;
//        cout << u << " -> " << v << endl;
        if(!dfs(v))return false;
        for(int j = 0;j < 26;j ++)
        {
            dp[u][j] = max(dp[u][j],dp[v][j]);
        }
    }
    vis[u] = 1;dp[u][s[u] - 'a'] ++;
    return true;
}
int n,m;
int solve()
{
    for(int i = 1;i <= n;i ++)
    {
        if(vis[i] == -1)
        {
            if(!dfs(i))return -1;
        }
    }
    int ans = 0;
    for(int i = 1;i <= n;i ++)
        for(int j = 0;j < 26;j ++)
        ans = max(ans,dp[i][j]);
    return ans;
}
int main()
{

    while( ~ scanf("%d%d",&n,&m))
    {
        Init();scanf("%s",s + 1);
        for(int i = 1;i <= m;i ++)
        {
            int x,y;scanf("%d%d",&x,&y);
            add_edge(x,y);
        }
        int ans = solve();
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xiaolonggezte/article/details/79225301