题目链接:戳我
题目大意:
有这么一行奶牛,每个奶牛有初始状态。
给出一个字符串,只包含大写字母B和F,B表示初始状态向后,F表示向前,现在有一个机器一次可以反转K个奶牛的朝向,求出最小的操作次数以及此时对应的K值;
看完题目容易想出N^3暴力算法,枚举每个k对于每个k,对原序列从1跑到n-k+1然后如果当前牛朝前那就不翻转,否则反转,然后反转就暴力反转。。。。这样复杂度比较高,我们着手优化暴力反转:
首先,一个区间翻转两次对该位置没有意义。
对于一个牛来说,如果他开始朝前并且被翻转了奇数次,那么它就一定得反转,否则不用反转,如果开始朝后并且翻转了偶数次也需要反转,这样的话,用一个新的f数组记录当前位置是否被翻转过,翻转过记为1,然后同时记录一下Sum,表示对当前奶牛有影响的f的和,翻转操作就可以O(1)了 (233很迷对吧)
我能说一直wr是因为输出反了?2333333333333333333333
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
const int size = 5010;
const int INF = 2 << 28;
int niu[size];
int f[size];
int main()
{
int n;
scanf("%d",&n);
for(int i = 1;i <= n;i ++)
{
char c;
cin>>c;
if(c == 'F') continue;
if(c == 'B') niu[i] = 1;
}
int sum,res;
int ansm = INF,ansk = 0;
for(int k = 1;k <= n;k ++)
{
memset(f,0,sizeof(f));
sum = 0;
res = 0;
int i;
for(i = 1;i <= n - k + 1;i ++)
{
if((niu[i] + sum) % 2 != 0)
{
res ++;
f[i] = 1;
}
sum += f[i];
if(i-k+1 >= 1) sum -= f[i-k+1];
}
bool flag = 0;
for(i;i <= n;i ++)
{
if((niu[i] + sum) % 2 != 0)
{
flag = 1;
break;
}
if(i-k+1 >= 1) sum -= f[i-k+1];
}
if(flag == 1) continue;
else if(res < ansm && res >= 0)
{
ansm = res;
ansk = k;
}
}
printf("%d %d",ansm,ansk);
return 0;
}
这里是一只大蒟蒻Seavot