题目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;
}



你可能感兴趣的:(题目1554:区间问题)