結果

問題 No.1404 誕生日プレゼント
ユーザー pockyny
提出日時 2022-01-31 06:08:56
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,245 bytes
コンパイル時間 1,093 ms
コンパイル使用メモリ 77,152 KB
最終ジャッジ日時 2025-01-27 18:00:30
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

using namespace std;
typedef long long ll;
ll a[100010],b[100010],d[100010],p = 223577;
ll pw(ll a,ll x){
    ll ret = 1;
    while(x){
        if(x&1) (ret *= a) %= p;
        (a *= a) %= p; x /= 2;
    }
    return ret;
}

vector<pair<int,int>> ans;
void solve(int i,int j){
    ans.push_back({i,j});
    (a[i] += a[j]) %= p;
}

void fin(){
    cout << ans.size() + 1 << endl;
    for(int i=0;i<ans.size();i++){
        cout << 1 << " " << ans[i].first + 1 << " " << ans[i].second + 1 << endl;
    }
    cout << 2 << endl;
}

int main(){
    int i,n,x; cin >> n >> x;
    for(i=0;i<n;i++) cin >> a[i];
    for(i=0;i<n;i++) cin >> b[i];
    bool f1 = false,f2 = false;
    for(i=0;i<n;i++){
        if(a[i]) f1 = true;
        if(b[i]) f2 = true;
    }
    if((f1 && !f2) || (!f1 && f2)){
        cout << -1 << endl;
        return 0;
    }
    if(!f1 && !f2){
        cout << 0 << endl;
        return 0;
    }
    vector<int> z1,z2;
    int j = -1,k = -1;
    for(i=0;i<n;i++){
        if(a[i]) z1.push_back(i);
        if(b[i]) z2.push_back(i);
    }
    if(z1.size()==1 && z2.size()==1 && z1[0]==z2[0]){
        j = z1[0];
        for(i=0;i<p;i++){
            if(a[j]!=b[j]) solve(j,j);
        }
        if(a[j]==b[j]) fin();
        else cout << -1 << endl;
        return 0;
    }
    j = z1[0],k = z2[0];
    if(j==k){
        if(z1.size()>1) j = z1[1];
        else k = z2[1];
    }
    //cout << "j = " << j << " k = " << k << endl;
    ll inv = pw(a[j],p - 2);
    for(i=0;i<n;i++){
        (a[i] *= inv) %= p; (b[i] *= inv) %= p;
    }
    /*cout << "rou1" << endl;
    for(i=0;i<n;i++) cout << a[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << b[i] << " ";
    cout << endl;*/
    if(a[k]>b[k]) a[k] -= p;
    ll dif = b[k] - a[k];
    while(dif){
        if(dif&1) solve(k,j);
        solve(j,j); dif /= 2;
    }
    /*cout << "rou2" << endl;
    for(i=0;i<n;i++) cout << a[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << b[i] << " ";
    cout << endl;*/
    ll inv2 = pw(a[k],p - 2);
    for(i=0;i<n;i++){
        (a[i] *= inv2) %= p;
        (b[i] *= inv2) %= p;
        d[i] = b[i] - a[i];
        if(d[i]<0) d[i] += p;
    }
    /*cout << "rou3" << endl;
    for(i=0;i<n;i++) cout << a[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << b[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << d[i] << " ";
    cout << endl;*/
    for(j=0;j<20;j++){
        bool f = true;
        //cout << a[k] << " ";
        for(i=0;i<n;i++){
            if(d[i]) f = false;
            if(d[i]&1) solve(i,k);
            d[i] /= 2;
            //cout << " ( " << a[i] << "," << d[i] << ")" << " ";
        }
        //cout << endl;
        if(f) break;
        solve(k,k);
    }
    /*cout << "rou4" << endl;
    for(i=0;i<n;i++) cout << a[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << b[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << d[i] << " ";
    cout << endl;*/
    for(i=0;i<p;i++){
        if(a[k]==b[k]) continue;
        solve(k,k);
    }
    fin();
    for(i=0;i<n;i++){
        if(a[i]!=b[i]) exit(1);
    }
    /*for(i=0;i<n;i++) cout << a[i] << " ";
    cout << endl;
    for(i=0;i<n;i++) cout << b[i] << " ";
    cout << endl;*/
}
0