您的位置:首页 > 博客中心 > 电脑问题 >

acwing 273. 分级

时间:2022-03-18 14:52

题面:

给定长度为N的序列A,构造一个长度为N的序列B,满足:

1、B非严格单调,即B1≤B2≤…≤BNB1≤B2≤…≤BN或B1≥B2≥…≥BNB1≥B2≥…≥BN。
2、最小化 S=∑Ni=1|Ai−Bi|S=∑i=1N|Ai−Bi|。

只需要求出这个最小值S。

输入格式

第一行包含一个整数N。

接下来N行,每行包含一个整数AiAi。

输出格式

输出一个整数,表示最小S值。

数据范围

1≤N≤20001≤N≤2000,
1≤|Ai|≤1091≤|Ai|≤109

输入样例:

7
1
3
2
4
5
3
9

输出样例:

3
题解:

(贪心,DP,序列型DP,前缀和) O(n2)O(n2)
对单调上升和单调下降分别求一遍,取最小值即可。

下面只讨论对单调上升的求法。

引理:

一定存在一组最优解,使得每个 BiBi 都是原序列中的某个值。

证明:

假设某个最优解如下图所示,其中 {Ai}{Ai} 是原序列,{A′i}{Ai′} 是将原序列排序后的序列,图中红色圆圈表示每个 BiBi。

技术图片


考虑每个位于 A′i,A′i+1Ai′,Ai+1′之间的一段 BiBi,比如上图中粉色框中的部分。
则我们在 {Ai}{Ai} 中粉色框对应的这段里统计出大于等于 A′i+1Ai+1′ 的数的个数 xx,小于等于 A′iAi′ 的数的个数 yy,那么:

如果 x>yx>y,将粉色框中的 BiBi 整体上移,使最高的一个圆圈达到上边界,结果会变好;
如果 x<yx<y,将粉色框中的 BiBi 整体下移,使最低的一个圆圈达到下边界,结果会变好;
如果 x=yx=y,则上面两种方式均可,结果不会变差;
综上所述,只要存在某个 BiBi 的值不在原序列中,我们一定可以将它调整成原序列中的值,且结果不会变差。

证毕。

剩下的问题就比较简单了。用闫氏DP分析法即可。

状态表示:

f[i][j] 代表所有给A[1] ~ A[i]分配好了值且最后一个B[i] = A‘[j]的方案的集合;
f[i][j] 的值是集合中所有方案的最小值;
状态计算:
依据倒数第二个数分配的是哪个A‘[i]将f[i][j]所代表的集合划分成j个不重不漏的子集:

倒数第二个数选取的是A‘[1]的所有方案的结合,最小值是 f[i - 1][1] + abs(A[i] - A‘[j]);
倒数第二个数选取的是A‘[2]的所有方案的结合,最小值是 f[i - 1][2] + abs(A[i] - A‘[j]);

倒数第二个数选取的是A‘[j]的所有方案的结合,最小值是 f[i - 1][j] + abs(A[i] - A‘[j]);
f[i][j]在所有子集的最小值中取min即可。

最终答案需要遍历最后一个数的所有取值,然后取min即可。

类似于AcWing 272. 最长公共上升子序列的优化方式,可以用前缀和思想优化掉一维循环。

时间复杂度
总共两重循环,时间复杂度是 O(n2)O(n2)。

代码实现:

#include<iostream>
#include<cstring>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
int n;int f[2010][2010];int a[2010],b[2010];
int work()
{
    for(int i=1;i<=n;i++)b[i]=a[i];
    sort(b+1,b+n+1);
    for(int i=1;i<=n;i++)
     {
         int minn=inf;
         for(int j=1;j<=n;j++)
         {
             minn=min(minn,f[i-1][j]);
             f[i][j]=minn+abs(a[i]-b[j]);
         }
     }
     int res=inf;
     for(int i=1;i<=n;i++)res=min(res,f[n][i]);
     return res;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    int res=work();
    reverse(a+1,a+1+n);
    res = min(res, work());
    printf("%d\n",res);
}

 

热门排行

今日推荐

热门手游