笔试题 热气球 可以最多拿多少个金币

题目描述

一个热气球，每次飞行的最远距离是M，每次降落后就得再重启一次，且热气球最多重启K次。

有N个宝藏，宝藏位置由近到远排列，是当前宝藏到第0个宝藏的距离，所以第0个宝藏的位置为0。每个宝藏里面含有a[i]个金币。

热气球初始处于0位置，求热气球在该条件下能获得的最大金币数。

输入N, M, K，表示有N为宝藏的个数，M为热气球每次重启能飞行的最远距离，K为热气球的重启次数。
然后输入N行，每行两个数，第一个数是当前宝藏距离第0个宝藏的距离，第二个数是当前宝藏的金币数。

5 10 2
0 5
8 6
10 8
18 12
22 15

输出为

25

思路

dp[k][j]
定义为第K次重启，落在第j个宝藏处的最大金币。
那么有递推关系

// 当且仅当i-1次重启可以到达jt，且jt位置通过一次重启可以达到j位置的宝藏时
dp[i][j] = max(dp[i][j], dp[i-1][jt]+coin[j]);

代码

#include <bits/stdc++.h>
using namespace std;

int dp[105][105]; // dp[k+1][n]
int dis[105];
int coin[105];
int main()
{
    int n, m, k;
    scanf("%d%d%d", &n, &m, &k);
    for (int i=1; i<=n; ++i) {
        scanf("%d%d", &dis[i], &coin[i]);
    }
    memset(dp, 0, sizeof(dp));
    dp[0][1] = coin[1];

    for (int i=1; i<=k; ++i) {
        for(int j=2; j<=n; ++j) {
            for(int jt=j-1; dis[j]-dis[jt]<=m && jt >0; --jt) {
                if (dp[i-1][jt] == 0) continue; // i-1跳是否可达jt位置
                dp[i][j] = max(dp[i][j], dp[i-1][jt]+coin[j]);
            }
        }
     }
    int maxi = 0;
    for (int j=0; j<=n; j++) {
        maxi = max(maxi, dp[k][j]);
    }

    printf("%d\n", maxi);
    system("pause");
    return 0;
}