bzoj 2933(区间dp)

来源:互联网 发布:电视直播软件pc版 编辑:程序博客网 时间:2024/05/29 08:02

传送门
题解:先按照a排序,预处理前缀和,然后预处理区间代价(每次贪心地选中位数然后计算),最后dp一次即可。dp[i][j]表示前i个地区分成j个颜色的最小代价。
吐槽一下,有位大哥做bzoj3065一道时限60秒的题TLE了8次算不算卡评测啊(╯‵□′)╯︵┻━┻

#include<bits/stdc++.h>using namespace std;typedef long long ll; int n,m;ll dp[3002][12],w[3002][3002],a[3002],sum[3002]={0};int main() {//  freopen("bzoj 2933.in","r",stdin);    scanf("%d%d",&n,&m);    for (int i=1;i<=n;++i) scanf("%lld",&a[i]);    sort(a+1,a+n+1);    for (int i=1;i<=n;++i) sum[i]=sum[i-1]+a[i];    for (int i=1;i<=n;++i)        for (int j=i;j<=n;++j) {            int mid=(i+j)>>1;            w[i][j]=a[mid]*(mid-i)-(sum[mid-1]-sum[i-1])-a[mid]*(j-mid)+(sum[j]-sum[mid]);        }    memset(dp,127,sizeof(dp));    dp[0][0]=0;    for (int i=1;i<=n;++i)        for (int j=1;j<=m;++j)            for (int k=0;k<i;++k)                dp[i][j]=min(dp[i][j],dp[k][j-1]+w[k+1][i]);    printf("%lld\n",dp[n][m]);    return 0;}
原创粉丝点击