原题链接:
https://vjudge.net/problem/HDU-5009
AC代码:
#include <algorithm>
#include <bits/stdc++.h>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
using namespace std;
const int INF = 0x3f3f3f3f;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e5 + 10;
void __init__() {
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
} //head
int a[maxn], pre[maxn], suf[maxn], dp[maxn];
map<int, int> mp;
int main() {
// __init__();
int n;
while (scanf("%d", &n) != EOF) {
mp.clear();
memset(dp, INF, sizeof(dp));
for (int i = 1; i <= n; i++) {
cin >> a[i];
pre[i] = i - 1;
suf[i] = i + 1;
}
pre[0] = -1;
dp[0] = 0;
for (int i = 1; i <= n; i++) {
if (!mp.count(a[i])) {
mp[a[i]] = i;
} else {
int id = mp[a[i]];
suf[pre[id]] = suf[id];
pre[suf[id]] = pre[id];
mp[a[i]] = i;
}
int cnt = 0;
for (int j = pre[i]; j != -1; j = pre[j]) {
cnt++;
dp[i] = min(dp[i], dp[j] + cnt * cnt);
if (cnt * cnt > i) {
break;
}
}
}
cout << dp[n] << endl;
}
return 0;
}
日常补题,参考了网上大佬的代码
讲道理这题挺难的,难在对于时间复杂度的优化,这道题的意思就是我们推导出动态规划的状态转移方程:
dp[i]=min{dp[i],dp[j]+num(j+1,i)^2}
但是单纯的n^2复杂度会超时,需要优化,怎么优化呢,我们发现数列的有些元素可能是重复的,对于本题来说重复的元素是没有意义的,于是我们就模拟双向链表的操作方式,将数列的重复元素’删除‘(不是真正意义的删除数组元素,而是用两个模拟指针表示每个元素的前后相接元素),这样可以很大的减少不必要的时间损耗,同时我们又发现,总共的耗费不会超过n(每一个珍珠是一个区间的话,就是1+1+…+1=n)因此我们计算cnt的时候,如果cnt*cnt>i的话,就进行剪枝操作,直接break,因为不可能是最优结果了,最终dp[n]即为最优解