結果

問題 No.2206 Popcount Sum 2
ユーザー lddlinanlddlinan
提出日時 2023-03-14 13:09:19
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 870 ms / 4,000 ms
コード長 1,993 bytes
コンパイル時間 1,548 ms
コンパイル使用メモリ 87,368 KB
実行使用メモリ 181,272 KB
最終ジャッジ日時 2023-10-18 11:38:08
合計ジャッジ時間 18,198 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
7,348 KB
testcase_01 AC 3 ms
7,348 KB
testcase_02 AC 296 ms
180,744 KB
testcase_03 AC 294 ms
178,580 KB
testcase_04 AC 296 ms
181,272 KB
testcase_05 AC 868 ms
179,976 KB
testcase_06 AC 862 ms
179,976 KB
testcase_07 AC 869 ms
179,976 KB
testcase_08 AC 867 ms
179,976 KB
testcase_09 AC 870 ms
179,976 KB
testcase_10 AC 849 ms
179,976 KB
testcase_11 AC 849 ms
179,976 KB
testcase_12 AC 845 ms
179,976 KB
testcase_13 AC 829 ms
179,976 KB
testcase_14 AC 836 ms
179,976 KB
testcase_15 AC 839 ms
179,976 KB
testcase_16 AC 845 ms
179,976 KB
testcase_17 AC 847 ms
179,976 KB
testcase_18 AC 846 ms
179,976 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include <map>
#include <vector>
#include <queue>
#include <deque>
#include <set>
#include <stack>
#include <algorithm>
#include <array>
#include <unordered_set>
#include <unordered_map>
#include <string>
using namespace std;

bool rcmp(int a, int b) { return a>b; }
typedef long long LL;

#define MOD 998244353
int expit(LL b, int e) {
    LL r=1;
    while(e) {
        if (e&1) { r*=b; r%=MOD; }
        b*=b; b%=MOD;
        e>>=1;
    }
    return r;
}
int bb[200004];
int ss[200004];
int ft[200004];
int ift[200004];
#define BB 500
int xx[401][200004]; // (500+200000)*400/2
int ns[200004];
int ms[200004];
int main() {
    int n, i, m, k, nn;
    int inv2 = expit(2, MOD-2);
    int xn;
    int tc; scanf("%d", &tc);
    for (i=0; i<tc; i++) scanf("%d %d", &ns[i], &ms[i]);
    xn=ns[0]; for (i=0; i<tc; i++) xn=max(xn, ns[i]);
    LL r, t, s;
    bb[0]=1;
    ft[0]=ift[0]=1;
    for (i=1; i<=xn+BB&&i<=200000; i++) {
        bb[i]=(bb[i-1]*2)%MOD;
        r=ft[i-1]; r*=i; r%=MOD;
        ft[i]=r;
        ift[i]=expit(r, MOD-2);
    }
    ss[0]=0; for (i=1; i<=xn; i++) ss[i]=(ss[i-1]+bb[i-1])%MOD;
    for (n=BB; n<=xn+BB&&n<=200000; n+=BB) {
        k = n/BB-1;
        s=0; for (m=0; m<=n; m++) {
            t=ft[n]; t*=ift[m]; t%=MOD; t*=ift[n-m]; t%=MOD;
            s+=t; s%=MOD;
            xx[k][m]=s;
        }
    }
    // while(tc) { tc--;
    //     scanf("%d %d", &n, &m);
    for (i=0; i<tc; i++) {
        n=ns[i]; m=ms[i];
        n--; m--;
        k = (n+BB-1)/BB-1;  if (k<0) k=0;
        nn = (k+1)*BB;
        r = xx[k][m];
        while(nn>n) {
            nn--;
            // move nn-> nn-1
            t=ft[nn]; t*=ift[m]; t%=MOD; t*=ift[nn-m]; t%=MOD;
            t+=r; t%=MOD; t*=inv2; t%=MOD;
            r=t;
        }
        // c(n-1, m-1)+...C(n-1, 0)
        // r=ft[n-1]; r*=ift[m-1]; r%=MOD; r*=ift[n-m]; r%=MOD;
        r*=ss[n+1]; r%=MOD;
        printf("%lld\n", r);
    }
    return 0;
}
0