Description
从数列中找出连续序列,使得和的绝对值与目标数之差最小
Input
多组用例,每组用例第一行两个整数n和m分别表示数列长度和查询次数,第二行为n个整数表示数列,第三行为m个整数表示目标数,以n=k=0结束输入
Output
对于每次查询,输出三个整数sum,l,r,分别表示其绝对值与目标数之差最小的连续序列值与此连续序列的左右端点
Sample Input
5 1
-10 -5 0 5 10
3
10 2
-9 8 -7 6 -5 4 -3 2 -1 0
5 11
15 2
-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1
15 100
0 0
Sample Output
5 2 2
5 2 8
9 1 1
15 1 15
15 1 15
Solution
因为前缀和不单调,所以需要先排个序,之后就是尺取法了:首尾分别逐步向前挪动,挪动过程中记录答案
Code
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 1000010
#define INF 100000000
int n,m;
pair<int,int> P[maxn];
void solve(int x)
{
int l=0,r=1,Min=INF;
int ansx,ansl,ansr;
while(l<=n&&r<=n)
{
int y=P[r].first-P[l].first;
if(abs(y-x)<Min)//更新差值的最小值
{
Min=abs(y-x);
ansx=y;
ansl=P[l].second;
ansr=P[r].second;
}
if(y<x) r++;//小于目标值右标前进
else if(y>x) l++;//大于目标值左标前进
else break;//等于目标值退出循环
if(l==r)//左右标相同则右标前进
r++;
}
if(ansl>ansr)//左标必须小于等于右标
swap(ansl,ansr);
printf("%d %d %d\n",ansx,ansl+1,ansr);
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==0&&m==0)
break;
P[0]=pair<int,int>(0,0);
int d,sum=0;
for(int i=1;i<=n;++i)
{
scanf("%d",&d);
sum+=d;//前缀和
P[i]=pair<int,int>(sum,i);
}
sort(P,P+n+1);//对前缀和排序
while(m--)//m次查询
{
int x;
scanf("%d",&x);
solve(x);
}
}
return 0;
}