bzoj2933 [Poi1999]地图(贪心+dp)

首先我们贪心的把所有数从小到大排序,那么最后每种颜色一定是一段连续的区间。
然后问题就变成了把这个序列分成m段,使得每段的这个和的和最小。
dp即可。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 3010
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,a[N];ll f[N][11],s[N];
inline ll cal(int i,int j){
    int mid=i+j>>1;
    return s[j]-s[mid]-(((j-i+1)&1)?s[mid-1]-s[i-1]:s[mid]-s[i-1]);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=n;++i) a[i]=read();sort(a+1,a+n+1);
    for(int i=1;i<=n;++i) s[i]=s[i-1]+a[i];memset(f,inf,sizeof(f));
    for(int i=1;i<=n;++i) f[i][1]=cal(1,i);
    for(int i=2;i<=n;++i)
        for(int j=2;j<=m;++j)
            for(int k=j-1;k<i;++k)
                f[i][j]=min(f[i][j],f[k][j-1]+cal(k+1,i));
    ll ans=1LL<<60;
    for(int i=1;i<=m;++i) ans=min(ans,f[n][i]);
    printf("%lld\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/icefox_zhx/article/details/80537957