HDU-5009 Paint Pearls

原题链接：

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]即为最优解