1. 程式人生 > >每日一題之 面試題 幸運巧克力(字首和+hash)

每日一題之 面試題 幸運巧克力(字首和+hash)

題目描述

在這裡插入圖片描述
在這裡插入圖片描述

思路:

注意到每個元素都是 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; }