洛谷[ZJOI2008]骑士(基环树二次DP法+树形DP)

简介: 洛谷[ZJOI2008]骑士(基环树二次DP法+树形DP)

原题链接

题意:

每个骑士有一个不可以同时上场的骑士,和一个战斗力。求最大战斗力。

思路:

类似没有上司的舞会。

这个题构成的是基环树森林,取每个基环树的最大值累加就是答案。

把每个基环树环上的一条边断开后,就构成了一棵树,这时候就相当于是没有上司的舞会那道题了。存图的时候默认的关系是两个人不能同时选择,也就需要分别强制选一个不选另一个,来跑两次DP,取max.

存图的时候存有向边,边的指向两种都可以。

注意答案会爆int。

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
inline ll read()
{
    ll x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
char F[200];
inline void out(I_int x)
{
    if (x == 0) return (void) (putchar('0'));
    I_int tmp = x > 0 ? x : -x;
    if (x < 0) putchar('-');
    int cnt = 0;
    while (tmp > 0)
    {
        F[cnt++] = tmp % 10 + '0';
        tmp /= 10;
    }
    while (cnt > 0) putchar(F[--cnt]);
    //cout<<" ";
}
ll ksm(ll a,ll b,ll p)
{
    ll res=1;
    while(b)
    {
        if(b&1)res=res*a%p;
        a=a*a%p;
        b>>=1;
    }
    return res;
}
const int inf=0x3f3f3f3f,mod=998244353;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double PI = atan(1.0)*4;
const int maxn=1e6+100;
int n,w[maxn],fa[maxn];
bool vis[maxn];
ll res;
int h[maxn],idx;
struct node{
    int e,ne;
}edge[maxn*2];
ll dp[maxn][2];
void add(int u,int v){
    edge[idx]={v,h[u]};h[u]=idx++;
}
void DP(int u,int root){
    vis[u]=1;
    dp[u][0]=0;dp[u][1]=w[u];
    for(int i=h[u];~i;i=edge[i].ne){
        int j=edge[i].e;
        if(j!=root){
            DP(j,root);
            dp[u][0]+=max(dp[j][0],dp[j][1]);
            dp[u][1]+=dp[j][0];
        }
        else dp[j][1]=-inf;
    }
}
void dfs(int u){
    vis[u]=1;
    int root=u;
    while(!vis[fa[root]]){
        root=fa[root];vis[root]=1;
    }
    DP(root,root);
    ll tmp=max(dp[root][0],dp[root][1]);
    vis[root]=1;root=fa[root];
    DP(root,root);
    tmp=max(tmp,max(dp[root][0],dp[root][1]));
    res+=tmp;
}
int main(){
    memset(h,-1,sizeof h);
    n=read();
    for(int i=1;i<=n;i++){
        w[i]=read();///该骑士的战斗力
        int x=read();///i讨厌x
        add(x,i);
        fa[i]=x;
    }
    for(int i=1;i<=n;i++)
        if(!vis[i]) dfs(i);///对每个没访问的点dfs
    printf("%lld\n",res);
  return 0;
}

参考:

目录
相关文章
|
1月前
|
机器学习/深度学习
【每日一题Day125】LC1326灌溉花园的最少水龙头数目 | 动态规划 贪心
【每日一题Day125】LC1326灌溉花园的最少水龙头数目 | 动态规划 贪心
37 0
|
20天前
蓝桥杯动态规划-第五弹 最大子数组和 买卖股票最佳时机IV 第N个泰波那契数 环形数组
蓝桥杯动态规划-第五弹 最大子数组和 买卖股票最佳时机IV 第N个泰波那契数 环形数组
|
5天前
【洛谷 P1002】[NOIP2002 普及组] 过河卒 题解(递归+记忆化搜索)
`NOIP2002`普及组的过河卒问题是一个棋盘路径计数挑战。卒从$(0,0)$出发到$(n,m)$,只能向下或向右移动,马在$(c1,c2)$固定,控制某些点。任务是计算不受马阻挡的路径数。输入是目标和马的位置,输出是路径总数。使用动态规划和记忆化搜索避免重复计算,样例输入$(6,6,3,3)$输出$6$。代码中定义了$f(x,y)$计算$(x,y)$处的路径数,利用边界条件和递推关系计算。
6 0
|
22天前
|
算法
【算法优选】 动态规划之两个数组dp——壹
【算法优选】 动态规划之两个数组dp——壹
|
1月前
|
算法
动态规划之第 N 个泰波那契数/三步问题【leetCode】【算法】
动态规划之第 N 个泰波那契数/三步问题【leetCode】【算法】
|
1月前
代码随想录Day36 动态规划05 LeetCode T1049最后一块石头的重量II T494 目标和 T474 一和零
代码随想录Day36 动态规划05 LeetCode T1049最后一块石头的重量II T494 目标和 T474 一和零
40 0
|
10月前
|
存储 算法 Java
dp算法 力扣174地下城游戏
dp算法 力扣174地下城游戏
|
1月前
【每日一题Day126】LC1140石子游戏Ⅱ | 博弈dp 记忆化搜索
【每日一题Day126】LC1140石子游戏Ⅱ | 博弈dp 记忆化搜索
35 0
|
8月前
|
算法
代码随想录Day28 贪心03 LeetCode T1005 K次取反后最大化的数组和 LeetCode T134 加油站 LeetCode T135 分发糖果
代码随想录Day28 贪心03 LeetCode T1005 K次取反后最大化的数组和 LeetCode T134 加油站 LeetCode T135 分发糖果
23 0
闫氏dp分析法(线性dp)AcWing 1015. 摘花生
闫氏dp分析法(线性dp)AcWing 1015. 摘花生
53 0