題目1554：區間問題

題目描述：

給定一個數組，判斷數組內是否存在一個連續區間，使其和恰好等於給定整數k。

輸入：

輸入包含多組測試用例，每組測試用例由一個整數n(1<=n<=10000)開頭，代表數組的大小。
接下去一行爲n個整數，描述這個數組，整數絕對值不大於100。
最後一行爲一個整數k(大小在int範圍內)。

輸出：

對於每組測試用例，若存在這個連續區間，輸出其開始和結束的位置，s，e(s <= e)。
若存在多個符合條件的輸出，則輸出s較小的那個，若仍然存在多個，輸出e較小的那個。
若不存在，直接輸出"No"。

樣例輸入：

5
-1 2 3 -4 9
5
3
-1 2 -3
7
2
-1 1
0

樣例輸出：

2 3
No
1 2

思路：求出累加和存入SumList[n],，用優先隊列存儲對應累加和的編號，從小到大遍歷累加和序列，在優先隊列中找到合適的解

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
int SumList[10010];
int NumList[10010];
priority_queue<int> mask[2000010];
int n, k;
int main()
{
	while(scanf("%d", &n) != EOF)
	{
		int sum;
		memset(SumList, 0, sizeof(SumList));
		for(int i = 1; i <= n; i++)
		{
			scanf("%d", &NumList[i]);
			SumList[i] = NumList[i] + SumList[i - 1];
			mask[SumList[i] + 1000001].push(i);
		}
		scanf("%d", &k);
		bool res = false;
		int temp, tempIndex;
		for(int i = 0; i < n; i++)
		{
			tempIndex = SumList[i] + k + 1000001;
			if(mask[tempIndex].size())
			{
				if(mask[tempIndex].top() < i + 1)
				{
					continue;
				}
				while(mask[tempIndex].size() && mask[tempIndex].top() >= i + 1)
				{
					temp = mask[tempIndex].top();
					mask[tempIndex].pop();
				}
				printf("%d %d\n", i + 1, temp);
				res = true;
				break;
			}
		}
		if(!res)
		{
			printf("No\n");
		}
		for(int i = 1; i <= n; i++)
		{
			tempIndex = SumList[i] + 1000001;
			if(mask[tempIndex].size())
			{
				mask[tempIndex].pop();
			}
		}
	}
	return 0;
}