洛谷 P4090 [USACO17DEC] Greedy Gift Takers P
题意
n n n 头牛排成一列,队头的奶牛 i i i 拿一个礼物并插到从后往前数 c i c_i ci 头牛的前面,重复无限次,问多少奶牛没有礼物。
题解
发现若一头牛无法获得礼物,那么它后面的牛都无法获得礼物。
发现获得礼物的牛构成一个循环。
二分获得礼物的牛的数量。假设有 x x x 头牛获得礼物,仅考虑第 x x x 头牛能否获得礼物。让它获得礼物,就要把它推到前面去。假设前 x − 1 x-1 x−1 头牛都能获得礼物。于是把前 x − 1 x-1 x−1 头牛按 a a a 从小到大排序,也就是把牛尽可能往后插。
假设牛 x x x 后面有 l i m lim lim 头牛。若 x x x 前面的牛 i i i 满足 a i > l i m a_i>lim ai>lim,那么这个 x x x 无法获得礼物。因为 i i i 会插入到 x x x 的前面,又因为 a a a 值从小到大,那么 i i i 后面的牛都会插到 x x x 前面。若 a i ≤ l i m a_i\le lim ai≤lim, x x x 后面的牛多一头, l i m lim lim 加 1 1 1。
代码
#include <bits/stdc++.h>
using namespace std;
template<typename Ty> void read(Ty &x) {
int c = getchar(), f = 1;
for (; c < '0' || c > '9'; c = getchar()) if (c == '-') f = -1;
for (x = 0; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);
x *= f;
}
const int N = 100005;
int n, a[N], b[N], ans = 0;
bool check(int x) {
for (int i = 1; i < x; i++) b[i] = a[i];
sort(b + 1, b + x);
int lim = n - x;
for (int i = 1; i < x; i++, lim++)
if (b[i] > lim)
return 0;
return 1;
}
int main() {
read(n);
for (int i = 1; i <= n; i++) read(a[i]);
int l = 0, r = n;
while (l <= r) {
int mid = l + r >> 1;
if (check(mid)) l = mid + 1, ans = mid;
else r = mid - 1;
}
printf("%d", n - ans);
return 0;
}