骑士:基环树

发布时间 2023-10-05 17:51:56作者: shirlybabyyy

https://www.bilibili.com/video/BV1Aa411Q7qp/?spm_id_from=333.999.0.0&vd_source=23dc8e19d485a6ac47f03f6520fb15c2

董老师讲的很清楚

https://www.luogu.com.cn/problem/P2607

 思路:

1、深搜找环

2、断环成树,对树深搜计算(断边:标记端点或者标记边)

O(N)

单向边:

// Luogu P2607 [ZJOI2008] 骑士
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e6+10;
typedef long long LL;
int n;
struct edge{int v,ne;}e[N];
int h[N],w[N],idx;
int r1,r2,vis[N];
LL f[N][2],sum;

void add(int a,int b){
  e[++idx]={b,h[a]};
  h[a]=idx;
}
void find(int u,int rt){//找两个根
  vis[u]=1;
  for(int i=h[u];i;i=e[i].ne){
    int v=e[i].v;
    if(v==rt){r1=u,r2=v;return;}
    if(vis[v])continue;
    find(v,rt);
  }
}
LL dfs(int u,int rt){//树上DP
  f[u][0]=0; f[u][1]=w[u];
  for(int i=h[u];i;i=e[i].ne){
    int v=e[i].v;
    if(v==rt)continue;
    dfs(v,rt);
    f[u][0]+=max(f[v][0],f[v][1]);
    f[u][1]+=f[v][0];
  }
  return f[u][0];
}
int main(){
  scanf("%d",&n);
  for(int v=1,u;v<=n;v++){
    scanf("%d%d",&w[v],&u);
    add(u,v);//单向边
  }
  for(int i=1;i<=n;i++){
    if(!vis[i]){
      r1=r2=0;
      find(i,i);
      if(r1){
        LL res1=dfs(r1,r1);
        LL res2=dfs(r2,r2);
        sum+=max(res1,res2);
      }
    }
  }
  printf("%lld",sum);
  return 0;
}

  

双向边:

// Luogu P2607 [ZJOI2008] 骑士
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e6+10;
typedef long long LL;
int n;
struct edge{int v,ne;}e[N<<1];
int h[N],w[N],idx=1;
int r1,r2,vis[N],be[N<<1];
LL f[N][2],sum;

void add(int a,int b){
  e[++idx]={b,h[a]};h[a]=idx;
}
bool find(int u,int ine){
    vis[u]=true;
    for(int i=h[u];i;i=e[i].ne){
      if(i==(ine^1))continue;
        int v=e[i].v;
        if(!vis[v]){//v尚未访问
            if(find(v,i)) return 1;
        } else{//v已访问
      r1=u,r2=v,be[i]=1,be[i^1]=1;
      return 1;
        }
    }
    return 0;
}
LL dfs(int u,int ine){//树上DP
  vis[u]=true;f[u][0]=0;f[u][1]=w[u];
  for(int i=h[u];i;i=e[i].ne){
    if(i==(ine^1)||be[i])continue;
    int v=e[i].v;
    dfs(v,i);
    f[u][0]+=max(f[v][0],f[v][1]);
    f[u][1]+=f[v][0];
  }
  return f[u][0];
}
int main(){
  scanf("%d",&n);
  for(int v=1,u;v<=n;v++){
    scanf("%d%d",&w[v],&u);
    add(u,v);add(v,u);
  }
  for(int i=1;i<=n;i++){
    if(!vis[i]){
      r1=r2=0;
      find(i,0);
      if(r1){
        LL res1=dfs(r1,0);
        LL res2=dfs(r2,0);
        sum+=max(res1,res2);
      }
    }
  }
  printf("%lld",sum);
  return 0;
}