結果

問題 No.2211 Frequency Table of GCD
コンテスト
ユーザー lddlinan
提出日時 2023-03-10 15:54:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,062 bytes
コンパイル時間 874 ms
コンパイル使用メモリ 95,736 KB
最終ジャッジ日時 2025-02-11 07:26:48
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 16
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:63:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   63 |     scanf("%d %d", &n, &m);
      |     ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:67:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   67 |         scanf("%d", &v); cs[v]++;
      |         ~~~~~^~~~~~~~~~

ソースコード

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;


map<int, int> rs;
int cpp[128];
int pp[128000];
char pmk[1024];
void build(int n, int p, int c) {
    if (n==0) {
        rs[p]+=c;
        return;
    }
    n--;
    build(n, p, c);
    build(n, p*cpp[n], c);
}
int gc=0;
int gps[200004], gk, gm;
int bb[200004];
int cs[200004];
#define MOD 998244353
void dfs(int n, int p, int c) {
    if (p>gm) return;
    // printf("dfs %d %d %d==>%d\n", n, p, c, rs[p]);
    if (n==0) {
        int x = bb[rs[p]]-1;
        if (x<0) x+=MOD;
        if (c&1) { gc-=x; if (gc<0) gc+=MOD; }
        else { gc+=x; if (gc>=MOD) gc-=MOD; }
        return;
    }
    n--;
    dfs(n, p, c);
    dfs(n, p*gps[n], c+1);
}
int main() {
    int n, i, c, j, p, k, v;
    int m, kk, vv;
    memset(pmk, 0, sizeof(pmk));
    c=0;
    for (i=2; i<sizeof(pmk); i++) if (pmk[i]==0) {
        pp[c++]=i;
        for (j=i+i; j<sizeof(pmk); j+=i) pmk[j]=1;
    }
    set<int> ps;
    scanf("%d %d", &n, &m);
    for (i=0; i<=m; i++) cs[i]=0;
    bb[0]=1; for (i=1; i<=n; i++) bb[i]=(bb[i-1]*2)%MOD;
    for (i=0; i<n; i++) {
        scanf("%d", &v); cs[v]++;
    }
    for (i=1; i<=m; i++) {
        ps.clear();
        rs.clear();
        for (j=i, v=1; j<=m; j+=i, v++) if (cs[j]) {
            vv=v;
            kk=0;
            for (k=0; k<c; k++) {
                p=pp[k];
                if (p*p>vv) break;
                if (vv%p) continue;
                while((vv%p)==0) vv/=p;
                cpp[kk++]=p;
                ps.insert(p);
            }
            if (vv>1) { cpp[kk++]=vv; ps.insert(vv); }
            build(kk, 1, cs[j]);
        }
        gc=0;
        k=0; for (auto x: ps) gps[k++]=x; gm=m/i;
        dfs(k, 1, 0);
        printf("%d\n", gc);
    }
    return 0;
}
0