[BZOJ1040][luogu2607][ZJOI2008]骑士（树形dp）

题目：

我是超链接

题解：

题目描述喜闻乐见
因为n个点n条边，环套树get？
边的两边只能选择一个点，但是因为有环，在判断的时候可能“不小心”加上相连的两个点权值，我们怎么办呢———把环中的一条边砍掉的话就没有问题了吧（成为树），在这棵树上跑树形dp，相当于是默认了这条边的两个端点不能同时选，分别以这两个点为起点并分别强制不选这两个点
正常树形dp最大值f[x][0/1]表示x点选不选
最后相连形成的树（环）不一定只有一个，所以要用for循环

代码：

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#define N 1000005
#define LL long long
using namespace std;
int tot=-1,nxt[N*2],point[N],v[N*2],l,r,s[N*2];LL f[N][2],a[N];
bool vis[N],fff;
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
    vis[x]=true;
    for (int i=point[x];i!=-1&&!fff;i=nxt[i])
      if (v[i]!=fa)
      {
        if (vis[v[i]]){l=v[i]; r=x; s[i]=s[i^1]=1; fff=true; break;}
        dfs(v[i],x);
      }
}
void treedp(int x,int fa,int cant)
{
    vis[x]=true;
    if (x==cant) f[x][1]=0;
    else f[x][1]=a[x];
    f[x][0]=0;
    for (int i=point[x];i!=-1;i=nxt[i])
      if (v[i]!=fa && !s[i])
      {
        treedp(v[i],x,cant);
        f[x][0]+=max(f[v[i]][0],f[v[i]][1]);
        f[x][1]+=f[v[i]][0];
      } 
}
int main()
{
    int n,i;
    memset(nxt,-1,sizeof(nxt));memset(point,-1,sizeof(point));
    scanf("%d",&n);
    for (i=1;i<=n;i++)
    {
        int y;scanf("%lld%d",&a[i],&y);
        addline(i,y);
    }
    LL ans=0;
    for (i=1;i<=n;i++)
      if (!vis[i])
      {
        fff=0;
        dfs(i,0);
        LL maxx=0;
        treedp(l,0,r);
        maxx=max(f[l][1],f[l][0]);
        treedp(r,0,l);
        maxx=max(maxx,max(f[r][1],f[r][0]));
        ans+=maxx;
      }
    printf("%lld",ans);
}