Codeforces Global Round 13 C

C. Pekora and Trampoline

题意：对于数组a，每次出发开始可以选择任意元素作为起始点，然后在数组上移动，落点为i + a[i]，直至超出数组范围，每次经过的点的值减一（先移动再减/直至减到1为止），求使数组元素全为1所用最少的出发次数

数据范围：数组大小n:$[1,5000]$ 元素大小 a[i]:$[1,10^9]$

思路：模拟+优化
首先观察元素大小的最大值远高于数组大小上限，此时站在上面会直接超出数组范围，直接模拟会多次重复该动作消耗大量时间，因此可以对其超出数组范围的部分批量一次性处理；
再者是数组不断趋向全为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;
}