src2020/1875_houseschoose/housechoose.cpp
2020-03-20 17:40:39 +08:00

32 lines
711 B
C++

#include <iostream>
using namespace std;
int main()
{
int n, k, a[501] = { 0 }, num[501] = { 0 }, sum = 0,f[501]={0},t=0,w=1,ans,now[501]={0};
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> a[i];
if (a[i] == 1) {
num[i] = ++sum;
}
}
now[1]=1;
do{
t++;
for(int i=now[t]+1;i<n;i++){
f[++w]=f[t]+(num[i]-num[now[t]-1])*num[i];
now[w]=i;
}
int bi=f[t]+(num[n]-num[now[t]-1])*num[n];
if(bi<sum){
sum=bi;
}
}while(t<w);
cout<<sum;
/*
for(int i=1;i<=n;i++){
for(int j=1;j<=k;j++){
}
}*/
}