CodeForces 1250G [貪心]

題目鏈接:https://codeforces.com/contest/1250/problem/G

 

解題思路:

題目的另一個意思就是每次按動reset意思就是原來的sa,sb(表示a,b現在的積分)變成了sa = sa -  min(sa,sb),sb = sb - min(sa,sb),根據這一的性質我們就可以都得到一個結論就是，當只執行到第i輪時，最後一次按下了reset是在第k輪，那麼這是sa[i] = pre_a[i] - min(pre_a[k],pre_b[k]),sb[i] = pre_b[i] - min(pre_a[k],pre_b[k])。pre[]表示數組前綴和，再記w[k] = min(pre_a[k],pre_b[k]

m 是題中最大積分的限制

那麼如果人想在第i輪贏得話，必須就要滿足存在k使得pre_a[i] - m < w[k] <= pre_b[i] - m, k < i。根據這個式子就可以去判斷如果到了第i輪是否有可以取勝的策略，那麼接下來我們就只要保證人到第i輪之前積分不會爆m就可以了，這個就直接貪心就好了，因爲已經確定了在第幾輪的時候可以贏了。

#include<bits/stdc++.h>
using namespace std;
const int mx = 2e5+5;
const int mod = 1e9 + 7; 
typedef long long ll;
int n,m;
int a[mx],b[mx],ans[mx];
ll sa[mx],sb[mx];
ll w[mx];
int get_id() {
	for (int i=1;i<=n;i++) {
		ll L = sa[i] - m;
		ll R = sb[i] - m;
		int l = upper_bound(w+1,w+i,L) - w;
		int r = upper_bound(w+1,w+i,R) - w;
		if (l < r) {
			//cout << l << " " << i << endl;
			return l;
		}
		w[i] = min(sa[i],sb[i]);
	}
	return -1;
}
int solve(int id) {
	ll now1 = 0,now2 = 0;
	int siz = 0;
	for (int i=1;i<id;i++) {
		now1 += a[i];
		now2 += b[i];
		if (now1 + a[i+1] >= m) {
			ll mi = min(now1,now2);
			ans[siz++] = i;
			now1 -= mi,now2 -= mi;
			if (now1 + a[i+1] >= m) return 0;
		}
	}
	ans[siz++] = id;
	return siz;
}
bool check() {
	for (int i=1;i<=n;i++)
		if (sa[i]<m&&sb[i]>=m)
			return 1;
	return 0;
}
int main(){
	int t;
	scanf("%d",&t);
	while (t--) {
		scanf("%d%d",&n,&m);
		for (int i=1;i<=n;i++) {
			scanf("%d",a+i);
			sa[i] = sa[i-1] + a[i];
		}
		for (int i=1;i<=n;i++) {
			scanf("%d",b+i);
			sb[i] = sb[i-1] + b[i];
		}
		if (check()) {
			puts("0\n");
			continue;
		}
		int id = get_id();
		int siz = solve(id);
		if (!siz || id == -1) {
			puts("-1");
			continue;
		}
		printf("%d\n",siz);
		for (int i=0;i<siz;i++) {
			printf("%d%c",ans[i],i==siz-1?'\n':' ');
		}
	}
	return 0;
}