。
#include <bits/stdc++.h>
using namespace std;
int w[1000000];
int main() {
long long n, q, z, sum = 0;
cin >> n >> q;
for (int i = 0; i < n; i++) {
cin >> w[i];
}
for (int i = 0; i < q; i++) {
cin >> z;
for (int j = 0; j < z; j++) {
sum = sum + w[j];
}
cout << sum << endl;
sum = 0;
}
return 0;
}
这里搜了一下前缀和,然后得到如下内容。
emmmmm类似于求1+2+.....+n;
这里用暴力算法的话时间复杂度就是O(n),然后如果n特别大再搞个嵌套的话就肯定会超时。
所以这里直接搞出来前缀和的作用:
降低代码的时间复杂度!
下面来看具体做法:
首先做一个预处理,定义一个sum[]数组,sum[i]代表a数组中前i个数的和。
然后代码大致就是这样滴
for(int i = 1; i <= n; i++){
s[i] = s[i-1] + a[i];
}
然后不难看出sum[i]代表a数组中前i个数的和。
然后这里放出一张图
这样,对于每个询问,只需要执行 sum[r]-sum[l-1]。输出原序列中从第l个数到第r个数的和的时间复杂度变成了O(1)。
emmmmm最后做一个小总结。
1.首先预处理,用sum[]来统计a[]里的前几项和。这里时间复杂度为O(n),
2.然后开始查询,s[r]-s[l-1]="a[l]到a[r]的和"。
emmm然后就没了吧?
最最后。。。。
#include <bits/stdc++.h>
using namespace std;
int w[1000000];
long long sum[1000000];
int main() {
long long n, q, z, p;
cin >> n >> q;
for (int i = 0; i < n; i++) {
cin >> w[i];
}
for (int i = 0; i < n; i++) {
sum[i] = sum[i - 1] + w[i];
}
for (int i = 0; i < q; i++) {
cin >> p;
cout << sum[p - 1] << endl;//这里我的i是从0开始的..并且刚开始输入的时候也输入了w[0]。
}
return 0;
}