oj.uz

Submission #754811

# Submission time Handle Problem Language Result Execution time Memory
754811 2023-06-08T16:04:33 Z Stickfish Costinland (info1cup19_costinland) C++17
67.8904 / 100
 1 ms 212 KB 
#include <iostream>
#include <vector>
#include <bitset>
using ll = long long;
using namespace std;

void solve_smallk(int k) {
    for (int m = 1; m < (1 << 16); m += 2) {
        vector<bitset<4>> v(4);
        for (int i = 0; i < 4; ++i) for (int j = 0; j < 4; ++j) {
            v[i][j] = m & (1 << (i * 4 + j));
        }
        vector<vector<pair<int, int>>> dp(5, vector<pair<int, int>>(5, {0, 0}));
        dp[0][0] = {1, 0};
        for (int i = 0; i < 4; ++i) for (int j = 0; j < 4; ++j) {
            if (v[i][j]) {
                dp[i + 1][j].first += dp[i][j].first + dp[i][j].second;
                dp[i][j + 1].second += dp[i][j].first + dp[i][j].second;
            } else {
                dp[i + 1][j].first += dp[i][j].first;
                dp[i][j + 1].second += dp[i][j].second;
            }
        }
        int ans = 0;
        for (int i = 0; i < 5; ++i) {
            ans += dp[4][i].first + dp[i][4].second;
        }
        if (ans == k) {
            cout << "5 5\n";
            for (int i = 0; i < 4; ++i) {
                for (int j = 0; j < 4; ++j) {
                    cout << (v[i][j] ? 'X' : '.');
                }
                cout << "d\n";
            }
            cout << "rrrr.\n";
            return;
        }
    }
    cout << "-1\n";
}

struct cell {
    int i;
    int j;
    int t;

    cell() {}

    cell(int i_, int j_, int t_): i(i_), j(j_), t(t_) {}

    cell reverse() {
        return {j, i, (3 - t) % 3};
    }
};

struct block {
    int n;
    int m;
    ll d;
    ll r;
    vector<cell> cells;

    block() {};

    block(int n_, int m_, ll d_, ll r_, vector<cell> cells_): n(n_), m(m_), d(d_), r(r_), cells(cells_) {}

    block(int n_, int m_, ll d_, ll r_, string s): n(n_), m(m_), d(d_), r(r_) {
        for (int i = 0; i <= n; ++i) for (int j = 0; j <= m; ++j) {
            char c = s[i * (m + 1) + j];
            if (c == 'X')
                cells.push_back({i, j, 0});
            else if (c == 'd')
                cells.push_back({i, j, 1});
            else if (c == 'r')
                cells.push_back({i, j, 2});
        }
    }
};

vector<block> blocks;

vector<cell> solve_recurs(ll k, int n, int m) {
    if (n <= 0 || m <= 0)
        return {};
    if (k == 1)
        return {{0, 0, 1}};
    if (n > m) {
        vector<cell> ans = solve_recurs(k, m , n);
        for (auto& x : ans)
            x = x.reverse();
        return ans;
    }

    for (auto b : blocks) {
        if (k < b.r || (k - b.r) % b.d || n < b.n || m < b.m)
            continue;
        vector<cell> ans = solve_recurs((k - b.r) / b.d, n - b.n, m - b.m);
        if (ans.empty())
            return {};
        for (auto& x : ans)
            x.i += b.n, x.j += b.m;
        for (auto x : b.cells)
            ans.push_back(x);
        return ans;
    }

    vector<cell> ans = solve_recurs(k - 1, n, m - 1);
    if (ans.empty())
        return {};
    for (auto& x : ans)
        ++x.j;
    ans.push_back({0, 0 ,0});
    return ans;
}

void init_blocks() {
    blocks.push_back({
        1, 2, 3, 1,
        "XXdXr."
    });
    //blocks.push_back({
        //2, 2, 4,
        //""
    //});
    blocks.push_back({
        1, 2, 3, 0,
        "XXdrr."
    });
    blocks.push_back({
        1, 1, 2, 0,
        "Xdr."
    });
}

signed main() {
    ll k;
    cin >> k;
    if (k <= 19) {
        solve_smallk(k);
        return 0;
    }
    init_blocks();
    vector<char> symbols = {'.', 'X', 'd', 'r'};
    for (int n = 1; n <= 200; ++n) {
        vector<cell> ans = solve_recurs(k, n - 1, n - 1);
        if (ans.empty())
            continue;
        vector<vector<int>> v(n, vector<int>(n, -1));
        for (int i = 0; i < n - 1; ++i)
            v[i][n - 1] = 1, v[n - 1][i] = 2;
        for (auto x : ans) if (x.t != -6)
            v[x.i][x.j] = x.t;
        
        cout << n << ' ' << n << '\n';
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < n; ++j) {
                cout << symbols[v[i][j] + 1];
            }
            cout << '\n';
        }
        return 0;
    }
}

Subtask #1
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Correct! Your size: 5
2 Correct 0 ms 212 KB Correct! Your size: 5
3 Correct 0 ms 212 KB Correct! Your size: 5
4 Correct 0 ms 212 KB Correct! Your size: 5
5 Correct 0 ms 212 KB Correct! Your size: 5
6 Correct 1 ms 212 KB Correct! Your size: 5
7 Correct 0 ms 212 KB Correct! Your size: 5
8 Correct 0 ms 212 KB Correct! Your size: 5
9 Correct 1 ms 212 KB Correct! Your size: 5

Subtask #2
47.89041096 / 80.0

# Verdict Execution time Memory Grader output
1 Partially correct 1 ms 212 KB Partially Correct! Your size: 58
2 Partially correct 1 ms 212 KB Partially Correct! Your size: 63
3 Partially correct 1 ms 212 KB Partially Correct! Your size: 60
4 Partially correct 1 ms 212 KB Partially Correct! Your size: 61
5 Partially correct 1 ms 212 KB Partially Correct! Your size: 61
6 Partially correct 1 ms 212 KB Partially Correct! Your size: 62
7 Partially correct 1 ms 212 KB Partially Correct! Your size: 62
8 Partially correct 1 ms 212 KB Partially Correct! Your size: 58