Submission #1313147

#TimeUsernameProblemLanguageResultExecution timeMemory
1313147joonwu04Mosaic (IOI24_mosaic)C++20
78 / 100
287 ms301924 KiB
#include "mosaic.h"
#include <vector>
#include <iostream>
#include <cassert>
using namespace std;

using ll = long long;

template <typename T>
class Array2D {
    int H = 0, W = 0;
    vector<T> data;

    public:
        Array2D() = default;

        Array2D(int h, int w): H(h), W(w), data(h * w) {}
        Array2D(int h, int w, T val): H(h), W(w), data(h * w, val) {}
        
        T& operator()(int i, int j) {
            assert(0 <= i && i < H  &&  0 <= j && j < W);
            return data[i * W + j];
        }
        const T& operator()(int i, int j) const {
            assert(0 <= i && i < H  &&  0 <= j && j < W);
            return data[i * W + j];
        }

        int height() const { return H; }
        int width() const { return W; }

        vector<T> row(int i) const {
            assert(0 <= i && i < H);

            vector<T> out;
            out.reserve(W);

            for (int j = 0; j < W; ++j) out.push_back(data[i * W + j]);
            
            return out;
        }

        vector<T> col(int j) const {
            assert(0 <= j && j < W);

            vector<T> out;
            out.reserve(H);

            for (int i = 0; i < H; ++i) out.push_back(data[i * W + j]);
            
            return out;
        }

        void print() const {
            for (int i = 0; i < H; ++i) {
                for (int j = 0; j < W; ++j) {
                    cout << (*this)(i, j) << " ";
                }
                cout << endl;
            }
        }

        void print1() const {
            for (int i = 1; i < H; ++i) {
                for (int j = 1; j < W; ++j) {
                    cout << (*this)(i, j) << " ";
                }
                cout << endl;
            }
        }
};

template <typename T, typename S = ll>
class PrefixSum1D {
    vector<S> dp;
        
    // dp: 1-indexed
    S sumDp(int l, int r) const {
        if (l > r) return 0;
        return dp[r] - dp[l-1];
    }

    public:
        PrefixSum1D() = default;

        PrefixSum1D(const vector<T>& arr) {   
            build(arr);
        }

        void build(const vector<T>& arr) {
            int n = (int)arr.size();
            dp = vector<S>(n + 1, (S)0);

            for (int i = 1; i <= n; ++i) {
                dp[i] = arr[i-1] + dp[i-1];
            }
        }

        // arr: 0-indexed
        S sum(int l, int r) const {
            return sumDp(l+1, r+1);
        }

        // void print() {
        //     print(dp);
        // }
};

template <typename T, typename S = ll>
class PrefixSum2D {
    Array2D<S> dp;
        
    // dp: 1-indexed
    S sumDp(int a, int b, int c, int d) const {
        return dp(c, d) - dp(c, b-1) - dp(a-1, d) + dp(a-1, b-1);
    }

    public:
        PrefixSum2D() = default;

        PrefixSum2D(const Array2D<T>& grid) {   
            build(grid);
        }

        void build(const Array2D<T>& grid) {   
            int h = grid.height(), w = grid.width();
            dp = Array2D<S>(h + 1, w + 1, (S)0);

            for (int i = 1; i <= h; ++i) {
                for (int j = 1; j <= w; ++j) {
                    dp(i, j) = (S)grid(i-1, j-1) + dp(i-1, j) + dp(i, j-1) - dp(i-1, j-1); 
                }
            }
        }

        // grid: 0-indexed
        S sum(int a, int b, int c, int d) const {
            return sumDp(a+1, b+1, c+1, d+1);
        }

        void print() const {
            dp.print1();
        }
};

struct Input {
    vector<int>& X;     vector<int>& Y; 
    vector<int>& T;     vector<int>& B; 
    vector<int>& L;     vector<int>& R;
};

namespace Subtask124 {
    
