1. 程式人生 > >創世紀 BZOJ3037 & [Poi2004]SZP BZOJ2068

創世紀 BZOJ3037 & [Poi2004]SZP BZOJ2068

algo add 樹形 spa i++ using ring urn for

分析:

樹形DP中的一種,基環樹DP

針對每一個環跑DP,f[i],g[i]分別表示選或者不選,之後我們註意每次遍歷的時候,不要重復遍歷。

附上代碼:

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <iostream>
#include <queue>
#include <cstring>
#include <cstdlib>
using namespace std;
#define N 1000005
struct node
{
    int to,next;
}e[N<<1];
int head[N],cnt,f[N],g[N],fa[N],now,n,ra[N],rb[N];
void add(int x,int y)
{
    e[cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt++;
    return ;
}
int find(int x){if(fa[x]==x)return x;return fa[x]=find(fa[x]);}
void dfs(int x)
{
    int minn=1<<30;
    g[x]=0;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int to1=e[i].to;
        if(to1!=now)dfs(to1);
        g[x]+=max(f[to1],g[to1]);
        minn=min(minn,max(f[to1],g[to1])-g[to1]);
    }
    f[x]=g[x]+1-minn;
}
int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d",&n);int cnt=0;
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        if(find(x)!=find(i))
        {
            add(x,i);
            fa[fa[x]]=fa[i];
        }else ra[++cnt]=x,rb[cnt]=i;
    }
    int ans=0;
    for(int i=1;i<=cnt;i++)
    {
        dfs(ra[i]);now=ra[i];
        dfs(rb[i]);
        int t=f[rb[i]];
        f[ra[i]]=g[ra[i]]+1;
        dfs(rb[i]);ans+=max(t,g[rb[i]]);
    }
    printf("%d\n",ans);
    return 0;
}

  

創世紀 BZOJ3037 & [Poi2004]SZP BZOJ2068