- 题目描述:
-
给定一个数组,判断数组内是否存在一个连续区间,使其和恰好等于给定整数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;
}