法丽达公主在斯波伊

Princess Farida on Spoj

本文关键字:公主      更新时间:2023-10-16

我卡在一个问题上 SPOJ。我检查了所有通过所有这些测试用例的测试用例,但我仍然在 spoj 上得到"WA"。
我知道它可以用动态编程来解决,但我正在练习记忆。

这是我的代码:

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector <int> dp(1000000);
long long int maxloot(vector<int> &loot, int n) {
    if (n == 0)
        return 0;
    if (n == 1)
        return loot[0];
    if (n == 2)
        return max(loot[0], loot[1]);
    if (dp[n] != -1)
        return dp[n];
    long long int take = loot[n - 1] + maxloot(loot, n - 2);
    long long int leave = maxloot(loot, n - 1);
    return dp[n]= max(take, leave);
}
int main() {
    int t;
    cin >> t;
    int p = 1;
    while (t--) {
        int n;
        cin >> n;
        vector <int> loot;
        for (int i = 0; i < n; i++) {
            int temp;
            cin >> temp;
            loot.push_back(temp);
        }
        dp.assign(1000000, -1);
        cout <<"Case "<<p<<": "<< maxloot(loot, n)<<endl;
        p++;
        dp.clear();

    }
}

测试用例 1:

5

1 2 3 4 5

测试用例 2:

1

10

输出 1:

9

产出2:

10

您使用了错误的数据类型在向量 dp 中存储值。由于硬币的总和可以达到(10^9*10^2=10^11),因此int将无法存储它。尝试改用长整型,因为它不会导致溢出情况。

与您的代码相同(使用长长 INT 被接受):

用途:向量<长 长=" int=">dp(1000000)

接受的代码:

#include<iostream>
#include<vector>
#include<algorithm>
#define ull unsigned long long
using namespace std;
vector <long long int> dp(1000000);
long long int maxloot(vector<int> &loot, int n) {
if (n == 0)
    return 0;
if (n == 1)
    return loot[0];
if (n == 2)
    return max(loot[0], loot[1]);
if (dp[n] != -1)
    return dp[n];
long long int take = loot[n - 1] + maxloot(loot, n - 2);
long long int leave = maxloot(loot, n - 1);
return dp[n]= max(take, leave);
}
int main() {
int t;
cin >> t;
int p = 1;
while (t--) {
    int n;
    cin >> n;
    vector <int> loot;
    for (int i = 0; i < n; i++) {
        int temp;
        cin >> temp;
        loot.push_back(temp);
    }
    dp.assign(1000000, -1);
    cout <<"Case "<<p<<": "<< maxloot(loot, n)<<endl;
    p++;
    dp.clear();

}
}