原题链接:http://ac.jobdu.com/problem.php?pid=1554
由数列的前缀和:
Sn=∑i=1nai i=1,2,3...n
由于:
Sn−Sn−1=an
所以区间 [i,j] 之和, Sj−Si−1=ai+ai+1+...+aj 1≤i,j,≤n
由题意给定一个数组,判断数组内是否存在一个连续区间,使其和恰好等于给定整 k
其实就是判断 Sum[j]−Sum[i]=k 是否成立。
暴力的方法直接枚举,由于 n=10000 会 tle 。
现在我们换个思路,试试二分,考虑开个结构体数组,保存原数组的前缀和+原来的位置。
然后对其排序,再进行二分。但这样可能会出错,所以再判断一下,即可。
懒得啰嗦了,直接看代码吧。。
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
using std::sort;
using std::lower_bound;
const int Max_N = 10010;
int arr[Max_N], sum[Max_N], temp[Max_N];
struct Node {
int v, pos;
bool operator<(const Node &x) const {
if (v == x.v) return pos < x.pos;
return v < x.v;
}
}rec[Max_N];
void solve(int n, int k) {
int p = 0;
for (int i = 0; i < n; i++) {
p = lower_bound(temp + 1, temp + n, sum[i] + k) - temp;
if (p == n && sum[n] != k) continue;
if (rec[p].pos > i && rec[p].v == sum[i] + k) {
printf("%d %d\n", i + 1, rec[p].pos);
return;
} else {
p++;
while (rec[p].v == sum[i] + k) {
if (rec[p].pos > i) {
printf("%d %d\n", i + 1, rec[p].pos);
return;
}
p++;
}
}
}
puts("No");
}
int main() {
#ifdef LOCAL
freopen("in.txt", "r", stdin);
freopen("out.txt", "w+", stdout);
#endif
int n, k;
while (~scanf("%d", &n)) {
for (int i = 1; i <= n; i++) {
scanf("%d", &arr[i]);
sum[i] = sum[i - 1] + arr[i];
rec[i].v = sum[i], rec[i].pos = i;
}
scanf("%d", &k);
sort(rec + 1, rec + n);
for (int i = 1; i <= n; i++) temp[i] = rec[i].v;
solve(n, k);
}
return 0;
}