码迷,mamicode.com
首页 > 其他好文 > 详细

【模板】缩点

时间:2018-09-07 23:16:47      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:init   oid   printf   scan   dfs   bre   pre   its   class   

题目背景

缩点+DP

题目描述

给定一个n个点m条边有向图,每个点有一个权值,求一条路径,使路径经过的点权值之和最大。你只需要求出这个权值和。

允许多次经过一条边或者一个点,但是,重复经过的点,权值只计算一次。

输入输出格式

输入格式:

 

第一行,n,m

第二行,n个整数,依次代表点权

第三至m+2行,每行两个整数u,v,表示u->v有一条有向边

 

输出格式:

 

共一行,最大的点权之和。

 

输入输出样例

输入样例#1: 复制
2 2
1 1
1 2
2 1
输出样例#1: 复制
2

说明

n<=10^4,m<=10^5,点权<=1000

算法:Tarjan缩点+DAGdp

 

缩点之后建新图,记忆化搜索

#include <bits/stdc++.h>
#define maxn 100005
using namespace std;
struct edges
{
    int u,v,next;
};
struct m
{
    edges edge[maxn];
    int head[maxn];
    int cnt=0;
    void init()
    {
        memset(head,-1,sizeof(head));
    }
    void addedge(int u,int v)
    {
        edge[cnt].u=u;
        edge[cnt].v=v;
        edge[cnt].next=head[u];
        head[u]=cnt++;
    }
}M,Mp;
int f[maxn],ind[maxn],dfn[maxn],low[maxn],a[maxn],in[maxn];
bool vis[maxn];
int cnt,tot,sum;
stack<int> s;
void Tarjan(int num)
{
    dfn[num]=low[num]=++tot;
    vis[num]=true;
    s.push(num);
    for(int i=M.head[num];i!=-1;i=M.edge[i].next)
    {
        int v=M.edge[i].v;
        if(!dfn[v])
        {
            Tarjan(v);
            low[num]=min(low[num],low[v]);
        }
        else if(vis[v])
        {
            low[num]=min(low[num],dfn[v]);
        }
    }
    if(low[num]==dfn[num])
    {
        sum++;
        while(true)
        {
            int now=s.top();
            s.pop();
            vis[now]=false;
            ind[now]=sum;
            if(now==num) break;
        }
    }
}
int dp[maxn];
void dfs(int num)
{
    if(dp[num]) return;
    dp[num]=f[num];
    int maxim=0;
    for(int i=Mp.head[num];i!=-1;i=Mp.edge[i].next)
    {
        int v=Mp.edge[i].v;
        if(!dp[v]) dfs(v);
        maxim=max(maxim,dp[v]);
    }
    dp[num]+=maxim;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int n,m,i,u,v;
    scanf("%d%d",&n,&m);
    M.init();
    Mp.init();
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        M.addedge(u,v);
    }
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i]) Tarjan(i);
    }
    for(int i=1;i<=n;i++)
    {
        f[ind[i]]+=a[i];
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=M.head[i];j!=-1;j=M.edge[j].next)
        {
            v=M.edge[j].v;
            if(ind[i]!=ind[v])
            {
                in[v]++;
                Mp.addedge(ind[i],ind[v]);
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        if(!dp[i])
        {   dfs(i);
            ans=max(ans,dp[i]);
        }
    }
    printf("%d\n",ans);
    return 0;
}

  

【模板】缩点

标签:init   oid   printf   scan   dfs   bre   pre   its   class   

原文地址:https://www.cnblogs.com/zyf3855923/p/9607452.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!