    void fillGrid(Array2D<int>& grid) {
        int n = (int)grid.height();
        for (int i = 1; i < n; i++) {
            for (int j = 1; j < n; j++) {
                grid(i, j) = !grid(i-1, j) && !grid(i, j-1);
            }
        }
    }

    vector<ll> solve(   vector<int>& X, vector<int>& Y,
                        vector<int>& T, vector<int>& B,
                        vector<int>& L, vector<int>& R) {

        int n = (int)X.size();
        int Q = (int)T.size();

        Array2D<int> grid(n, n, 0);

        for (int j = 0; j < n; j++) grid(0, j) = X[j];
        for (int i = 0; i < n; i++) grid(i, 0) = Y[i];

        fillGrid(grid);

        PrefixSum2D<int, ll> ps2d(grid);

        vector<ll> C(Q, 0LL);
        for (int k = 0; k < Q; k++) {
            int a = T[k], b = L[k], c = B[k], d = R[k];
            C[k] = ps2d.sum(a, b, c, d);
        }

        return C;
    }

    inline vector<ll> solve(Input& in) {
        return solve(in.X, in.Y, in.T, in.B, in.L, in.R);
    }
};

namespace Subtask3 {
    bool canApply(vector<int>& T, vector<int>& B) {
        int Q = (int)T.size();
        for (int k = 0; k < Q; ++k) {
            if (T[k] != 0  ||  B[k] != 0) return false;
        }
        return true;
    }
    inline bool canApply(Input& in) { return canApply(in.T, in.B); }

    vector<ll> solve(   vector<int>& X, vector<int>& Y,
                        vector<int>& T, vector<int>& B,
                        vector<int>& L, vector<int>& R) 
    {
        int Q = (int)T.size();
        
        PrefixSum1D<int, ll> ps1d(X);

        vector<ll> C(Q, 0LL);
        for (int k = 0; k < Q; k++) {
            int l = L[k], r = R[k];
            C[k] = ps1d.sum(l, r);
        }

        return C;
    }

    inline vector<ll> solve(Input& in) {
        return solve(in.X, in.Y, in.T, in.B, in.L, in.R);
    }
};

namespace Subtask5 {
    bool canApply(vector<int>& X, vector<int>& Y) {
        int n = (int)X.size();
        for (int i = 0; i < n; ++i) {
            if (X[i] != 0  ||  Y[i] != 0) return false;
        }
        return true;
    }
    inline bool canApply(Input& in) { return canApply(in.X, in.Y); }

    vector<ll> solve(   vector<int>& X, vector<int>& Y,
                        vector<int>& T, vector<int>& B,
                        vector<int>& L, vector<int>& R) 
    {
        int Q = (int)T.size();
        
        vector<ll> C(Q, 0LL);
        for (int k = 0; k < Q; k++) {
            int a = T[k], b = L[k], c = B[k], d = R[k];
            if (a == 0) a += 1;
            if (b == 0) b += 1;
            ll area = ((ll)c - a + 1) * (d - b + 1);

            C[k] = area / 2;
            if (area % 2 == 1  &&  (a + b) % 2 == 0) C[k] += 1;
        }

        return C;
    }

    inline vector<ll> solve(Input& in) {
        return solve(in.X, in.Y, in.T, in.B, in.L, in.R);
    }
};

Array2D<int> makeRows(vector<int>& X, vector<int>& Y) {
    int n = X.size();
    Array2D<int> rows(3, n);

    for (int j = 0; j < n; ++j) rows(0, j) = X[j];
    rows(1, 0) = Y[1];  
    rows(2, 0) = Y[2];
    for (int i = 1; i < 3; ++i) {
        for (int j = 1; j < n; ++j) {
            rows(i, j) = (!rows(i-1, j) && !rows(i, j-1));
        }
    }

    return rows;
}

Array2D<int> makeCols(vector<int>& X, vector<int>& Y) {
    int n = X.size();
    Array2D<int> cols(n, 3);

    for (int i = 0; i < n; ++i) cols(i, 0) = Y[i];
    cols(0, 1) = X[1];  
    cols(0, 2) = X[2];
    for (int i = 1; i < n; ++i) {
        for (int j = 1; j < 3; ++j) {
            cols(i, j) = (!cols(i-1, j) && !cols(i, j-1));
        }
    }

    return cols;
}

