1. 程式人生 > 其它 >【題解】[LNOI2022] 吃

【題解】[LNOI2022] 吃

給定 \(n\) 個二元組 \((a_i,b_i)\),任意排列,初始化 \(S = 1\),每次可以選擇讓 \(S \leftarrow S + b_i\) 或者 \(S \leftarrow S \times a_i\)。求最大值。

首先加法一定在乘法前面,因為所有數都是正數,把加法前移只會更優。

顯然對於 \(a_i = 1\) 的二元組我們一定選擇加 \(b_i\),因為使用 \(a_i\) 等於沒有變化。所以我們可以預處理並刪去所有 \(a_i = 1\) 的二元組。

並且對於 \(a_i \neq 1\) 的二元組,我們最多隻會選擇一個數進行加法。反證,假設進行兩次加法 \(b_i, b_j\)

,不失一般性另 \(b_i \ge b_j\),由於 \(a_j \ge 2\),所以 \(2b_i \ge b_i + b_j\),第二次選擇乘法會更優。

這樣我們只用列舉哪個數加即可,另 \(K = \prod a_i\),如果不加答案為 \(KS\),如果加 \(b_i\) 那麼答案為 \(\dfrac{K(S+b_i)}{a_i}\),所以我們只用選出 \(\dfrac{S+ b_i}{a_i}\) 最大的二元組即可。

時間複雜度 \(\mathcal{O}(N)\),交了六發,要是考場上就可以原地退役了。

#define N 500005
int n, m, u[N]; LL S = 1;
struct node{int a, b; }a[N];
int main() {
	read(n);
	rp(i, n)read(u[i]);
	rp(i, n){
		int x = u[i], y; read(y);
		if(x == 1)S += y;
		else a[++m] = {x, y};
	}
	int pv = 0; long double w = S;
	rp(i, m){
		long double cur = (S + a[i].b) * 1. / a[i].a;
		if(cur > w)w = cur, pv = i;
	}
	S %= P;
	if(pv)S = (S + a[pv].b) % P;
	rp(i, m)if(i != pv)S = S * a[i].a % P;
	cout << S << endl;
	return 0;
}