51nod 1094 和为k的连续区间

1094 和为k的连续区间

一整数数列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

题解：

自己的思路是前缀和，但是题目说是map,不知道怎么写。

先上自己写的前缀和。

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int maxn =10005;
LL k;
LL a[maxn];
LL sum[maxn];

int main()
{
int n;
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>a[i];
sum[0]=0;
bool flag=1;
for(int i=1;i<=n;i++)
{
sum[i]+=sum[i-1]+a[i];
}
for(int i=1;i<=n;i++){
if(!flag) break;
for(int j=i+1;j<=n;j++)
{
LL tmp = sum[j]-sum[i-1];
else if(tmp==k) {cout<<i<<" "<<j<<endl;flag=0;break;}
}
}
if(flag) cout<<"No Solution"<<endl;
return 0;
}

其实Map优化的时这个和存不存在，不存在则不需要进行下一步

map优化

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int maxn =10005;
LL k;
LL a[maxn];
LL sum[maxn];
map<LL,LL> m;
int main()
{
int n;
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>a[i];
sum[0]=0;
bool flag=1;
for(int i=1;i<=n;i++)
{
sum[i]+=sum[i-1]+a[i];
m[sum[i]]++;
}
for(int i=1;i<=n;i++){
if(!flag) break;
if(m[sum[i-1]+k])
for(int j=i+1;j<=n;j++)
{
LL tmp = sum[j]-sum[i-1];
if(tmp==k) {cout<<i<<" "<<j<<endl;flag=0;break;}
}
}
if(flag) cout<<"No Solution"<<endl;
return 0;
}