【BZOJ】1040 [ZJOI2008]骑士 树形DP

xiaoxiao2021-02-28  86

题目传送门

好难啊,我太菜了,连题解都看不懂啊……orz教主JN,把一道如此难题讲得如此精简,JN是最强的!(这是JN邪教的信条!)

翻开题解,都是一群大佬在刷什么基环外向树,吓得我抱起了抱着抱着抱着我的我的我的我。(这TMD什么玩意儿?!)

这道题给出的是一张n个节点、n条边的有向图。但只要稍微脑补一下,其实是无向图。

题目并没有说整张图是联通的,所以这是一片基环外向树森林。

对于普通的树,我们直接树形DP就行了;

对于基环外向树,我们要考虑删去环上的一条边,这样这棵基环外向树就变成了普通的树了,然后对于删去的边的两端节点分别做一次树形DP。

令f[i][0]表示不取第i号节点的最大价值,f[i][1]则反之。

因为这两个节点之间是有边的,也就是这两个节点不能同时出现在答案集合中,设这两个节点分别为u和v,所以我们取max(f[u][0],f[v][0])加入答案即可。

附上AC代码:

#include <cstdio> #include <cctype> #include <algorithm> #define N 1001000 using namespace std;   struct side{     int to,nt; }s[N<<1]; int n,x,a[N],num=1,h[N],fx,fy,edge; long long f[N][2],ans,tmp; bool b[N];   inline char nc(){     static char ch[100010],*p1=ch,*p2=ch;     return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++; }   inline void read(int &a){     static char c=nc();int f=1;     for (;!isdigit(c);c=nc()) if (c=='-') f-1;     for (a=0;isdigit(c);a=a*10+c-'0',c=nc());     a*=f;return; }   inline void add(int x,int y){     s[++num]=(side){y,h[x]},h[x]=num;     s[++num]=(side){x,h[y]},h[y]=num; }   inline void so(int x,int e){     b[x]=1;     for (int i=h[x]; i; i=s[i].nt)         if ((i^1)!=e){             if (b[s[i].to]){                 fx=x,fy=s[i].to,edge=i;                 continue;             }             so(s[i].to,i);         }     return; }   inline void work(int x,int e,int ed){     f[x][1]=a[x],f[x][0]=0;     for (int i=h[x]; i; i=s[i].nt)         if ((i^1)!=e&&i!=ed&&(i^1)!=ed){             work(s[i].to,i,ed);             f[x][0]+=max(f[s[i].to][0],f[s[i].to][1]);             f[x][1]+=f[s[i].to][0];         }     return; }   int main(void){     read(n);     for (int i=1; i<=n; ++i) read(a[i]),read(x),add(x,i);     for (int i=1; i<=n; ++i)         if (!b[i]){             so(i,0);             work(fx,0,edge),tmp=f[fx][0];             work(fy,0,edge),tmp=max(tmp,f[fy][0]);             ans+=tmp;         }     printf("%lld",ans);     return 0; }

转载请注明原文地址: https://www.6miu.com/read-53686.html

最新回复(0)