bzoj1179 Atm

题意

有向图，从起点出发到终点集合的某个点结束，求路径上的点权之和的最大值（如果某点重复经过，点权只被计算一次）

分析

Tarjan缩点重构图，再用Spfa求最长路
有向图用Tarjan一定要有vis标记

#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#include<cstring>
using namespace std;
#define MAXN 500010
int head[MAXN],nxt[MAXN],p[MAXN],val[MAXN];
int low[MAXN],dfn[MAXN],c[MAXN],stack[MAXN],tot,cnt;
bool vis[MAXN];
void Tarjan(int u)
{
    dfn[u]=low[u]=++tot;
    stack[++cnt]=u;
    vis[u]=1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=p[i];
        if(!dfn[v])
        {
            Tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(vis[v])
        {
            low[u]=min(low[u],dfn[v]);
        }
    }
    if(low[u]==dfn[u])
    {
        do
        {
            val[u]+=c[stack[cnt]];
            c[stack[cnt]]=u;
            vis[stack[cnt--]]=0;
        }while(u!=stack[cnt+1]);
    }
}
map<int,bool>nw[MAXN];
int Adj[MAXN],nx[MAXN],V[MAXN],n;
void Add(int u,int v)
{
    n++;
    V[n]=v;
    nx[n]=Adj[u];Adj[u]=n;
}
void DFS(int u)
{
    vis[u]=1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=p[i];
        if(!vis[v]) DFS(v);
        if(c[u]!=c[v]&&nw[c[u]][c[v]]==0)
            Add(c[u],c[v]),nw[c[u]][c[v]]=1;
    }
}
//************Tarjan缩点重构图************************
int dis[MAXN];
void SPFA(int s)
{
    memset(vis,0,sizeof vis);
    queue<int>Q;
    Q.push(s);
    vis[s]=1;
    dis[s]=val[s];
    while(!Q.empty())
    {
        int u=Q.front();Q.pop();
        vis[u]=0;
        for(int i=Adj[u];i;i=nx[i])
        {
            int v=V[i];
            int D=val[v];
            if(dis[v]<dis[u]+D)
            {
                dis[v]=dis[u]+D;
                if(!vis[v])
                {
                    Q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
}
int bar[MAXN];
int main()
{
    int N,M,u,v,s,m;
    scanf("%d%d",&N,&M);
    for(int i=1;i<=M;i++)
    {
        scanf("%d%d",&u,&v);
        p[i]=v,nxt[i]=head[u];head[u]=i;
    }
    for(int i=1;i<=N;i++)
        scanf("%d",&c[i]);
    scanf("%d%d",&s,&m);
    Tarjan(s);
    DFS(s);
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&u);
        bar[i]=c[u];
    }
    SPFA(c[s]);
    int ans=0;
    for(int i=1;i<=m;i++)
        ans=max(ans,dis[bar[i]]);
    printf("%d\n",ans);
}