CF 1703 G. Good Key, Bad Key 思维 1600

题意：从1 到 n 依次打开 n 个宝箱，每个宝箱分别有 ai 个金币，可以用两种方式打开宝箱：1.用好钥匙花费k个金币。2.用坏钥匙不花费金币，使当前宝箱到后面所有宝箱金币数减半。问怎么开启宝箱可以获得最多金币（金币可以是负数）。

思路：假设先用一把坏钥匙，再用一把好钥匙，获得的金币是  ⌊ai / 2⌋ + ⌊ai+1 / 2⌋ − k。而先用一把好钥匙，再用一把坏钥匙，获得的金币是 ai + ⌊ai+1 / 2⌋ − k。所以在用坏钥匙之前全部都是用好钥匙，用坏钥匙后，后面全都用坏钥匙。log2(109)≈32 所以最多会有32个宝箱中还有金币，后面所有的宝箱金币都变0，所以暴力计算从每个 i 开始使用坏钥匙计算中后面32个宝箱一共有多少金币就可以。

代码

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 2e5 + 10, P = 1e9 + 7, mod = 998244353;
ll a[N];
void solve(){
	int n, k;
	cin >> n >> k;
	for(int i = 1; i <= n; i++) cin >> a[i];
	ll sum = 0;
	ll ans = -0x3f3f3f3f;
	for(int i = 1; i <= n; i++){
		ll s = 0;
		for(int j = 0; j <= 32; j++){
			if(i + j > n) break;
			s += a[i + j] >> (j + 1);
		}
		ans = max(ans, sum + s);
		sum += a[i] - k;
	} 
	cout << max(ans, sum) << endl;
}                 
int main(){
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int t;
	cin >>t;
	while(t--){
		solve();
	}
	return 0;
}