POJ 2566 尺取法

题意

传送门 POJ 2566

题解

由于数列元素不单调，难以使用依次推进元素的尺取法。

考虑前缀和，升序排序后，满足 $S_i-S_j\ge 0,j<i$，代表所有的区间求和的绝对值。满足单调性，就可以使用尺取法处理区间了。

对于每一个查询值 $t$，只要用小于它的最大值以及大于等于它的最小值更新答案即可。但查询值会出现 $0$，而对于非法区间 $(i,i]$ 求和值也为 $0$，方便起见，每一次尺取合法区间就更新答案，非法区间则返回一个不可能出现的求和值。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define min(a,b)    (((a) < (b)) ? (a) : (b))
#define max(a,b)    (((a) > (b)) ? (a) : (b))
#define abs(x)    ((x) < 0 ? -(x) : (x))
#define INF 0x3f3f3f3f
#define delta 0.85
#define eps 1e-5
#define PI 3.14159265358979323846
#define MAX_N 100005
using namespace std;
typedef pair<int, int> P;
int N, K, rs, rt, res;
P S[MAX_N];

int update(int n, int s, int t){
	if(s >= t) return -INF;
	int sum = S[t].first - S[s].first;
	// 对于每一个搜索到的合法区间都更新答案
	if(abs(sum - n) < abs(res - n)){
		res = sum, rs = S[s].second, rt = S[t].second;
	}
	return sum;
}

int main(){
	while(~scanf("%d%d", &N, &K) && (N | K)){
		int s = 0;
		S[0] = P(s, 0);
		for(int i = 1; i <= N; i++){
			int v;
			scanf("%d", &v);
			s += v;
			S[i] = P(s, i);
		}
		sort(S, S + N + 1);
		while(K--){
			int n, s = 0, t = 0, sum = -INF;
			scanf("%d", &n);
			// 尺取法
			res = -INF;
			for(;;){
				while(t < N && sum < n){
					sum = update(n, s, ++t);
				}
				if(sum < n) break;
				sum = update(n, ++s, t);
			}
			printf("%d %d %d\n", res, min(rs, rt) + 1, max(rs, rt));
		}
	}
	return 0;
}