結果

問題 No.5013 セクスタプル (open)
コンテスト
ユーザー wanui
提出日時 2022-12-29 15:10:18
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,966 ms / 2,000 ms
コード長 6,313 bytes
コンパイル時間 2,234 ms
実行使用メモリ 5,164 KB
スコア 18,408
最終ジャッジ日時 2022-12-29 15:13:44
合計ジャッジ時間 202,255 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 100
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
// clang-format off
using namespace std;
void print0(){}; template<typename H,typename... T> void print0(H h,T... t){cout<<h;print0(t...);}
void print(){print0("\n");}; template<typename H,typename... T>void print(H h,T... t){print0(h);if(sizeof...(T)>0)print0(" ");print(t...);}
#define debug1(a) { cerr<<#a<<":"<<a<<endl; }
#define debug2(a,b) { cerr<<#a<<":"<<a<<" "<<#b<<":"<<b<<endl; }
#define debug3(a,b,c) { cerr<<#a<<":"<<a<<" "<<#b<<":"<<b<<" "<<#c<<":"<<c<<endl; }
#define debug4(a,b,c,d) { cerr<<#a<<":"<<a<<" "<<#b<<":"<<b<<" "<<#c<<":"<<c<<" "<<#d<<":"<<d<<endl; }
// clang-format on

namespace marathon {
mt19937 engine(0);
clock_t start_time;
double now() {
    return 1000.0 * (clock() - start_time) / CLOCKS_PER_SEC;
}
void marathon_init() {
    start_time = clock();
    random_device seed_gen;
    engine.seed(seed_gen());
}
int randint(int mn, int mx) {
    int rng = mx - mn + 1;
    return mn + (engine() % rng);
}
bool anneal_accept(double new_score, double old_score, double cur_time, double begin_time, double end_time, double begin_temp, double end_temp) {
    const int ANNEAL_RND = 1e8;
    const double ANNEAL_EPS = 1e-6;
    double temp = (begin_temp * (end_time - cur_time) + end_temp * (cur_time - begin_time)) / (end_time - begin_time);
    return (exp((new_score - old_score) / temp) > double(engine() % ANNEAL_RND) / ANNEAL_RND + ANNEAL_EPS);
}
}  // namespace marathon

const int N2 = 36;
const int N = 6;
const int SCORE_BASE = 3;
const double END_TIME = 1900;

struct six {
    int val[6];
};

int calc_score(vector<vector<int>> &perm, vector<six> &rolls) {
    int score = 0;
    for (int i = 0; i < N; i++) {
        for (int d = 0; d < N; d++) {
            int cnt = 0;
            for (int j = 0; j < N; j++) {
                int rollid = perm[i][j];
                int roll_cnt = rolls[rollid].val[d];
                if (!roll_cnt) {
                    cnt = 0;
                    break;
                }
                cnt += roll_cnt;
            }
            if (!cnt) continue;
            score += cnt - N + SCORE_BASE;
        }
    }
    for (int j = 0; j < N; j++) {
        for (int d = 0; d < N; d++) {
            int cnt = 0;
            for (int i = 0; i < N; i++) {
                int rollid = perm[i][j];
                int roll_cnt = rolls[rollid].val[d];
                if (!roll_cnt) {
                    cnt = 0;
                    break;
                }
                cnt += roll_cnt;
            }
            if (!cnt) continue;
            score += cnt - N + SCORE_BASE;
        }
    }
    return score;
}
int col_score(vector<vector<int>> &perm, vector<six> &rolls, int xj) {
    int score = 0;
    {
        for (int d = 0; d < N; d++) {
            int cnt = 0;
            for (int i = 0; i < N; i++) {
                int rollid = perm[i][xj];
                int roll_cnt = rolls[rollid].val[d];
                if (!roll_cnt) {
                    cnt = 0;
                    break;
                }
                cnt += roll_cnt;
            }
            if (!cnt) continue;
            score += cnt - N + SCORE_BASE;
        }
    }
    return score;
}
int row_score(vector<vector<int>> &perm, vector<six> &rolls, int xi) {
    int score = 0;
    {
        for (int d = 0; d < N; d++) {
            int cnt = 0;
            for (int j = 0; j < N; j++) {
                int rollid = perm[xi][j];
                int roll_cnt = rolls[rollid].val[d];
                if (!roll_cnt) {
                    cnt = 0;
                    break;
                }
                cnt += roll_cnt;
            }
            if (!cnt) continue;
            score += cnt - N + SCORE_BASE;
        }
    }
    return score;
}
void anneal(vector<six> rolls) {
    vector<vector<int>> perm(N, vector<int>(N));
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            perm[i][j] = i * N + j;
        }
    }
    int old_score = calc_score(perm, rolls);

    vector<vector<int>> best_perm = perm;
    int best_score = old_score;

    double begin_time = marathon::now();
    double end_time = END_TIME;
    double begin_temp = 2.0;
    double end_temp = begin_temp * 0.1;

    int anneal_iter = 0;
    int anneal_accept = 0;
    while (marathon::now() < END_TIME) {
        anneal_iter++;
        int x = 0;
        int y = 0;
        while (x == y) {
            x = marathon::engine() % N2;
            y = marathon::engine() % N2;
        }
        int xi = x / N;
        int xj = x % N;
        int yi = y / N;
        int yj = y % N;

        int new_score = old_score;
        if (xi != yi) {
            new_score -= row_score(perm, rolls, xi);
            new_score -= row_score(perm, rolls, yi);
        }
        if (xj != yj) {
            new_score -= col_score(perm, rolls, xj);
            new_score -= col_score(perm, rolls, yj);
        }
        swap(perm[xi][xj], perm[yi][yj]);
        if (xi != yi) {
            new_score += row_score(perm, rolls, xi);
            new_score += row_score(perm, rolls, yi);
        }
        if (xj != yj) {
            new_score += col_score(perm, rolls, xj);
            new_score += col_score(perm, rolls, yj);
        }

        if (best_score < new_score) {
            best_perm = perm;
            best_score = new_score;
        }
        if (marathon::anneal_accept(new_score, old_score, marathon::now(), begin_time, end_time, begin_temp, end_temp)) {
            anneal_accept++;
            old_score = new_score;
        } else {
            swap(perm[xi][xj], perm[yi][yj]);
        }
    }
    debug3(anneal_iter, anneal_accept, best_score);

    vector<pair<int, int>> revperm(N2);
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            revperm[best_perm[i][j]] = {i, j};
        }
    }
    for (int i = 0; i < N2; i++) {
        print(revperm[i].first + 1, revperm[i].second + 1);
    }
}

int main() {
    marathon::marathon_init();
    vector<six> rolls(N2);
    for (int i = 0; i < N2; i++) {
        six s;
        for (int k = 0; k < N; k++) {
            s.val[k] = 0;
        }
        for (int j = 0; j < N; j++) {
            int d;
            cin >> d;
            d--;
            s.val[d]++;
        }
        rolls[i] = s;
    }
    anneal(rolls);
    return 0;
}
0