P3183 [HAOI2016]食物链 DP&图论 DAY 1 上午

上接：DP&图论 DAY 1 上午

 

P3183 [HAOI2016]食物链

题解

◦          给定n个点m条边的有向无环食物网，求其中有多少条极长食物链。

◦          n<=10^5，m<=2*10^5

>Solution

◦          拓扑图dp经典题

◦          设f[u]为以节点u为终点的食物链数量。

           

◦          按照拓扑序的顺序转移即可。

 处理的时候有些小细节：

扫描二维码关注公众号，回复： 7028811 查看本文章

1.记录每个点有多少入度：方便topsort

2.有多少出度：出度为0的非单点就是一条食物链的结束，这里可以用一个数组 vis 打标记（万一你记上单点了怎么办QAQ）

3.拓扑序转移：食物链起点的 f[u] 标记为 1 ，不然你以后咋转移？？？

 

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>

using namespace std;

inline int read()
{
    int ans=0;
    char last=' ',ch=getchar();
    while(ch<'0'||ch>'9') last=ch,ch=getchar();
    while(ch>='0'&&ch<='9') ans=ans*10+ch-'0',ch=getchar();
    if(last=='-') ans=-ans;
    return ans;
}

const int maxn=1e5+10;
const int maxm=2e5+10;
int n,m,ans=0;
int in[maxn],out[maxn],f[maxn];
int head[maxn],cnt=0;
bool vis[maxn];
struct node
{
    int u,v,nxt;
}edge[maxm];

void addedge(int u,int v)
{
    edge[++cnt].u =u;edge[cnt].v =v;
    edge[cnt].nxt =head[u];head[u]=cnt;
}

void topsort()
{
    queue<int>q;
    for(int u=1;u<=n;u++)
      if(!in[u])
        q.push(u),f[u]=1;
    while(!q.empty() )
    {
        int u=q.front() ;
        q.pop() ;
        for(int i=head[u];i;i=edge[i].nxt )
        {
            int v=edge[i].v ;
            f[v]+=f[u];
            in[v]--;
            if(!in[v]) q.push(v); 
        }
    } 
}

int main()
{
    n=read();m=read();
    for(int i=1,u,v;i<=m;i++)
    {
        u=read();v=read();
        addedge(u,v);
        in[v]++;
        out[u]++;
    }
    for(int i=1;i<=n;i++){  //打标记
        if(!out[i]&&in[i]) 
           vis[i]=1;
    } 
    topsort();
    for(int i=1;i<=n;i++){
    //    printf("f[%d]=%d ",i,f[i]);
        if(vis[i])
           ans+=f[i];
    }
      
    printf("%d",ans);
    
    
    return 0;
}