namespace Subtask6 {
    bool canApply(  vector<int>& T, vector<int>& B,
                    vector<int>& L, vector<int>& R) 
    {
        int Q = (int)T.size();

        for (int k = 0; k < Q; ++k) {
            if (T[k] != B[k]  ||  L[k] != R[k]) return false;
        }
        return true;
    }
    inline bool canApply(Input& in) { return canApply(in.T, in.B, in.L, in.R); }

    vector<ll> solve(   vector<int>& X, vector<int>& Y,
                        vector<int>& T, vector<int>& B,
                        vector<int>& L, vector<int>& R) 
    {
        int Q = (int)T.size();
    
        auto rows = makeRows(X, Y);
        auto cols = makeCols(X, Y);
        
        vector<ll> C(Q, 0LL);
        for (int k = 0; k < Q; k++) {
            int r = T[k], c = L[k];
            if (r <= 2) C[k] = rows(r, c);
            else if (c <= 2) C[k] = cols(r, c);
            else {
                if (r >= c) {
                    r -= c - 2;
                    c = 2;
                    C[k] = cols(r, c);
                }
                else {
                    c -= r - 2;
                    r = 2;
                    C[k] = rows(r, c);
                }
            }
        }
    
        return C;
    }

    inline vector<ll> solve(Input& in) {
        return solve(in.X, in.Y, in.T, in.B, in.L, in.R);
    }
};

namespace Subtask7 {
    bool canApply(vector<int>& T, vector<int>& B) 
    {
        int Q = (int)T.size();

        for (int k = 0; k < Q; ++k) {
            if (T[k] != B[k]) return false;
        }
        return true;
    }
    inline bool canApply(Input& in) { return canApply(in.T, in.B); }
    
    vector<ll> solve(   vector<int>& X, vector<int>& Y,
                        vector<int>& T, vector<int>& B,
                        vector<int>& L, vector<int>& R) 
    {
        int Q = (int)T.size();
    
        auto rows = makeRows(X, Y);
        auto cols = makeCols(X, Y);

        PrefixSum1D<int, ll> rowPs[3], colPs[3];
        for (int i = 0; i < 3; ++i) {
            rowPs[i] = PrefixSum1D<int, ll>(rows.row(i));
            colPs[i] = PrefixSum1D<int, ll>(cols.col(i));
        }
        
        vector<ll> C(Q, 0LL);
        for (int k = 0; k < Q; k++) {
            int i = T[k], l = L[k], r = R[k];

            if (i <= 2) C[k] = rowPs[i].sum(l, r);
            else {
                ll &ans = C[k];
                
                int s = l;
                int e = min(r, 2);
                for (int j = s; j <= e; ++j) ans += cols(i, j);
                
                s = max(l, 3);
                e = min(r, i);
                ans += colPs[2].sum(i - (e - 2), i - (s - 2));

                s = max(l, i+1);
                e = r;
                ans += rowPs[2].sum(s - (i - 2), e - (i - 2));                
            }
        }

        return C;
    }

    inline vector<ll> solve(Input& in) {
        return solve(in.X, in.Y, in.T, in.B, in.L, in.R);
    }
};

vector<ll> mosaic(vector<int> X, vector<int> Y,
                  vector<int> T, vector<int> B,
                  vector<int> L, vector<int> R) 
{
    int n = (int)X.size();
    int Q = (int)T.size();

    Input in{X, Y, T, B, L, R};

    if (n <= 5000)
        return Subtask124::solve(in);
    else if(Subtask3::canApply(in))
        return Subtask3::solve(in);
    else if(Subtask5::canApply(in))
        return Subtask5::solve(in);
    else if(Subtask6::canApply(in))
        return Subtask6::solve(in);
    else if(Subtask7::canApply(in))
        return Subtask7::solve(in);

    vector<ll> C(Q, 0ll);
    return C;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...