1. 程式人生 > 其它 >Codeforces Round #782 (Div. 2) C. Line Empir

Codeforces Round #782 (Div. 2) C. Line Empir

C. Line Empire

思路:本題的核心需要考慮:每次佔領新的王國以後是否需要更換首都的位置?

只需要比較一下:更換軍隊位置的花費和更換後可以減少佔領後續王國的花費進行比較,如果前者大就不換,後者大就換,這樣總花費是最小的。

我們需要先預處理出\(x_i\)的字首和。

更換位置的花費: 假如是從第 \(pre\)個王國佔領到第\(i\)個王國, \(w1 = abs(c[i] - c[pre]) *a\)

假如更換以後可以減少佔領後續王國的花費:\(w2 = (n - i) * b * (c[i] - c[pre])\)

只要比較這兩個的花費然後選擇方案即可。

LL w[N], sum[N];
 
void solve()
{
	cin >> n >> a >> b;
	for(int i = 0; i <= n; i ++ ) sum[i] = 0;
	int pre = 0;
	for(int i = 1; i <= n; i ++ ){
		cin >> w[i];
		sum[i] = sum[i - 1] + w[i];
	}
	LL ans = 0;
	for(int i = 1; i <= n; i ++ ){
		ans += abs(w[i] - w[pre]) * b;
		if((w[i] - w[pre]) * b * 1ll * (n - i) > a * abs(w[i] - w[pre])){
			ans += a * abs(w[i] - w[pre]);
			pre = i;
		}
	}
	cout << ans << endl;
}