#P3309. 队伍选人

队伍选人

问题描述

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

输入格式

第一行,两个整数N,代表队伍中有(0<N<=100000)个小朋友,铭铭想选K(K<N)个人。 第二行:有N个0或1(0代表男,1代表女),每个数用空格隔开。

输出格式

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

样例输入

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

样例输出

4

样例解释

样例中队伍的小朋友性别序列为0 1 1 0 1 0 0 1 0 1,需要选择连续的3个小朋友,且男(0)的数量多于女(1)的数量。我们依次检查所有连续的3个小朋友的组合:

  1. 第1-3位:0、1、1 → 男1人,女2人 → 不满足
  2. 第2-4位:1、1、0 → 男1人,女2人 → 不满足
  3. 第3-5位:1、0、1 → 男1人,女2人 → 不满足
  4. 第4-6位:0、1、0 → 男2人,女1人 → 满足
  5. 第5-7位:1、0、0 → 男2人,女1人 → 满足
  6. 第6-8位:0、0、1 → 男2人,女1人 → 满足
  7. 第7-9位:0、1、0 → 男2人,女1人 → 满足
  8. 第8-10位:1、0、1 → 男1人,女2人 → 不满足

满足条件的组合共有4种,因此答案为4。