01数组

01数组

题目描述

有一个长度为 nn 且仅由 11 或 00 组成的数组。每次操作允许移除数组第一个或最后一个元素。

请问最少要进行几次这样的操作，才能使得数组中所有元素之和为 ss。

如果无法通过操作达到目标，则输出 −1−1。

输入格式

第一行一个整数 TT ，表示共有 TT 组数据。每组数据格式如下：

第一行两个整数，nn 和 ss。

第二行 nn 个整数 aiai，即为数组中的元素。

输出格式

共 TT 行，每行对应一组数据的答案

样例数据

输入样例 #1

7
3 1
1 0 0
3 1
1 1 0
9 3
0 1 0 1 1 1 0 0 1
6 4
1 1 1 1 1 1
5 1
0 0 1 1 0
16 2
1 1 0 0 1 0 0 1 1 0 0 0 0 0 1 1
6 3
1 0 1 0 0 0

输出样例 #1

0
1
3
2
2
7
-1

数据范围

对于 20%20% 的数据，aiai 皆为 11

对于 100%100% 的数据，1≤t≤1000,1≤n,s≤2×104,0≤ai≤1

#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int t;
    cin>>t;
    while(t--){
        int n, s;
        cin>>n>>s;
        vector<int> arr(n);
        int sumOfArr=0;
        for(int i=0;i<n;i++){
            cin>>arr[i];
            sumOfArr+=arr[i];
        }
        if(sumOfArr<s){
            cout<<-1<<'\n';
            continue;
        }
        int left=0, right=0;
        int currS=0;
        int ans=0;
        while(right<n){
            while(right<n && currS<=s){
                currS+=arr[right];
                right++;
            }
            if(currS==s){
                ans=max(ans, right-left);
            }
            if(currS==s+1){
                ans=max(ans, right-left-1);
            }
            while(left<right && currS>s){
                currS-=arr[left];
                left++;
            } 
        }
        cout<<n-ans<<'\n';
    }
    return 0;
}