问题描述
地面上有一排长度为 n 的格子 1−n,每个格子上都有一个数 xi,开始时你在位置 0 ,每次你可以向前跳 1-2 格,然后取走格子上的数,直到跳到位置n+1。
取走的数的和就是你的得分,现在你想知道你可能的最大得分是多少。
输入
一行四个整数 n,A,B,C (n≤100000,0≤A,B,C≤10000),其中n表示格子的数量。
x[i] 由如下方式生成:
for (int i = 1; i <= n; i++){
int tmp = ((long long)A * i * i + B * i + C) % 20000;
x[i] = tmp - 10000;
}
输出
一行一个整数 ans表示可能的最大得分。
样例
输入
3 1 1 1
输出
-9993
代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
int n, A, B, C;
cin >> n >> A >> B >> C;
vector<int> x(n+2); // 格子1-n,0和n+1为边界
for (int i = 1; i <= n; i++) {
long long tmp = (A * 1LL * i * i + B * 1LL * i + C) % 20000;
x[i] = tmp - 10000;
}
vector<int> dp(n+2);
dp[0] = 0; // 起点
dp[1] = x[1];
if (n >= 2) {
dp[2] = max(x[2], dp[1] + x[2]);
}
for (int i = 3; i <= n+1; i++) {
dp[i] = max(dp[i-1], dp[i-2]) + (i <= n ? x[i] : 0);
}
cout << dp[n+1] << endl;
return 0;
}