結果

問題 No.1404 誕生日プレゼント
ユーザー pockynypockyny
提出日時 2022-01-31 08:40:45
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,279 bytes
コンパイル時間 817 ms
コンパイル使用メモリ 81,292 KB
実行使用メモリ 21,444 KB
最終ジャッジ日時 2023-09-02 02:04:22
合計ジャッジ時間 21,939 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 120 ms
21,440 KB
testcase_01 AC 5 ms
13,644 KB
testcase_02 AC 119 ms
21,048 KB
testcase_03 AC 129 ms
21,084 KB
testcase_04 AC 151 ms
21,052 KB
testcase_05 AC 179 ms
21,240 KB
testcase_06 AC 211 ms
21,056 KB
testcase_07 AC 246 ms
21,264 KB
testcase_08 AC 277 ms
21,148 KB
testcase_09 AC 310 ms
21,056 KB
testcase_10 AC 339 ms
21,176 KB
testcase_11 AC 386 ms
21,444 KB
testcase_12 AC 423 ms
21,124 KB
testcase_13 TLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

using namespace std;
typedef long long ll;
ll a[100010],b[100010],d[100010],p = 223577;
int cnt[300010];
vector<int> pa[300010];
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];
    }
    ll inv = pw(a[j],p - 2);
    for(i=0;i<n;i++){
        (a[i] *= inv) %= p; (b[i] *= inv) %= p;
    }
    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;
    }
    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;
    }
    int rr = 1;
    for(i=0;i<p;i++){
        cnt[rr]++; rr *= 2; rr %= p;
    }
    cnt[0]++;
    vector<int> v;
    for(i=0;i<p;i++){
        if(cnt[i]) v.push_back(i);
    }
    for(i=0;i<p;i++){
        for(j=0;j<v.size();j++){
            if(i>=v[j] && cnt[i - v[j]]){
               pa[i].push_back(v[j]); pa[i].push_back(i - v[j]);
               break; 
            }
            if(j==v.size() - 1){
                //cout << i << endl;
                //cout << " xxx " << endl;
            }
        }
    }
    pa[13].push_back(1); pa[13].push_back(4); pa[13].push_back(8);
    pa[25].push_back(1); pa[25].push_back(8); pa[25].push_back(16);
    for(j=0;j<p;j++){
        bool f = true;
        for(i=0;i<n;i++){
            if(a[i]==b[i]) continue;
            f = false;
            if(pa[d[i]][0]==a[k]) solve(i,k);
            if(pa[d[i]].size()>1 && pa[d[i]][1]==a[k]) solve(i,k);
            if(pa[d[i]].size()>2 && pa[d[i]][2]==a[k]) solve(i,k);
            if(pa[d[i]].size()>3 && pa[d[i]][3]==a[k]) solve(i,k);
        }
        if(f) break;
        solve(k,k);
    }
    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]) return 0;
    }
    if(ans.size()>p) 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