hdu5009 Paint Pearls 题解

备注:这道题完全是qbxt老师+dalao博客做法,但代码完全手敲!!!(还调了半个小时才调出来orz)

【题目描述】(原题是英文)

给你一个数组,每个值代表一种颜色,每次选一个区间涂颜色,代价是区间内颜色种类数的平方,涂完所有数组,问你最小代价是多少。 n<=50000

10

3 4 2 4 4 2 4 3 2 2

ans=7

【冷静分析】

如果n的范围小点,可以考虑O(n^2)算法,比较容易想到:

dp[i]表示到i为止,消耗的最小值

求dp[i],只需令j从1到i-1枚举,dp[i]=dp[j]+num^2

*num表示该区间内的颜色数

考虑到j从i-1倒序枚举到1时,若出现重复的a[i],除第一次出现外应该删去,因为枚举到的a[i]加入num^2部分不会使num变大。(优化1)

还有一个小优化,答案一定不会大于n,因为每个数字单独一组的答案是 n*(1)^2=n(优化2)

可以用双向链表来实现

扫到相同的a[i]时,令其退出链表,使i及之前的链表中,每个颜色只出现一次。(性质1)

标程

​
#include<bits/stdc++.h>
using namespace std;
int a[50005],dp[50005],pre[50005],nxt[50005],n;
map<int, int>m;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        pre[i]=i-1;nxt[i]=i+1;
        dp[i]=9999999;//初始化 
    }
    dp[0]=0,pre[0]=-1;//不涂的消耗为0 
    for(int i=1;i<=n;i++){
        if(!m.count(a[i]))  m[a[i]]=i;//a[i]没有在之前的链表里出现过 
        else{//之前已经有过了,踢掉之前的 
            int tmp=m[a[i]];//优化1
            // 踢出链表 
            nxt[pre[tmp]]=nxt[tmp];
            pre[nxt[tmp]]=pre[tmp];
            m[a[i]]=i;//换成现在的,保证在链表里出现的a[i]最靠右 
        }
        int cnt=0;//记录颜色个数。性质1 
        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;//优化2 
        }
    }
    cout<<dp[n]<<endl;
    return 0;
}

​

猜你喜欢

转载自www.cnblogs.com/erutsiom/p/9904853.html
今日推荐