[Tarjan][并查集][dp] Jzoj P4253 QYQ在艾泽拉斯

Description

在艾泽拉斯的无尽之海里，有着一群不为人知的由各个种族的冒险者统治的岛屿，这些岛屿都很庞大，足以在上面建造许多的城市，城市之间有一些单向道路连接。
有一天，QYQ无意中发现了这些岛屿，并且发现在每个城市的地下都或多或少埋藏着一些装备、金币、宝物……
可是正当QYQ兴奋不已打算全部把它们拿走时，他却惊奇的发现你的魔法在这里被限制住了，唯一可用的技能就是闪现，而且魔法只够他使用K次这个技能了，每次使用这个技能QYQ只能从一个岛屿上闪现到另外一个岛屿上。每一个岛屿只能登上一次，QYQ可以从任何一个城市开始旅程，在任何一个城市结束旅程。
城市的数量共有n个，有m条道路，每一条道路有两个参数u，v，表示从u到v有一条道路，但你只能由u到v走，两个城市属于相同的岛屿当且仅当暂时将所有道路视为双向道路时可以从其中一个城市走到另一个城市（可以途径其它城市）。
每一个城市都有一个宝物的总价值v[i]，你的任务是帮助QYQ得到最大总价值的宝物，并输出这个值。

 

Input

从文件azeroth.in中输入数据。
输入的第一行包含两个整数n，m
输入的第二行到第m+1行，每行包含2个整数u,v，代表你可以从城市u走到城市v
输入的第m+2行包含n个整数，第i个整数代表v[i]，即这个城市的宝物总价值。
输入的第m+3行包含一个整数K，代表你可以使用技能的次数。

Output

输出到文件azeroth.out中。
输出的第一行包含一个整数，代表QYQ能获得的最大的宝物总价值

 

Sample Input

3 2
1 2
3 1
1 2 1
0

Sample Output

4
样例说明：
QYQ从3号点开始，走到2号点，最后走到1号点，结束旅程，共获得1+2+1=4价值的宝物

 

Data Constraint

对于30%的数据：n<=10，K=0
对于50%的数据：n<=100，m<=100，K<=1
对于100%的数据：1<=n<=100000，1<=m<=1000000，1<=v[i]<=1000，0<=K<=100000
图中可能会有重边、自环。

 

题解

• 题目大意：给定一个有向有环图，每个环只能走一次，有k次在任意环上跳的次数，问可以得到的最大价值
• 这种题，先来个Tarjan缩点，然后就变成了有向无环图，就是一个DAG
• 然后我们只用处理每个环，也就是每个有向的连通块，进行dp，设f[i]为从第i个连通块开始能获得的最大价值
• 那么一个有向的连通块的最大价值，就是所有在这个连通块中的所有点能获得的最大价值
• 最后答案就是所有连通块价值的前k个之和

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e5+10,M=1e6+10;
int n,m,num,tot,K,Ans,cnt[2],sum[N],head[2][N],f[N],a[N],Num[N],bel[N],ans[N],fa[N],dfn[N],low[N],p[N];
bool b[N],vis[N];
struct edge { int x,y; }e[2][M];
void insert(int p,int x,int y) { e[p][++cnt[p]].x=y,e[p][cnt[p]].y=head[p][x],head[p][x]=cnt[p]; }
int getfather(int x) { return fa[x]==x?x:fa[x]=getfather(fa[x]); }
void Tarjan(int x)
{
    dfn[x]=low[x]=++dfn[0],p[++tot]=x,vis[x]=1;
    for (int i=head[0][x];i;i=e[0][i].y)
        if (!bel[e[0][i].x])
        {
            if (vis[e[0][i].x]) low[x]=min(low[x],dfn[e[0][i].x]);
            else Tarjan(e[0][i].x),low[x]=min(low[x],low[e[0][i].x]);
        }
    if (dfn[x]==low[x])
    {
        num++; int v;
        do
        {
            v=p[tot--],bel[v]=num,sum[num]+=a[v],vis[v]=0;
        }while(v!=x);
    }
}
void work(int x)
{
    b[x]=1,f[x]+=sum[x];
    for (int i=head[1][x];i;i=e[1][i].y)
    {
        getfather(x),getfather(e[1][i].x);
        fa[fa[x]]=fa[e[1][i].x],Num[e[1][i].x]--,f[e[1][i].x]=max(f[x],f[e[1][i].x]);
        if (!Num[e[1][i].x]) work(e[1][i].x);
    }
}
int main()
{
    freopen("azeroth.in","r",stdin),freopen("azeroth.out","w",stdout),scanf("%d%d",&n,&m);
    for (int i=1,x,y;i<=m;i++) { scanf("%d%d",&x,&y); if (x==y) continue; insert(0,x,y); }
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    scanf("%d",&K);
    for (int i=1;i<=n;i++) if (!bel[i]) Tarjan(i);
    for (int i=1;i<=n;i++)
        for (int j=head[0][i];j;j=e[0][j].y)
            if (bel[i]!=bel[e[0][j].x])
                insert(1,bel[i],bel[e[0][j].x]),Num[bel[e[0][j].x]]++;
    for (int i=1;i<=num;i++) fa[i]=i;
    for (int i=1;i<=num;i++) if (!Num[i]&!b[i]) work(i);
    for (int i=1;i<=num;i++) getfather(i),ans[fa[i]]=max(ans[fa[i]],f[i]);
    sort(ans+1,ans+num+1);
    for (int i=num;i>=max(1,num-K);i--) Ans+=ans[i];
    printf("%d",Ans);
}