Problem A:队伍(team)

Problem A:队伍(team)

Time Limit:1000MS Memory Limit:65536K
Total Submit:46 Accepted:18

Description

问题描述：
蛋糕分好了，小朋友排着队去领蛋糕。铭铭想从 N 人的队伍中选出 K 位小朋友帮忙分发蛋糕。但铭铭选人 的方法有点特别，他想从队伍中选连续的 K 个小朋友， 而且必须是男比女多，你知道铭铭有多少种选择吗？
输入格式：
第一行， 两个整数 N， 代表队伍中有 N [N:0~1000000] 个小朋友，铭铭想选 K[K小于N]个人。 第二行：有 N 个 0 或 1[0 代表男，1 代表女]，每个数用空格隔开。
输出格式：
输出一个整数。代表铭铭可以有多少种选择方案。
输入样例：
10
3
0 1 1 0 0 1 0 1 0 1
输出样例：
4

#include<cstdio>using namespace std;int a[1000010];int main(){int n,k,s=0,l=1,h,nan=0,nv=0;scanf("%d%d",&n,&k);h=k;for(int i=1;i<=n;i++) scanf("%d",&a[i]);for(int i=1;i<=k;i++) {if(a[i]==1) nv++;else nan++;}for(int i=1;i<=n-k+1;i++){if(nan>nv) s++;h++;if(a[h]==1) nv++;else nan++;if(a[l]==1) nv--;else nan--;l++;} printf("%d",s);return 0;}