Codeforces 1420 E. Battle Lemmings —— 暴力DP,想法

202 篇文章 5 订阅

This way

题意:

给你n个数,如果有一对0之间有至少一个1,那么这对0记录答案,问你可以将1左右移动的话,移动最多i步的时候,答案是多少。

题解:

首先可以确定的就是将这个数组按照1为分隔划分成多段,然后每段的贡献就是(段内0数量)*(段外0数量),加起来除二就行了。
设段的数量是m,b[i]表示第i段的0数量
那么答案其实就是 ( ∑ i = 1 m ( b [ i ] ∑ j = 1 m b [ i ] ) − ∑ i = 1 m ( b [ i ] ∗ b [ i ] ) ) / 2 (\sum\limits_{i=1}^{m}(b[i]\sum\limits_{j=1}^{m}b[i])-\sum\limits_{i=1}^{m}(b[i]*b[i]))/2 (i=1m(b[i]j=1mb[i])i=1m(b[i]b[i]))/2
首先前半段其实就相当于所有0的数量的平方,那么我们DP只需要求出后半段的最小值即可。
dp[i][j][k]表示到了第i个1,前面有j个0,走了k步的时候的 ∑ x = 1 i ( b [ x ] ∗ b [ x ] ) \sum\limits_{x=1}^{i}(b[x]*b[x]) x=1i(b[x]b[x])最小值
转移就是暴力枚举下一个位置的0的数量

#include<bits/stdc++.h>
using namespace std;
const int N=85;
int dp[N][N][N*N],a[N],s[N];
int main()
{
    int n,m=1;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        if(a[i])m++;
        else s[m]++;
    }
    memset(dp,-1,sizeof(dp));
    dp[0][0][0]=0;
    for(int i=1;i<=m;i++)s[i]+=s[i-1];
    for(int i=0;i<m;i++){
        for(int j=0;j<=s[m];j++){
            for(int k=0;k<=n*(n-1)/2;k++){
                if(dp[i][j][k]==-1)continue;
                for(int l=j;l<=s[m];l++){
                    int step=k+abs(l-s[i+1]);
                    if(step>n*(n-1)/2)continue;
                    if(dp[i+1][l][step]==-1)dp[i+1][l][step]=dp[i][j][k]+(l-j)*(l-j);
                    else dp[i+1][l][step]=min(dp[i+1][l][step],dp[i][j][k]+(l-j)*(l-j));
                }
            }
        }
    }
    int mi=1e9;
    for(int i=0;i<=n*(n-1)/2;i++){
        if(dp[m][s[m]][i]!=-1)
            mi=min(mi,dp[m][s[m]][i]);
        printf("%d%c",(s[m]*s[m]-mi)/2," \n"[i==n*(n-1)/2]);
    }
    return 0;
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值