N头奶牛排成了一列。每头奶牛向前或者向后。为了让所有的奶牛都面向前方,农夫买了一台自动转向机,机器在购买的时候必须设定
一个数值K,机器没操作一次恰好使连续的K头奶牛转向,求使得所有奶牛都转向前的最小的操作次数M和对应的K。
首先考虑以下问题:
1.各个区间的操作顺序对结果无关,所以可以从左到右来考虑;
2.每个区间最多进行一次操作,超过一次的操作完全没有意义;
3.对于每个区间的最左端的牛,由于从左到右来考虑区间,所以当前只有一个区间包含它(处在中间位置时的情况前面的区间已经包含进去了),
如果朝前则次区间不需反转,如果朝后则需要反转。
有了这几条我们就可以进行从左到右的动态规划,考虑到一头牛反转奇数次反向,反转偶数次不变,所以只要求牛被反转的次数便可。
考虑优化:
所以可以将复杂度减为O(n^2)
题源来自《挑战程序竞赛》第二版152页。
//
// 150_face the right way.cpp
// changlle
//
// Created by user on 1/9/16.
// Copyright (c) 2016 user. All rights reserved.
//
#include <iostream>
using namespace std;
int N=7;
int dir[7]={1,1,0,1,0,1,1};//牛的方向 0为F, 1为B
int f[7]; //是否反转第i头牛
//固定k, 求对应的最小操作数
//无解返回-1
int calc (int K) {
memset (f,0,sizeof(f));
int res=0;
int sum=0;
for (int i=0;i+K<=N;i++) {
//计算区间[i,i+K-1]
if( (dir[i]+sum)%2!=0) {
//经过操作后现在牛朝后方
res++;
f[i]=1;
}
sum+=f[i];
if(i-K+1>=0) {
sum-=f[i-K+1];
}
}
//检查尾部的牛是否符合要求。
for (int i=N-K+1;i<N;i++) {
if ((dir[i]+sum)%2!=0) {
return -1;
}
if (i-K+1>=0) {
sum-=f[i-K+1];
}
}
return res;
}
int main () {
int K=1, M=N;
for (int k=1;k<=N;k++) {
int m=calc(k);
if(m>=0 && M>m) {
M=m;
K=k;
}
}
cout<<"K="<<K<<endl<<"M="<<M<<endl;
}