51nod 1094 和为k的连续区间 （暴力||map数组优化）

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

因为是求连续的区间和，想当然的开个数组维护一下前n个数字的和。。。 然后开始大暴力 ，O(n*n),,, 这里的测试数据太水。。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e4+100;
ll a[maxn],pre[maxn];
int main()
{
	ios::sync_with_stdio(false); 
	int n,i,j,ok;
	ok=1;
	ll k;
	cin>>n>>k;
	for(i=1;i<=n;i++) {
		cin>>a[i];
		pre[i]=pre[i-1]+a[i];
	}
	for(i=1;i<=n;i++) {
		for(j=i;j<=n;j++) {
			if(pre[j]-pre[i-1]==k) {
				ok=0;
				break;
			}
		}
		if(!ok) break;
	}
	if(ok) cout<<"No Solution"<<endl;
	else cout<<i<<" "<<j<<endl;
	return 0;
}

这个有个O(n) 的算法，开个map数组维护一下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e4+100;
map<ll,int>mp;
ll a[maxn],pre[maxn];
int main()
{
	int ok,n,i,j;
	ok=1;
	ll k;
	cin>>n>>k;
	for(i=1;i<=n;i++) {
		cin>>a[i];
		pre[i]=pre[i-1]+a[i];
		mp[pre[i]]++;
	}

	for(i=0;i<=n;i++) {
		if(mp[pre[i]+k]>0) {
			for(j=i;j<=n;j++){
				if(pre[j]-pre[i]==k) {
					cout<<i+1<<" "<<j<<endl;
					return 0;
				}
 			}
		}
		mp[pre[i-1]]--;
	}
    cout<<"No Solution"<<endl;
	return 0;
}