結果

問題 No.5024 魔法少女うなと宝集め
コンテスト
ユーザー Sqrt10
提出日時 2026-05-02 17:37:22
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 3,837 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,786 ms
コンパイル使用メモリ 343,236 KB
実行使用メモリ 6,400 KB
スコア 1,372,522
最終ジャッジ日時 2026-05-02 17:37:39
合計ジャッジ時間 6,899 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge1_1
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

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

int a[21][21] = {};
int s[22][22] = {};

int sum(int x1, int y1, int x2, int y2) { return s[x2][y2] - s[x1][y2] - s[x2][y1] + s[x1][y1]; }

int main() {
        int n, t;
        cin >> n >> t;
        for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                        cin >> a[i][j];
                }
        }

        for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                        s[i + 1][j + 1] = s[i + 1][j] + s[i][j + 1] - s[i][j] + a[i][j];
                }
        }

        double c[21][21] = {};
        const double alpha = .8;
        for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                        c[i][j] = 0;
                        for (int x = 0; x < n; x++) {
                                for (int y = 0; y < n; y++) {
                                        int d = abs(i - x) + abs(j - y);
                                        c[i][j] += a[x][y] * pow(alpha, d);
                                }
                        }
                }
        }
        // (1) Find the cell with the largest c
        int si = 0, sj = 0;
        double maxc = -1e18;
        for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                        if (c[i][j] > maxc) {
                                maxc = c[i][j];
                                si = i;
                                sj = j;
                        }
                }
        }

        vector<vector<bool>> visited(n, vector<bool>(n, false));
        int ci = si, cj = sj;
        visited[ci][cj] = true;
        vector<pair<int, int>> path;
        path.emplace_back(ci, cj);

        int moves = 1;
        int di[4] = {-1, 1, 0, 0}, dj[4] = {0, 0, -1, 1};
        while (moves < t) {
                int ni = -1, nj = -1;
                double best = -1e18;
                vector<pair<int, int>> candidates;
                for (int d = 0; d < 4; d++) {
                        int ti = ci + di[d], tj = cj + dj[d];
                        if (ti < 0 || ti >= n || tj < 0 || tj >= n) continue;
                        if (visited[ti][tj]) continue;
                        int adj_visited = 0;
                        for (int dd = 0; dd < 4; dd++) {
                                int ai = ti + di[dd], aj = tj + dj[dd];
                                if (ai < 0 || ai >= n || aj < 0 || aj >= n) continue;
                                if (visited[ai][aj]) adj_visited++;
                        }
                        if (adj_visited < 3) {
                                if (c[ti][tj] > best) {
                                        best = c[ti][tj];
                                        ni = ti;
                                        nj = tj;
                                }
                        } else {
                                candidates.emplace_back(ti, tj);
                        }
                }
                // If no "good" candidate, allow those with 3+ adjacent visited
                if (ni == -1 && !candidates.empty()) {
                        for (auto [ti, tj] : candidates) {
                                if (c[ti][tj] > best) {
                                        best = c[ti][tj];
                                        ni = ti;
                                        nj = tj;
                                }
                        }
                }
                if (ni == -1) break;
                ci = ni;
                cj = nj;
                visited[ci][cj] = true;
                path.emplace_back(ci, cj);
                moves++;
        }

        cout << path.size() << endl;
        for (auto [x, y] : path) {
                cout << x << ' ' << y << endl;
        }
}
0