版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u014046022/article/details/83662698
题目描述
思路:
注意到每个元素都是 x >= 0, 所以该数组的前缀和是递增的。要找到连续的子序列之和等于k 即找到 sum[i] - sum[j] = k 且 0 <= j < i , 两重for循环会超时,注意到前缀和是递增的,那么将上式改写成 sum[i] - k = sum[j] ,这时候只要将前缀和映射成下标,这样只需要一重for循环,也就是遍历sum[i]的时候,只要查看mp[sum[i] - k] 是否存在,如果存在就存在一组解,且起点是 j+1 终点是 i。
#include <cstdio>
#include <iostream>
#include <vector>
#include <map>
using namespace std;
const int maxn = 1e5+7;
typedef long long llint;
llint A[maxn];
llint sum[maxn];
map<llint,int>mp;
int main() {
int n,k;
llint x;
cin >> n >> k;
for (int i = 1; i <= n; ++i) {
cin >> A[i];
if (i == 1) {
sum[i] = A[i];
}
else
sum[i] = sum[i-1] + A[i];
mp[sum[i]] = i;
}
if (sum[n] < k) {
cout << "0" << endl;
}
else {
int res = 0;
int l = 0, r = 0;
for (int i = 1; i <= n; ++i) {
if (sum[i] > k && mp[sum[i]-k] != 0) {
int j = mp[sum[i]-k];
if (i - j > res) {
res = i - j ;
l = j;
r = i;
}
}
else if (sum[i] == k) {
if (i > res) {
l = 0;
r = i;
}
}
}
cout << l << " " << r << endl;
}
return 0;
}