c++ - 法丽达公主 on Spoj

Question

我遇到了一个问题 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

Answer 1

您使用错误的数据类型将值存储在 vector dp 中。
由于硬币的总和可以达到 (10^9*10^2=10^11) 因此 int 将无法存储它。尝试使用 long long int 代替，因为它不会导致溢出情况。
与您的代码相同(使用 long long int 被接受):
使用: vector < long long 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();


}
}