codeforces croc 2016 C. Enduring Exodus
题目链接 题意：给出n和k,给出一个长度为n的字符串表示房间的占用情况（0表示没占用，1表示已占用），从n个房间中找出k+1个，使得k+1中的k个距离k+1个中的1个的距离和最小。
思路：只需要考虑没被占用的位置。所以用pos[]数组记录0的位置。 找到第一个能住下的位置后向前平移即可。
/* ***********************************************
Author :111qqz
Created Time :2016年03月19日 星期六 00时21分28秒
File Name :code/cf/croc2016/C.cpp
************************************************ */
1#include <cstdio>
2#include <cstring>
3#include <iostream>
4#include <algorithm>
5#include <vector>
6#include <queue>
7#include <set>
8#include <map>
9#include <string>
10#include <cmath>
11#include <cstdlib>
12#include <ctime>
13#define fst first
14#define sec second
15#define lson l,m,rt<<1
16#define rson m+1,r,rt<<1|1
17#define ms(a,x) memset(a,x,sizeof(a))
18typedef long long LL;
19#define pi pair < int ,int >
20#define MP make_pair
1using namespace std;
2const double eps = 1E-8;
3const int dx4[4]={1,0,0,-1};
4const int dy4[4]={0,-1,1,0};
5const int inf = 0x3f3f3f3f;
6const int N=1E5+7;
7int ban[N];
8int k,n;
9int sum[N];
10int pos0[N];
11string st;
1int dis(int x,int l,int r)
2{
3 return max(pos0[x]-pos0[l],pos0[r]-pos0[x]);
4}
5int main()
6{
7 #ifndef ONLINE_JUDGE
8 freopen("code/in.txt","r",stdin);
9 #endif
10 cin>>n>>k;
11 cin>>st;
12 ms(ban,0);
13 ms(pos0,0);
14 k++;
15 int len = st.length();
16 int cnt0 = 0;
17 for ( int i = 0 ; i < len ; i++)
18 {
19 if (st[i]=='0')
20 {
21 ban[i+1] = 0;
22 cnt0++;
23 pos0[cnt0] = i ;
24 }
25 else
26 {
27 ban[i+1] = 1;
28 }
29 }
sum[0] = 0 ;
1 for ( int i = 1 ; i <= n ; i++)
2 {
3 sum[i] = sum[i-1] + ban[i];
4 }
5 int ans = inf;
6 int l = 1 ; int r = k;
7 int mid = 0;
8 for ( int i = 1 ; i <= r ; i++)
9 {
10 if (dis(i,l,r)<dis(mid,l,r))
11 {
12 mid = i;
13 }
14 }
1 ans = dis(mid,l,r);
2 for ( ; r <= cnt0 ; r++,l++)
3 {
4 while (dis(mid+1,l,r)<dis(mid,l,r)) mid++;
5 ans = min(ans,dis(mid,l,r));
6 }
7 cout<<ans<<endl;
1 #ifndef ONLINE_JUDGE
2 fclose(stdin);
3 #endif
4 return 0;
5}