[PKUSC2018]真实排名 数论

Description
给你n个人的成绩，定义一个人的成绩为：大于等于他的人的个数。
现在可能会有k个人的成绩乘于二，
对于第i个人，现在问有多少种方案使得第i个人的排名不发生改变。

---

Sample Input
3 2
1 2 3

---

Sample Output
3
1
2

---

先按权值排序。
对于第i个人如果他乘于二，那么他就可能会超过他前面的一些人，那么要想让他排名不变，他超过的这些人就都要乘于二，剩下的人随便分，二分出这个位置即可。
对于第i个人不乘于二，那么他就可能被后面一些人超过，你仍然是而分出一个位置，其他人就随便分即可。

---

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
typedef long long LL;
const LL mod = 998244353;
int read() {
	int s = 0, f = 1; char ch = getchar();
	while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
	while(ch >= '0' && ch <= '9') s = s * 10 + ch - '0', ch = getchar();
	return s * f;
}

struct node {
	int x, y, id;
} a[210000];
LL jc[210000], inv[210000];
LL ans[210000];

bool cmp(node a, node b) {return a.x < b.x;}

LL pow_mod(LL x, LL k) {
	LL ans = 1;
	while(k) {
		if(k & 1) (ans *= x) %= mod;
		(x *= x) %= mod; k /= 2;
	} return ans;
}

LL C(int n, int m) {
	if(m > n) return 0;
	return jc[n] * inv[m] % mod * inv[n - m] % mod;
}

int main() {
	int n = read(), k = read();
	for(int i = 1; i <= n; i++) a[i].x = read(), a[i].y = a[i].x * 2, a[i].id = i;
	sort(a + 1, a + n + 1, cmp);
	jc[0] = inv[0] = 1; for(int i = 1; i <= n; i++) jc[i] = (LL)jc[i - 1] * i % mod, inv[i] = pow_mod(jc[i], mod - 2);
	for(int i = 1; i <= n; i++) {
		if(a[i].x == a[i - 1].x && i != 1) {ans[a[i].id] = ans[a[i - 1].id]; continue;}
		int l = i + 1, r = n, hh = i;
		while(l <= r) {
			int mid = (l + r) / 2;
			if(a[mid].x < a[i].y) l = mid + 1, hh = mid;
			else r = mid - 1;
		} ans[a[i].id] += C(n - (hh - i + 1), k - (hh - i + 1));
		l = 1, r = i - 1, hh = i;
		while(l <= r) {
			int mid = (l + r) / 2;
			if(a[mid].y >= a[i].x) r = mid - 1, hh = mid;
			else l = mid + 1;
		} ans[a[i].id] += C(n - (i - hh + 1), k);
	} for(int i = 1; i <= n; i++) printf("%lld\n", ans[i] % mod);
	return 0;
}