XTU 1414

科普伊特瑞

已于 2023-05-28 16:18:26 修改

阅读量54

点赞数

文章标签： c++ 算法

于 2023-04-09 15:17:55 首次发布

本文链接：https://blog.csdn.net/weixin_51552375/article/details/130042597

版权

该程序段是用C++实现的一个算法，主要运用了前缀和和双指针技巧。它读取输入的整数n和k，以及一串数字ch，计算每个sh[i]对应在ch数组中满足条件的子数组个数。通过对sum数组的更新，双指针l和r动态调整，找到和等于sh[i]的子数组个数并输出结果。

摘要由CSDN通过智能技术生成

前缀和+同向双指针

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

int ch[10001];
int sh[101];
int sum[10001];
int t;
int n,k;

int main(){
    cin>>t;
    while(t--){
        cin>>n>>k;
        for(int i=1;i<=n;i++){
            cin>>ch[i];
            sum[i]=sum[i-1]+ch[i];
            //cout<<"sum:"<<i<<" "<<sum[i]<<endl;
        }
        for(int i=1;i<=k;i++){
            cin>>sh[i];
            int l=0,r;
            int ans=0;
            for(r=0;r<=n;r++){
            	while((sum[r]-sum[l])>sh[i]){
            		l++;
				}
				if((sum[r]-sum[l])==sh[i]){
					ans++;
				}
			}
            cout<<ans<<endl;
        }
        
    }
}