51Nod - 1094 和为k的连续区间

Posted 2020-09-16 zhang--yd

51Nod - 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

 

 

题解： 

　　　使用c++ map 来求区间前后端口。 

 

#include <iostream> 
#include <map> 
#include <vector> 
#include <algorithm> 
using namespace std; 
const int MAXN = 10005; 

long long n, k, num[MAXN]; 

int main(){
	while(cin >> n >> k){
		map<long long, vector<int>> mp; 
		mp[0] = vector<int>{0}; 
		vector<pair<int,int> > ans; 
		num[0] = 0; 
		for(int i=1; i<=n; ++i){
			cin >> num[i]; 
			num[i] = num[i-1] + num[i];
			if( mp.find( num[i] ) == mp.end() ){
				mp[ num[i] ] = vector<int>{i}; 
			}else{
				mp[num[i]].push_back(i); 
			}

			if( mp.find( num[i] - k ) != mp.end() ){
				for(int j=0; j<mp[ num[i] - k ].size(); ++j ){
					ans.push_back( make_pair(mp[ num[i]-k ][j] + 1, i) ); 
				}
			}
		}
		if(ans.size() == 0){
			cout << "No Solution" << endl; 
		}else{
			sort(ans.begin(), ans.end()); 
			cout << ans[0].first << " " << ans[0].second << endl; 
		}
	}
	return 0; 
}