每日一题之 面试题 幸运巧克力（前缀和+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;
}