https://www.luogu.com.cn/problem/CF521D
有赋值,加,乘这3个操作,让最后的连乘积最大
首先乘可以同一计算,不管乘在那个数都一样,可以考虑把加和赋值转成乘
肯定是先赋值再加,所以可以把赋值转换为加,即每个位置保存最大的那个赋值
对于每个位置,肯定先加最大的,所以可以把加法从小到达排序,然后加又可以转换成乘
x+=y 变成 x*=(x+y)/x;
想清楚顺序就行了,主要是要意识到这几个操作是可以互相转换的
code:
#include<bits/stdc++.h>
#define N 200005
#define ll long long
#define pi pair<int, int>
#define pd pair<double, int>
#define mp make_pair
using namespace std;
vector<pi> ad[N];
vector<pd> ml;
pi as[N];
int n, m, k, a[N], ans[N], op[N];
int cmp(int x, int y) {
return op[x] < op[y];
}
int main() {
scanf("%d%d%d", &n, &m, &k);
for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
for(int i = 1; i <= m; i ++) {
int c, x, y;
scanf("%d%d%d", &c, &x, &y); op[i] = c;
if(c == 1) as[x] = max(as[x], mp(y, i));
if(c == 2) ad[x].push_back(mp(y, i));
if(c == 3) ml.push_back(mp(y, i));
}
for(int i = 1; i <= n; i ++) {
if(as[i].first > a[i]) {
ad[i].push_back(mp(as[i].first - a[i], as[i].second));
}
}
for(int i = 1; i <= n; i ++) {
sort(ad[i].begin(), ad[i].end(), greater<pi>());
ll sum = a[i];
for(int j = 0; j < ad[i].size(); j ++) {
ll val = ad[i][j].first, id = ad[i][j].second;
ml.push_back(mp(1.0 * (val + sum) / sum, id));
sum += val;
}
}
sort(ml.begin(), ml.end(), greater<pd>());
k = min(k, (int)ml.size());
printf("%d\n", k);
for(int i = 0; i < k; i ++) ans[i + 1] = ml[i].second;
sort(ans + 1, ans + 1 + k, cmp);
for(int i = 1; i <= k; i ++) printf("%d ", ans[i]);
return 0;
}