LOJ #6432. 「PKUSC2018」真实排名

最新推荐文章于 2023-11-21 12:28:31 发布

weixin_30436101

最新推荐文章于 2023-11-21 12:28:31 发布

阅读量86

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/miecoku/p/9171932.html

本文介绍了一种快速计算组合数的方法，并应用于解决特定问题。通过预处理和使用费马小定理，在O(nlogn+logmod)的时间复杂度内解决了节点变化情况下的组合数计算问题。

题目在这里......

对于这道题，现场我写炸了......谁跟我说组合数O（n）的求是最快的？（~！@#￥￥%……&

#include <cstdio>
#include <algorithm>
using namespace std;

typedef long long ll;
const ll mod = 998244353;
const int N = 1e5 + 10;

ll ksm(ll x, ll n) {
    ll ans = 1ll; //printf("%lld ", x);
    while( n) {
        if( n&1) ans = ans * x % mod; x = x * x % mod;
        n >>= 1;
    }
//    printf("ans = %lld, %lld\n", ans, x);
    return ans;
}
ll zt[N];
ll lucas(int m, int n) {
    ll x = zt[m] * zt[n-m] % mod; //printf("%lld\n", ksm(x, mod-2));
    return zt[n] * ksm(x, mod-2) % mod;
}
int data[N], a[N], n, k;
ll query1(int x) {
    int r = lower_bound(data+1, data+1+n, a[x]) - data;
    int l = upper_bound(data+1, data+1+n, (a[x] + 1 >> 1) - 1) - data - 1;
    if( l + n - r < k) return 0ll;
    return lucas(k, l + n - r);
}

ll query2(int x) {
    int l = lower_bound(data+1, data+1+n, a[x]) - data;
    int r = lower_bound(data+1, data+1+n, a[x] << 1) - data;
    if( l == r) r ++;
    if( r - l > k) return 0ll;
    return lucas(k-r+l, l + n - r);
}

int main() {
    zt[1] = zt[0] = 1ll;
    for( int i = 2; i <= N-2; i ++) zt[i] = zt[i-1] * i % mod;
//    for( int i = 1; i <= 10; i ++) printf("%d ", zt[i]); puts("");
    scanf("%d%d", &n, &k);
    for ( int i = 1; i <= n; i ++) scanf("%d", a+i), data[i] = a[i];
    sort(data+1, data+1+n);
    for ( int i = 1; i <= n; i ++) 
        printf("%lld\n", (query1(i) + query2(i)) % mod);
}