PAT甲级1044 Shopping in Mars (25 分)（前缀和，双指针）

Posted 2020-11-15 ldudxy

题意：

输入一个正整数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<<" "<<ans[i].first<<"-"<<ans[i].second;
return 0;
}