每日一題之 面試題 幸運巧克力(字首和+hash)
阿新 • • 發佈:2018-11-19
題目描述
思路:
注意到每個元素都是 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;
}