【问题描述】
对于一个具有 *n* 个元素的数组 *a* , 执行以下操作:
-
首先,选择下标 i* (1 ≤ *i* ≤**n*)—— 设置为数组的开始位置,放一个标记在 *i** 处(值 为**ai***)
-
当 i* ≤ *n* 时,你的得分将增加 *ai ,*且将标记向右移动 *ai* 个位置即下个位置是 *i* +**ai*** ,继续这个过程直到 i>n
-
如果 i* >**n,***则结束操作
【输入形式】
输入的第一行为一个整数 *t* (1≤ *t* ≤ 104),表示测试用例的组数。
每个测试用例的第一行为一个整数 *n* (1≤ *n* ≤ 2×105),表示数组 *a* 的元素个数
接下来一行包含 n* 个整数 *a1*、**a2*、...、*an*** (1 ≤ a***i*** ≤ 109),表示数组 *a* 的元素
【输出形式】
对于每个测试用例,输出独立一行,表示选择合适的开始位置后经过上述操作可以获得的最大分数。
【样例输入】
4 5 7 3 1 2 3 3 2 1 4 6 2 1000 2 3 995 1 5 1 1 1 1 1
【样例输出】
7 6 1000 5
【判例说明】
20%的测试数据 n≤10000
70%的测试数据 n≤100000
100%的测试数据 n≤200000
题目大概就是这么个事儿,这个加强主要体现在测试数据部分,可以看到数据规模很大,如果直接暴力解题,或许很难全部通过。其实我一开始就知道暴力求解无法通过,但还是义无反顾地A了上去。你知道的,人就是这样,当然这样做也没错,毕竟很多事试试才知道,万一就成了呢?人不能失去尝试的勇气,就像只狼不能失去忍义手。
但是现实很残酷,只过了6组测试数据。于是我开始思考,怎么优化呢?首先想到的是提升I/O流输入的性能,于是我添加了ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);这一行代码,取消C++的I/O库与C的I/O库(即stdio.h
)的同步,将cin和cout解绑。然而并没有什么卵用,还是只能通过6组测试数据。
看来只能从算法方面下手了啊,也是,原本时间复杂度接近O(n^2)的算法会超时是理所当然的。思索过后,我发现其实可以一边读入数据一边处理,于是就有了下面的代码。代码很简单,就算不写注释相信聪明的你也一定能看懂的🤗🤗🤗
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
int arr[200010]={0};
int main(){
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int t,n,sum,tmp;
cin>>t;
for(int i=0;i<t;i++){
memset(arr,0,sizeof(arr));
cin>>n;
sum=0;
for(int j=0;j<n;j++){
cin>>tmp;
if(j+tmp<n)
arr[j+tmp]=max(arr[j+tmp],arr[j]+tmp);
else
sum=max(sum,arr[j]+tmp);
}
sum=max(sum,arr[n-1]);
printf("%d\n",sum);
}
}