結果

問題 No.2719 Equal Inner Products in Permutation
ユーザー Jeroen Op de BeekJeroen Op de Beek
提出日時 2024-04-05 23:39:28
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 49 ms / 2,000 ms
コード長 2,533 bytes
コンパイル時間 1,980 ms
コンパイル使用メモリ 209,464 KB
実行使用メモリ 8,192 KB
最終ジャッジ日時 2024-10-01 03:10:30
合計ジャッジ時間 6,300 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 1 ms
6,816 KB
testcase_05 AC 2 ms
6,820 KB
testcase_06 AC 1 ms
6,816 KB
testcase_07 AC 1 ms
6,816 KB
testcase_08 AC 2 ms
6,816 KB
testcase_09 AC 2 ms
6,816 KB
testcase_10 AC 2 ms
6,816 KB
testcase_11 AC 2 ms
6,820 KB
testcase_12 AC 2 ms
6,820 KB
testcase_13 AC 2 ms
6,816 KB
testcase_14 AC 2 ms
6,820 KB
testcase_15 AC 41 ms
7,424 KB
testcase_16 AC 47 ms
7,936 KB
testcase_17 AC 29 ms
6,816 KB
testcase_18 AC 49 ms
7,808 KB
testcase_19 AC 37 ms
7,296 KB
testcase_20 AC 26 ms
6,820 KB
testcase_21 AC 46 ms
8,192 KB
testcase_22 AC 33 ms
6,820 KB
testcase_23 AC 31 ms
6,816 KB
testcase_24 AC 25 ms
6,820 KB
testcase_25 AC 47 ms
8,112 KB
testcase_26 AC 48 ms
8,064 KB
testcase_27 AC 48 ms
8,192 KB
testcase_28 AC 49 ms
8,076 KB
testcase_29 AC 48 ms
8,100 KB
testcase_30 AC 49 ms
8,192 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define all(x) x.begin(),x.end()
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << p.first << " " << p.second; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { string sep; for (const T &x : v) os << sep << x, sep = " "; return os; }
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) 42
#define ASSERT(...) 42
#endif
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int,int> pi;
const int oo = 1e9;
std::mt19937 rng(std::chrono::steady_clock::now().time_since_epoch().count());
template<class I> I rnd(I l,I r){return std::uniform_int_distribution<I>(l,r)(rng);}
void out(vi p) {
    int n = p.size()/3;
    for(int i=0;i<n;++i) swap(p[i],p[i+n]);
    cout << p << '\n';
    exit(0);
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n; cin >> n;
    vi p(n*3);
    iota(all(p),1);
    if(n<=6) {
        do {
            ll ans=0;
            for(int i=0;i<n;++i) {
                ans+=(ll)p[i]*p[i+n];
            }
            ll res=0;
            for(int i=0;i<n;++i) {
                res+=(ll)p[i]*p[i+2*n];
            }
            if(ans==res) {
                out(p);
            }

        } while(next_permutation(all(p)));
        cout << "-1\n";
        exit(0);
    }
    

    while(true) {
        int v = n+1;
        for(int i=n;i<2*n;++i) {
            p[i]=v;
            p[i+n]=v+1;
            v+=2;
        }
    const int k= 4;
    for(int i=0;i<k;++i) {
        if(rnd(0,1)) {
            swap(p[-i-1+n*3],p[-i-1+n*2]);
        }
    }
    shuffle(p.begin()+n*2-k,p.begin()+n*2,rng);
    shuffle(p.begin()+n*3-k,p.begin()+n*3,rng);

    ll ans=0;
    for(int i=0;i<n;++i) {
        ans+=(ll)p[i]*p[i+n];
    }
    ll res=0;
    for(int i=0;i<n;++i) {
        res+=(ll)p[i]*p[i+2*n];
    }
    ll need = res-ans;
    for(int i=2*n-k-1;i>=n;--i) {
        ll dif = 2 *(i+1-n);
        if(dif<=need) {
            need-=dif;

            swap(p[i],p[i+n]);
            assert(abs(p[i]-p[i+n])==1);
        }
    }
    ans=0;
    for(int i=0;i<n;++i) {
        ans+=(ll)p[i]*p[i+n];
    }
    res=0;
    for(int i=0;i<n;++i) {
        res+=(ll)p[i]*p[i+2*n];
    }
    debug(need,ans,res);
        if(need==0) {
            assert(ans==res);
            out(p);
        }
        
    }
}
0