51nod 1094 和为k的连续区间【前缀和/区间差/map】
基准时间限制: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
【代码】:
#include <bits/stdc++.h> using namespace std; #define inf 1e18+100 #define LL long long const int maxn = 1e4+100; int main() { int n,k; int a[maxn],sum[maxn]; memset(sum,0,sizeof(sum)); cin>>n>>k; for(int i=1;i<=n;i++) { cin>>a[i]; sum[i]=sum[i-1]+a[i]; } for(int i=0;i<n;i++) { for(int j=i;j<n;j++) { if(sum[j]-sum[i]==k) { printf("%d %d\n",i+1,j); return 0; } } } printf("No Solution\n"); return 0; }
map优化可达到O(n)