結果

問題 No.434 占い
ユーザー ytftytft
提出日時 2021-03-22 14:41:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,738 bytes
コンパイル時間 1,768 ms
コンパイル使用メモリ 175,328 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-24 01:03:11
合計ジャッジ時間 4,169 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other WA * 22 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

vector<int> merge(vector<int> a,int p,int MOD){
    if(p==0){
        int N=a.size()+1;
        vector<int> ret(N,1);
        for(int i=1;i<N-1;i++){
            ret[i]=(a[i-1]+a[i])%MOD;
        }
        return ret;
    }
    int A=a.size(),N=A+pow(3,p);
    vector<int> ret(N);
    for(int i=0;i<N;i++){
        if(0<=i-pow(3,p)+0 && i-pow(3,p)+0<a.size()){
            ret[i]+=a[i-pow(3,p)+0];
        }
        if(0<=i-pow(3,p)+pow(3,p-1) && i-pow(3,p)+pow(3,p-1)<a.size()){
            ret[i]+=a[i-pow(3,p)+pow(3,p-1)]*3;
        }
        if(0<=i-pow(3,p)+pow(3,p-1)*2 && i-pow(3,p)+pow(3,p-1)*2<a.size()){
            ret[i]+=a[i-pow(3,p)+pow(3,p-1)*2]*3;
        }
        if(0<=i-pow(3,p)+pow(3,p-1)*3 && i-pow(3,p)+pow(3,p-1)*3<a.size()){
            ret[i]+=a[i-pow(3,p)+pow(3,p-1)*3];
        }
        ret[i]%=MOD;
    }
    return ret;
}

vector<int> C(int step,int MOD){
    vector<int> three;
    while(step>0){
        three.push_back(step%3);
        step=step/3;
    }
    vector<int> cur={1};
    for(int i=0;i<three.size();i++){
        for(int j=0;j<three[i];j++){
            cur=merge(cur,i,MOD);
        }
    }
    return cur;
}

int main(){
    int T;
    cin>>T;
    string S;
    vector<int> c;
    cout<<endl;
    int temp;
    for(int i=0;i<T;i++){
        cin>>S;
        c=C(S.size()-1,9);
        temp=0;
        for(int j=0;j<S.size();j++){
            if(S[i]!='0'){
                break;
            }
            if(j==S.size()-1){
                cout<<0<<endl;
                goto lab;
            }
        }
        for(int j=0;j<S.size();j++){
            temp+=(S[j]-'0')*c[j];
        }
        cout<<(temp+8)%9+1<<endl;
        lab:{}
    }
}
0