题186.pat甲级练习-1044 Shopping in Mars (25 分)
一、题目
二、题解
依题目大意,说白了就是要你输出连续子列和等于M的所有区间,要是没有等于M的就输出大于M中最小的(比M大但最接近M的)。其实上述可看作一个问题,即求连续子列和最小值问题,这个最小值需要>=M且越接近M越好。
因为是对于本题且为连续子列和问题,所以我们可以单纯用一个普通的队列queue来做滑动窗口,用滑窗来解决上述问题就好。
基本思路就是queue存储序列元素对应的下标,用一个变量记住当前队列所表示的连续子列的和,一个变量记着所谓的最小子列和,正着遍历序列,当当前子列和小于M时就访问下一个序列元素,并让其入队,即扩展窗口(还没到M,所以扩展);当当前子列和大于或等于M时,就把子列和与所谓的最小子列和比较,如果相等,那么添加区间结果,如果更小且不小于M,那么重置区间结果再添加并更新所谓的最小子列和。之后,将队头删除,当前子列和减少,即缩小窗口(已经到M甚至比M大了,要想扩展,得先缩小)。待序列元素遍历完成且队列为空时,过程结束,输出区间结果就好。
#include <bits/stdc++.h>
using namespace std;
const int Inf=0x3f3f3f3f;
int value[100001];
queue<int> q;
int min_valuesum=Inf;
vector<pair<int,int>> res;
int main()
{
int N,M;
cin>>N>>M;
for(int i=1;i<=N;i++)
{
scanf("%d",&value[i]);
}
int valuesum=0;
for(int i=1;i<=N||!q.empty();)//切记要等队列空了才退出循环
{
if(valuesum<M&&i<=N)//切记限制i<=N才能执行这个操作
{
q.push(i);
valuesum+=value[i];
i++;
}
else
{
if(valuesum<min_valuesum&&valuesum>=M)//切记限制vlauesum>=M才执行这些操作,否则会更新成小于M的值,就完犊子了
{
min_valuesum=valuesum;
res.clear();
res.push_back(make_pair(q.front(),i-1));
}
else if(valuesum==min_valuesum&&valuesum>=M)
{
res.push_back(make_pair(q.front(),i-1));
}
valuesum-=value[q.front()];
q.pop();
}
}
for(int i=0;i<res.size();i++)
{
if(i>0)
{
putchar('\n');
}
printf("%d-%d",res[i].first,res[i].second);
}
}