队 伍(NHOI2016xy6)

队 伍(NHOI2016xy6)

描述

蛋糕分好了，小朋友排着队去领蛋糕。铭铭想从 N人的队伍中选出 K 位小朋友帮忙分发蛋糕。但铭铭选人的方法有点特别，他想从队伍中选连续的 K 个小朋友，而且必须是男比女多，你知道铭铭有多少种选择吗？

输入

第一行，两个整数 N，代表队伍中有 N （0<N<=1000000）个小朋友，铭铭想选 K（K<N）个人。

第二行：有 N 个 0 或 1（0 代表男，1 代表女），每个数用空格隔开。

数据范围：

50%的数据 0<N<1000,k<N;

80%的数据 0<N<1000000，k<=100;

100%的数据 0<N<1000000,k<N;

输出

输出一个整数。代表铭铭可以有多少种选择方案。

输入样例 1 

10 3
0 1 1 0 1 0 0 1 0 1

输出样例 1

4

输入样例 2

7 3
0 1 0 1 0 0 1

输出样例 2

2

输入样例 3

7 3
0 1 1 0 0 1 0

输出样例 3

2

输入样例 4

5 3
0 0 1 0 1

输出样例 4

1

输入样例 5

8 3
1 1 0 1 0 0 1 0

输出样例 5

3

代码如下：

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int n, k;
    cin >> n >> k;
    vector<int> nums(n);
    for (int i = 0; i < n; i++) {
        cin >> nums[i];
    }
    int ans = 0;
    int men = 0, women = 0;
    for (int i = 0; i < k; i++) {
        if (nums[i] == 0) men++;
        else women++;
    }
    if (men > women) ans++;
    for (int i = k; i < n; i++) {
        if (nums[i-k] == 0) men--;
        else women--;
        if (nums[i] == 0) men++;
        else women++;
        if (men > women) ans++;
    }
    cout << ans << endl;
    return 0;
}

 思路：

思路解析：

本题依旧是一道经典的滑动窗口问题，连续的 K 个小朋友可以抽象为一个大小为 K 的窗口，我们依次将窗口从队伍的第一个人移动到最后一个人，每个窗口的男女人数就可以通过统计来得到。接着只需判断男人数是否大于女人数，若是则答案加 1。

代码中的变量 men 和 women 分别表示当前窗口中男人和女人的数量，ans 表示符合条件的窗口数量。具体实现中，我们先计算前 K 个人的男女数量，然后循环遍历队列，每次移动窗口时，先将窗口最左边的人的性别从相应的人数中减掉，再将窗口最右边的人的性别从相应的人数中加上，就得到当前窗口中男女的数量。最后，只需判断男人数是否大于女人数即可。