G2. Teleporters (Hard Version) codeforces 1791G2

Problem - G2 - Codeforces

题目大意：有n个传送门，每个门的使用费用为ai，可以传送到i=0或i=n+1的位置，不用门每次移动一个单位花费1，给定费用限额c，问从i=0处出发，最多使用几个门

1<=n<=2e5；1<=ai<=1e9

思路：因为无论选哪个门对答案的贡献都是1，所以我们在限额允许的情况下，选费用最小的即可，所以对于每一个门，最优费用就是ai+i和ai+n-i+1的最小值，但是因为起点是0，所以我们得找出哪个点作为起点，然后再依次选费用最低的门，所以我们先记录下每个门从0出发到它的费用和他的最优费用，然后将最优费用排序并求前缀和，这样我们再遍历1到n，枚举将每个门作为第一个门之后剩余的费用，再二分答案，找出满足剩余费用的最大前缀和，前缀和的下标就是门的数量，同时如果当前下标大于等于当前枚举的初始门的坐标，需要先减去初始门的费用，然后再判断加上从0到初始门的费用在不在限额之内，

#include<__msvc_all_public_headers.hpp>
//#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
typedef long long ll;
pair<ll, int>a[N];
ll sum[N];
int main()
{
	cin.tie(0);
	cout.tie(0);
	ios::sync_with_stdio(false);
	int t;
	cin >> t;
	while (t--)
	{
		int n, c;
		cin >> n >> c;
		ll ans = 0;
		for (int i = 1; i <= n; i++)
		{
			ll x;
			cin >> x;
			a[i].first = x + min(i, n - i + 1);//记录最优费用
			a[i].second = x + i;//记录作为初始门的费用
		}
		sort(a + 1, a + 1 + n);//按最优费用从小到大排序
		for (int i = 1; i <= n; i++)
		{//求最优费用的前缀和
			sum[i] = sum[i - 1] + a[i].first;
		}
		for (int i = 1; i <= n; i++)
		{
			int l = 0, r = n;//二分枚举除起始门之外走过了多少门，所以边界是0~n
			int mid;
			ll tans = 0;
			while (l <= r)
			{
				mid = (l + r) >> 1;
				ll temp = sum[mid];
				ll teans = mid+1;//记录的是所有门的数量，所以要加上起始门
				if (mid >= i)
				{//当前门在前缀和中被统计过
					temp -= a[i].first;
					teans--;
				}
				if (temp + a[i].second <= c)
				{//当前数量的门满足限额
					tans = max(tans, teans);//维护二分中的最大值
					l = mid + 1;
				}
				else
					r = mid - 1;
			}
			ans = max(ans, tans);//维护所有门中的最大值
		}
		cout << ans << endl;
	}
	return 0;
} 

对所有门取最大值即可