1094 和为k的连续区间
基准时间限制:1 秒 空间限制:131072 KB 分值: 10 难度:2级算法题 收藏 关注
一整数数列a1, a2, … , an(有正有负),以及另一个整数k,求一个区间[i, j],(1 <= i <= j <= n),使得a[i] + … + a[j] = k。
Input
第1行:2个数N,K。N为数列的长度。K为需要求的和。(2 <= N <= 10000,-10^9 <= K <= 10^9)
第2 - N + 1行:A[i](-10^9 <= A[i] <= 10^9)。
Output
如果没有这样的序列输出No Solution。
输出2个数i, j,分别是区间的起始和结束位置。如果存在多个,输出i最小的。如果i相等,输出j最小的。
Input示例
6 10
1
2
3
4
5
6
Output示例
1 4
本来要睡,想着看一个题,思考一下,免得自己闲下来。发现这个题,感觉恰巧可以用刚刚写过的那个方法,简单的dp,然后利用两个for循环枚举一下。最后输出,这样的思路代码如下:
#include"cstdio"
#include"algorithm"
#include"cstring"
#include"cmath"
using namespace std;
int main ()
{
int n, k, i, j;
scanf("%d %d",&n,&k);
int a[n+1];
long long int sum[n+1];
sum[0]=0;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
int flag=0;
int minn=0;
for(i=1;i<=n;i++)
{
for(j=i;j<=n;j++)
{
if((sum[j-1]-sum[i-1])==k)
{
printf("%d %d\n",i,j-1);
flag=1;
break;
}
}
if((sum[j-1]-sum[i-1])==k)
break;
}
if(flag==0)
printf("No Solution\n");
return 0;
}
比较容易想到,但,写到这里,我突然想到之前做的一个题,也是51Nod上的一个题,求数组中等于K的数组,那个题是利用两个指针,从两边遍历,我印象也比较深。这个题可不可以也利用这个再一次的优化呢。我试一下。好吧,我试了一下,不行,那个的基础是需要排序,这个一旦排序,顺序乱了,无法确定区间端点。我搜了下其他人的代码。有利用map什么的,我目前还看不懂,今天已经看不下去了,睡觉,明天再看。/好吧,其实没什么的,整个map我还没看,不知道,单就这个题而言,无非是先判断这个数字是否存在,然后再找区间端点。我还以为有多难,都是纸老虎。不过复杂度从O(n*n)降到了O(n)。蛮不错。。/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
typedef long long LL;
const int MAXN = 1e4+5;
LL a[MAXN], sum[MAXN];
map<LL,LL>mp;
int main()
{
int N, K, i, j;
while(~scanf("%d%d",&N,&K))
{
memset(sum, 0, sizeof(sum));
for(i=1; i<=N; i++)
{
scanf("%lld",&a[i]);
sum[i] = sum[i-1] + a[i];
mp[sum[i]]++;
}
for(i=0; i<=N; i++)
if(mp[sum[i]+K])
for(j=i; j<=N; j++)
if(sum[j]-sum[i] == K)
{
printf("%d %d\n",i+1, j);
goto endW;
}
puts("No Solution");
endW:;
}
return 0;
}
自取。