链接:https://www.nowcoder.com/acm/contest/159/C
来源:牛客网
题目描述
现在给你N个正整数ai,每个数给出一“好数程度” gi(数值相同但位置不同的数之间可能有不同的好数程度)。对于在 i 位置的数,如果有一在j位置的数满足 j < i 且 ai=aj,则你可以将位于[i,j]闭区间内的序列评为“好序列”,然后获得∑gk(j≤k≤i)(此闭区间内“好数程度”之和)分数。
注意: 在所有情况下,每个数都只能被一个”好序列”包含(只能与其他相应数被评为”好序列”一次);在符合要求的情况下,”好序列”的评定次数不受限制,且通过不同”好序列”获得的分数可以累加。
输入描述:
第一行有一个正整数N。
接下来的一行有N个正整数ai,表示意义如上。
(保证ai在32位整型范围内)
接下来的一行有N个正整数gi,表示ai的”好数程度”。
(保证gi在64位整型范围内)
输出描述:
一个整数,你可以获得的最大分数(通过不同”好序列”获得的分数可以累加),保证答案在64位整型范围内。
示例1
输入
复制
7
1 2 1 2 3 2 3
1 4 3 4 3 4 5
输出
复制
23
备注:
数据范围 2≤N≤300000
分析:
浅显易懂,维护
(1,i−1)
(
1
,
i
−
1
)
取得最大价值的那个位置,map记录
log2(N)
l
o
g
2
(
N
)
.
#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long LL;
map<LL, int> mp;
const int INF = 0x3f3f3f3f;
const int MAXN = 3e5 + 10;
LL dp[MAXN], a[MAXN], g[MAXN];
LL sum[MAXN];
int main() {
int n, k; LL ans;
scanf("%d", &n);
for(int i = 1; i <= n; ++i) scanf("%lld", &a[i]);
for(int i = 1; i <= n; ++i) {
scanf("%lld", &g[i]);
sum[i] = sum[i - 1] + g[i];
}
for(int i = 1; i <= n; ++i) {
dp[i] = dp[i - 1];
if(!mp.count(a[i])) {
mp[a[i]] = i - 1;
continue;
}
int id = mp[a[i]];
dp[i] = max(dp[i], dp[id] + sum[i] - sum[id]);
if(sum[i - 1] - dp[i - 1] < sum[id] - dp[id]) mp[a[i]] = i - 1;
}
printf("%lld\n", dp[n]);
return 0;
}
数据比较水,DFS+减枝:
#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long LL;
map<LL, int> mp;
const int MAXN = 3e5 + 10;
LL dp[MAXN], a[MAXN], g[MAXN], pre[MAXN];
LL sum[MAXN];
int main() {
int n, k; LL ans;
scanf("%d", &n);
for(int i = 1; i <= n; ++i) {
scanf("%lld", &a[i]);
pre[i] = mp[a[i]];
mp[a[i]] = i;
}
for(int i = 1; i <= n; ++i) {
scanf("%lld", &g[i]);
sum[i] = sum[i - 1] + g[i];
}
for(int i = 1; i <= n; ++i) {
k = i;
dp[i] = dp[i - 1];
while(pre[k]) {
ans = sum[i] - sum[pre[k] - 1];
dp[i] = max(dp[i], dp[pre[k] - 1] + ans);
if(dp[i] == sum[i]) break;
k = pre[k];
}
}
printf("%lld\n", dp[n]);
return 0;
}