UESTC - 方伯伯的儿童节

本文探讨了使用动态规划(DP)算法解决路径优化问题的策略,详细介绍了算法实现步骤,包括初始化、递推公式设定及边界条件处理。通过实例分析展示了如何应用DP算法在特定场景下求解最优路径,提供了实用的编程代码示例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

DP

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>
#define MOD 10007
#define eps 1e-10
#define maxn 100010
#define INF (int)(1e5+7)
using namespace std;
typedef long long ll;
struct node {
    int to, w, next;
};
node edge[maxn << 2];
int head[maxn];
int dp[maxn][2];
ll num[maxn][2];
int vis[maxn];
int dfs(int cur, int col) {
    if (dp[cur][col] != -1) return dp[cur][col];
    vis[cur] = 1;
    int res = col;
    ll sum = 1;
    for (int k = head[cur]; k != -1; k = edge[k].next) if (!vis[edge[k].to]) {
        if (col == 1) {
            res += min(dfs(edge[k].to,1), dfs(edge[k].to,0));
            if (dfs(edge[k].to,1) == dfs(edge[k].to,0)) {
                sum *= (num[edge[k].to][0] + num[edge[k].to][1]);
            } else if (dfs(edge[k].to,1) < dfs(edge[k].to,0)) {
                sum *= num[edge[k].to][1];
            } else {
                sum *= num[edge[k].to][0];
            }
        }
        else {
            res += dfs(edge[k].to, 1);
            sum *= num[edge[k].to][1];
        }
        sum %= MOD;
    }
    vis[cur] = 0;
    num[cur][col] = sum;
    return dp[cur][col] = res;
}
int main() {
    int t;
    scanf("%d", &t);
    while (t --) {
        int n;
        scanf("%d", &n);
        int p = 0;
        memset(dp, -1, sizeof(dp));
        memset(vis, 0, sizeof(vis));
        memset(num, 0, sizeof(num));
        memset(head, -1, sizeof(head));
        for (int i = 0; i < n-1; ++ i) {
            int u, v;
            scanf("%d%d", &u, &v);
            edge[p].to = v;
            edge[p].next = head[u];
            head[u] = p;
            ++ p;
            edge[p].to = u;
            edge[p].next = head[v];
            head[v] = p;
            ++ p;
        }
        int res = min(dfs(1,1), dfs(1,0));
        ll sum = 0;
        if (dfs(1,1) == dfs(1,0)) {
            sum = num[1][1] + num[1][0];
        }
        else if (dfs(1,1) <= dfs(1,0)) {
            sum = num[1][1];
        }
        else {
            sum = num[1][0];
        }
        sum %= MOD;
        printf("%d %lld\n", res, sum);

    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值