基准时间限制: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
先预处理前缀和,并用map容器记录该前缀和的数量。因为a[j]=a[i]+k,所以枚举i,当前缀和为a[j]的数在map容器内时(即键值不为0),枚举j找到最小的i和j。
1 #include <iostream> 2 #include <map> 3 using namespace std; 4 typedef long long ll; 5 ll sum[10005]={0}; 6 int n,k; 7 map<ll,ll> m; 8 int flag=0; 9 int main() 10 { 11 ios::sync_with_stdio(false);//加快cin读入 12 cin>>n>>k; 13 for(int i=1;i<=n;i++) 14 { 15 int tmp; 16 cin>>tmp; 17 sum[i]=sum[i-1]+tmp;//前缀和 18 m[sum[i]]++;//前缀和为sum[i]的数量 19 } 20 for(int i=0;i<n;i++) 21 if(m[k+sum[i]]) 22 for(int j=i+1;j<=n;j++) 23 if(sum[j]-sum[i]==k) 24 { 25 cout<<i+1<<" "<<j<<endl; 26 flag=1; 27 return 0; //要求只输出一个区间 28 } 29 if(!flag) 30 cout<<"No Solution"<<endl; 31 return 0; 32 }