洛谷P2016 战略游戏 (树形dp)

简介: 洛谷P2016 战略游戏 (树形dp)

原题链接

思路:

dp[i][0]表示第i个点不放士兵所需的最少士兵数,dp[i][1]表示第i个点放士兵所需的最少士兵数。

转移:

如果当前节点不放士兵,子节点要放士兵;如果当前节点放士兵,子节点取min。

代码:

//#pragma GCC optimize(2)
#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;}
#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
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 maxn=1e6+7;
ll n,h[maxn],idx;
struct node{
    ll e,ne;
}edge[maxn];
void add(ll u,ll v){
    edge[idx]={v,h[u]},h[u]=idx++;
}
ll dp[maxn][2];
void dfs(ll u,ll fa){
    dp[u][0]=0;///不放
    dp[u][1]=1;
    for(ll i=h[u];i!=-1;i=edge[i].ne){
        int j=edge[i].e;
        if(j==fa) continue;
        dfs(j,u);
        dp[u][0]+=dp[j][1];
        dp[u][1]+=min(dp[j][0],dp[j][1]);
    }
}
void solve(){
    memset(h,-1,sizeof h);
    n=read;
    rep(t,0,n-1){
        int i=read;
        int m=read;
        rep(j,1,m){
            int x=read;
            add(x,i);add(i,x);
        }
    }
    dfs(0,-1);
    cout<<min(dp[0][0],dp[0][1])<<endl;
}
int main(){
    int T=1;
    while(T--) solve();
    return 0;
}
目录
相关文章
|
6月前
|
C语言
快速幂+高精乘(填坑)洛谷1226+1045
快速幂+高精乘(填坑)洛谷1226+1045
|
6月前
LeetCode题:174. 地下城游戏
LeetCode题:174. 地下城游戏
64 0
LeetCode题:174. 地下城游戏
|
1月前
acwing 110 抓住那头牛
acwing 110 抓住那头牛
9 0
|
4月前
【蓝桥杯】[递归]母牛的故事
蓝桥杯——[递归]母牛的故事
57 1
【蓝桥杯】[递归]母牛的故事
|
5月前
【洛谷 P1002】[NOIP2002 普及组] 过河卒 题解(递归+记忆化搜索)
`NOIP2002`普及组的过河卒问题是一个棋盘路径计数挑战。卒从$(0,0)$出发到$(n,m)$,只能向下或向右移动,马在$(c1,c2)$固定,控制某些点。任务是计算不受马阻挡的路径数。输入是目标和马的位置,输出是路径总数。使用动态规划和记忆化搜索避免重复计算,样例输入$(6,6,3,3)$输出$6$。代码中定义了$f(x,y)$计算$(x,y)$处的路径数,利用边界条件和递推关系计算。
68 0
|
算法
【过河卒】回溯算法保姆式解题
【过河卒】回溯算法保姆式解题
95 0
|
算法 C++
【每日算法Day 105】打家劫舍第二弹:看好你的电瓶车!
【每日算法Day 105】打家劫舍第二弹:看好你的电瓶车!
|
算法
【递归与递推】洛谷[NOIP2002 普及组] 过河卒
前言 本题来自洛谷P1002. 题目链接:[NOIP2002 普及组] 过河卒 - 洛谷
222 0