题意:
给你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=1∑m(b[i]j=1∑mb[i])−i=1∑m(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=1∑i(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;
}