TOJ 6121: 学长的情书 ( 二分）

传送门：

点我

6121: 学长的情书 

时间限制(普通/Java):2000MS/6000MS     内存限制:65536KByte
总提交: 79            测试通过:2

描述

 

抹布收到了一封情书，但是这封情书上只有一串数字，发信人多半是将想要表达的意思隐藏在这串数字里，但是直男抹布怎么看得出来呢。抹布对着这串数字百思不得其解，十分苦恼，当然了，一味的苦恼不是抹布的性格，他决定苦中作乐，他想知道，这串数字中有多少组"优美子序列"。他定义，如果某个连续 的子序列中恰好包含k个偶数数字，就称这个序列为"优美子序列"。

 

现在你看到抹布竟然对着这样一封有着特殊含义的情书玩起了毫不相干的游戏，你看不下去了，你决定...帮助他解出这个问题。

输入

 

输入数据多组，以EOF结束。

每组数据第一行，包含两个正整数n，k，表示n个数字，k的含义与描述中相同。（1<=n<=10^5，0<=k<=5*10^4）

接下来一行，包含n个正整数m，每个数字之间用一个空格隔开。（1<=m<=10^5）

数据组数小于110组。

输出

对于每组输入，输出优美子序列的组数，占一行。

样例输入

5 3
2 2 1 2 2

样例输出

 2

提示

【样例说明】恰好包含3个偶数的连续子序列只有2个，分别是[2,2,1,2]，[2,1,2,2]。

【数据范围】

对于45%的数据，1<=n<=10^3，0<=k<=10^3。

对于90%的数据，1<=n<=10^4，0<=k<=10^4。

对于100%的数据，1<=n<=10^5，0<=k<=5*10^4。

吐槽：

看到2KMS，想了个nlogn的二分去大力莽了。。

思路：

老年人做法
预处理偶数的前缀和。
比如说
样例 2 2 1 2 2
前缀和sum数组是1 2 2 3 4
对每个位置pos，对前缀和数组的操作是对pos之后的位置进行二分查找到sum[pos-1]+k，第一个等于这个数位置posLeft和最后一个等于这个数的位置posRight
差值就是当前位置对答案的贡献：ans += posRight-posLeft+1

trick：

1、当K等于0时候，如果二分写的不好要过滤一下（二分写的好的话可能不用？存疑）

2、ans要用long long

复杂度O(nlogn)

应该有双指针的贡献算法更快（待补）

代码：

#include"bits/stdc++.h"
using namespace std;
typedef long long LL;
int a[100010],k,n;
int sum[100010];
int SolveLeft(int begin){
    int l = begin,r = n;
    int key = sum[begin-1] + k;
    int ans = 0;
    while(l <= r){
        int mid = (l+r) >> 1;
        if(sum[mid] >= key){
            r = mid-1;
            ans = mid;
        }else{
            l = mid+1;
        }
    }
    return ans;
}
int SolveRight(int begin){
    int l = begin,r = n;
    int key = sum[begin-1] + k;
    int ans = 0;
    while(l <= r){
        int mid = (l+r) >> 1;
        if(sum[mid] > key){
            r = mid-1;
        }else{
            l = mid+1;
            ans = mid;
        }
    }
    return ans;
}

int main(){
    while(~scanf("%d%d",&n,&k)){
        LL ans = 0;
        sum[0] = 0;
        for(int i = 1 ; i <= n ; i ++){
            scanf("%d",a+i);
            a[i] = (a[i]&1) ? 0:1;
            sum[i] = 0;
        }
        for(int i = 1 ; i <= n ; i ++){
            sum[i] = sum[i-1]+a[i];
        }
        for(int i = 1 ; i <= n ; i ++){
            int L = SolveLeft(i);
            int R = SolveRight(i);
            if(R < i) continue; //当K等于0的时候会过滤掉误加 
            if(L != 0){
                ans += (R-L+1L);
            }
        }
        cout<<ans<<endl;
    }
}
/*
5 3
2 2 1 2 2

6 3
2 2 1 2 1 2

5 0
1 0 1 0 1

5 1
2 2 1 2 2

10 0
1 1 0 1 0 1 1 1 0 0

2 1
1 0

2 0 
1 0
*/