法丽达公主在斯波伊
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();
}
}
相关文章: