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;
}