結果

問題 No.2075 GCD Subsequence
ユーザー HIcoderHIcoder
提出日時 2023-07-12 08:37:07
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,403 ms / 4,000 ms
コード長 2,604 bytes
コンパイル時間 1,241 ms
コンパイル使用メモリ 109,564 KB
実行使用メモリ 24,576 KB
最終ジャッジ日時 2024-09-13 22:26:22
合計ジャッジ時間 41,780 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 81 ms
22,984 KB
testcase_01 AC 80 ms
22,948 KB
testcase_02 AC 81 ms
23,012 KB
testcase_03 AC 80 ms
22,868 KB
testcase_04 AC 80 ms
22,936 KB
testcase_05 AC 81 ms
22,912 KB
testcase_06 AC 82 ms
23,012 KB
testcase_07 AC 82 ms
22,896 KB
testcase_08 AC 1,020 ms
23,808 KB
testcase_09 AC 1,632 ms
24,560 KB
testcase_10 AC 1,069 ms
23,936 KB
testcase_11 AC 1,386 ms
24,192 KB
testcase_12 AC 1,211 ms
24,064 KB
testcase_13 AC 901 ms
23,680 KB
testcase_14 AC 1,367 ms
24,140 KB
testcase_15 AC 956 ms
23,696 KB
testcase_16 AC 1,021 ms
23,744 KB
testcase_17 AC 1,620 ms
24,412 KB
testcase_18 AC 2,363 ms
24,448 KB
testcase_19 AC 2,374 ms
24,400 KB
testcase_20 AC 2,368 ms
24,524 KB
testcase_21 AC 2,366 ms
24,520 KB
testcase_22 AC 2,388 ms
24,400 KB
testcase_23 AC 2,369 ms
24,516 KB
testcase_24 AC 2,383 ms
24,448 KB
testcase_25 AC 2,377 ms
24,448 KB
testcase_26 AC 2,374 ms
24,448 KB
testcase_27 AC 2,374 ms
24,576 KB
testcase_28 AC 2,403 ms
24,460 KB
testcase_29 AC 80 ms
22,912 KB
testcase_30 AC 81 ms
22,868 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
AC

余事象を用いる
yukicoder 2055の知識


*/
#include<iostream>
#include<set>
#include<algorithm>
#include<vector>
#include<string>
#include<set>
#include<map>
#include<numeric>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
const ll INF=1LL<<60;
typedef pair<int,int> P;
typedef pair<int,P> PP;
const ll MOD=998244353;
const double PI=acos(-1);

struct Mobius{

    std::vector<int> lists;
    std::vector<bool> isprime;
    int n;
    Mobius(int n_):n(n_){
        lists=std::vector<int>(n+1,1);
        isprime=std::vector<bool>(n+1,true);

        init();
    }

    void init(){

        isprime[0]=isprime[1]=false;
        for(long long i=2;i<=n;i++){
            if(!isprime[i]) continue;//iが素数でない場合


            lists[i]=(-1)*lists[i];

            for(long long j=2*i;j<=n;j+=i){
                isprime[j]=false;

                if( ((j/i)%i)==0 ){
                    //jは (i*i)で割れる
                    lists[j]=0;
                }
                lists[j]*=(-1);
            }
        }
    
    }

    int operator()(int n){
        return lists[n];
    }

    
};

Mobius mobius(1e6+10);

int main(){
    int N;
    cin>>N;
    vector<ll> A(N);
    for(int i=0;i<N;i++){
        cin>>A[i];
    }
    //map<ll,ll> dp; これだとTLEする
    //map<ll,ll> g;

    vector<ll> dp(1000000+10,0);//dp[x]=部分列がxで終わる数列
    vector<ll> g(1000000+10,0);

    ll tot=0;
    for(int i=0;i<N;i++){
        

        vector<ll> d;
        for(ll x=1;x*x<=A[i];x++){
            if(A[i]%x==0){
                d.push_back(x);
                if(A[i]/x!=x){
                    d.push_back(A[i]/x);
                }
            }
        }


        ll f1=0;

        for(ll x:d){

            //f1+=g[x]*mobius(x);
            if(mobius(x)>0){
                f1+=g[x]%MOD;
                f1%=MOD;
            }else if(mobius(x)<0){
                f1+=(MOD-g[x])%MOD;
                f1%=MOD;
            }
        }

        //ll add=1+tot-f1;
        ll add=0;
        add+=1;
        add%=MOD;
        add+=tot;
        add%=MOD;
        add+=(MOD-f1)%MOD;
        add%=MOD;


        
        dp[A[i]]+=add;
        dp[A[i]]%=MOD;
        tot+=add;
        tot%=MOD;

        for(ll x:d){
            //A[i]はxで割れる 
            g[x]+=add;//同時にgも更新.
            g[x]%=MOD;
        }

    }

    ll ans=0;
    // for(auto [key,val]:dp){
    //     ans+=val;
    //     ans%=MOD;
    // }
    for(int i=1;i<=1000000;i++){
        ans+=dp[i];
        ans%=MOD;
    }
    cout<<ans<<endl;

}
0