C. Pekora and Trampoline
题意:对于数组a
,每次出发开始可以选择任意元素作为起始点,然后在数组上移动,落点为i + a[i]
,直至超出数组范围,每次经过的点的值减一(先移动再减/直至减到1为止),求使数组元素全为1所用最少的出发次数
数据范围:数组大小n
:
[
1
,
5000
]
[1,5000]
[1,5000] 元素大小 a[i]
:
[
1
,
1
0
9
]
[1,10^9]
[1,109]
思路:模拟+优化
首先观察元素大小的最大值远高于数组大小上限,此时站在上面会直接超出数组范围,直接模拟会多次重复该动作消耗大量时间,因此可以对其超出数组范围的部分批量一次性处理;
再者是数组不断趋向全为1的状态,中间态的数组难免出现大量的连续的1,直接模拟经过连续1时只能一步步走则会消耗大量时间,因此可以另开一个数组去对应记录当该值为1时最终会跳那个非1的值的位置
#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
#define fastio() ios_base::sync_with_stdio(0);cin.tie(0);
typedef long long ll;
const int N = 5010;
int a[N], n;
using namespace std;
void SetLeft(vector<int>& b, int j) //当数组出现新的1时,去更新该1及相邻的左边的1的跳转点
{
b[j] = b[j + 1];
for (int i = j - 1; i >= 1; i--)
{
if (a[i] == 1) b[i] = b[i + 1];
else return;
}
}
int main()
{
fastio();
int t;
cin >> t;
while (t--)
{
cin >> n;
for (size_t i = 1; i <= n; i++) cin >> a[i];
ll ans = 0;
vector<int> b(n + 2);
b[n + 1] = n + 1; //使最后一个元素a[n]为1时跳到n+1(即超出数组范围)
for (int i = n; i >= 1; i--)
{
if (a[i] != 1) b[i] = i;
else b[i] = b[i + 1];
}
while (1)
{
int beg = b[1]; //每次从最左端出发(最优性待证明)
if (beg > n) break; //此时说明数组全为1
if (a[beg] > max(n - beg, 1)) //批量处理超出数组范围的部分
{
ans += a[beg] - max(n - beg, 1); //max函数避免当beg=n时出现0
a[beg] = max(n - beg, 1);
if (a[beg] == 1) SetLeft(b, beg);
continue;
}
int j = beg, k;
while (j <= n)
{
if (a[j] == 1) k = b[j]; //元素为1时快速跳转
else k = j + a[j];
if (a[j] == 2) SetLeft(b, j); //此时说明将数组出现新的1
a[j] = max(a[j] - 1, 1);
j = k;
}
ans++;
}
cout << ans << "\n";
}
return 0;
}