Codeforces 812C Sagheer and Nubian Market[二分]

题意：给了n个物品，每个物品的价值为ai，你可以选取其中的k个，代价是k个ai的和加上他们的下标乘k的和。

问，在给定最大代价S的条件下，最多能取几个，取的最多的情况下，代价是多少。

分析：对能取几个进行二分，每次判断的时候，将原来的数组对ai+i*mid排序，取前mid个，看是否大于S。

以下是代码：

ll n, s, ans1, ans2;struct lx {ll id, v, k;friend bool operator<(const lx &a, const lx &b) {return a.v + a.id*a.k < b.v + b.id*b.k;}}tmp[MAXN], a[MAXN];bool check(int mid) {for (int i = 1; i <= n; ++i)tmp[i] = a[i], tmp[i].k = mid;sort(tmp + 1, tmp + 1 + n);ll sum = 0;for (int i = 1; i <= mid; ++i) {sum += tmp[i].v + tmp[i].id*tmp[i].k;}ans2 = sum;if (sum > s)return false;else return true;}int main() {scanf("%I64d%I64d", &n, &s);for (int i = 1; i <= n; ++i) {scanf("%I64d", &a[i].v);a[i].id = i;}int l = 0, r = n + 1;while (l < r) {int mid = (l + r) >> 1;if (check(mid)) {l = mid + 1;}else r = mid;}l--; ans1 = l;check(l);printf("%I64d %I64d\n", ans1, ans2);}