POJ - 2566 Bound Found（尺取法+前缀和）

 1 #include <cstdio>
 2 #include <algorithm>
 3 using namespace std;
 4 
 5 const int INF=0x3f3f3f3f;
 6 const int N=100011;
 7 pair <int,int> T[N];
 8 
 9 //前缀和初始化
10 void init(int n){
11     T[0]=make_pair(0,0);//这里需要放一个0,0进去,不然有一些值就取不到了. 
12     int x,tmp=0;
13     for(int i=1;i<=n;i++){
14         scanf("%d",&x);
15         tmp+=x;
16         T[i]=make_pair(tmp,i);
17     }
18     sort(T,T+n+1);
19 }
20 
21 int main(){
22     int n,m;
23     while(scanf("%d %d",&n,&m)){
24         if(n==0&&m==0) break;
25         init(n);
26         for(int i=1;i<=m;i++){
27             int l=0,r=1,t=INF;
28             int tmp,ansl,ansr,ans;
29             scanf("%d",&tmp);
30             while(r<=n&&t){
31                 int temp=T[r].first-T[l].first;
32                 if(abs(temp-tmp)<t){
33                     t=abs(temp-tmp);
34                     ans=temp;
35                     ansl=T[l].second;
36                     ansr=T[r].second;
37                 }
38                 if(temp<tmp) r++;
39                 if(temp>tmp) l++;
40                 if(temp==tmp) break;
41                 if(l==r) r++; //保证r>=l 
42             }
43             if(ansl>ansr) swap(ansl,ansr);//因为是经过排序过的,有可能左边的比右边的大. 
44             printf("%d %d %d\n",ans,ansl+1,ansr);//左边多加了一个0,需要+1 
45         }
46     }
47     return 0;
48 }