当前位置 : 主页 > 编程语言 > c++ >

【PAT甲级】1044 Shopping in Mars (25 分)(前缀和,双指针)

来源:互联网 收集:自由互联 发布时间:2021-06-23
题意: 输入一个正整数N和M(N=1e5,M=1e8),接下来输入N个正整数(=1e3),按照升序输出"i-j",i~j的和等于M或者是最小的大于M的数段。 代码: #define HAVE_STRUCT_TIMESPEC #includebits/stdc++.h us

题意:

输入一个正整数N和M(N<=1e5,M<=1e8),接下来输入N个正整数(<=1e3),按照升序输出"i-j",i~j的和等于M或者是最小的大于M的数段。

代码:

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
int a[100007];
int sum[100007];
vector<pair<int,int> >ans;
int main(){
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
int n,m;
cin>>n>>m;
for(int i=1;i<=n;++i){
cin>>a[i];
sum[i]=sum[i-1]+a[i];
}
int l=0,t=0,mn=1e9;
for(int i=0;i<=n;++i){
t-=a[i];
while(t<m&&l<=n)
t+=a[l++];
if(t>=m&&t<mn){
mn=t;
ans.clear();
ans.push_back({i+1,l-1});
}
else if(t==mn)
ans.push_back({i+1,l-1});
}
cout<<ans[0].first<<"-"<<ans[0].second;
for(int i=1;i<ans.size();++i)
cout<<"\n"<<ans[i].first<<"-"<<ans[i].second;
return 0;
}

网友评论