#521963 (C++17) No.1141 田グリッド

提出ソース

結果

問題	No.1141 田グリッド
ユーザー	ganariya
提出日時	2020-07-31 22:07:05
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	201 ms / 2,000 ms
コード長	11,358 bytes
コンパイル時間	1,913 ms
コンパイル使用メモリ	162,996 KB
実行使用メモリ	14,208 KB
最終ジャッジ日時	2024-07-06 18:18:46
合計ジャッジ時間	6,979 ms
ジャッジサーバーID （参考情報）	judge2 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	1 ms 6,820 KB
testcase_02	AC	2 ms 6,816 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	1 ms 6,940 KB
testcase_05	AC	1 ms 6,944 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	1 ms 6,944 KB
testcase_08	AC	5 ms 6,940 KB
testcase_09	AC	4 ms 6,940 KB
testcase_10	AC	5 ms 6,944 KB
testcase_11	AC	6 ms 6,940 KB
testcase_12	AC	4 ms 6,944 KB
testcase_13	AC	182 ms 7,004 KB
testcase_14	AC	201 ms 14,208 KB
testcase_15	AC	186 ms 6,944 KB
testcase_16	AC	180 ms 6,940 KB
testcase_17	AC	177 ms 6,944 KB
testcase_18	AC	176 ms 6,940 KB
testcase_19	AC	182 ms 6,940 KB
testcase_20	AC	185 ms 6,944 KB
testcase_21	AC	183 ms 6,940 KB
testcase_22	AC	187 ms 6,940 KB
testcase_23	AC	189 ms 6,944 KB
testcase_24	AC	185 ms 6,940 KB
testcase_25	AC	184 ms 6,940 KB
testcase_26	AC	185 ms 6,944 KB
testcase_27	AC	185 ms 6,940 KB
testcase_28	AC	181 ms 6,940 KB
testcase_29	AC	179 ms 6,940 KB
testcase_30	AC	175 ms 6,940 KB
testcase_31	AC	177 ms 6,944 KB
testcase_32	AC	180 ms 6,944 KB
testcase_33	AC	178 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

//include
//------------------------------------------
#include <vector>
#include <list>
#include <map>
#include <unordered_map>
#include <climits>
#include <set>
#include <unordered_set>
#include <deque>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <ctime>
#include <queue>
#include <random>
#include <chrono>
#include <complex>
#include <regex>
#include <locale>
#include <random>
#include <cassert>
#include <type_traits>

using namespace std;

// typedef
//------------------------------------------
typedef long long LL;
typedef vector<int> VI;
typedef vector<bool> VB;
typedef vector<char> VC;
typedef vector<double> VD;
typedef vector<string> VS;
typedef vector<LL> VLL;
typedef vector<VI> VVI;
typedef vector<VB> VVB;
typedef vector<VS> VVS;
typedef vector<VLL> VVLL;
typedef vector<VVI> VVVI;
typedef vector<VVLL> VVVLL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
typedef pair<int, string> PIS;
typedef pair<string, int> PSI;
typedef pair<string, string> PSS;
typedef vector<PII> VPII;
typedef vector<PLL> VPLL;
typedef vector<VPII> VVPII;
typedef vector<VPLL> VVPLL;
typedef vector<VS> VVS;
typedef map<int, int> MII;
typedef map<LL, LL> MLL;
typedef map<string, int> MSI;
typedef map<int, string> MIS;

// container util
//------------------------------------------

typedef int pInt[4];
#define ALL(a)  (a).begin(),(a).end()
#define SZ(a) int((a).size())
#define EACH(i, arr) for(typeof((arr).begin()) i=(arr).begin(); i!=(arr).end(); ++i)
#define EXIST(str, e) ((str).find(e)!=(str).end())
#define COUNT(arr, v) count((arr).begin(), (arr).end(), v)
#define SEARCH(v, w) search((v).begin(), (v).end(), (w).begin(), (w).end())
#define SORT(c) sort((c).begin(),(c).end())
#define RSORT(c) sort((c).rbegin(),(c).rend())
#define REVERSE(c) reverse((c).begin(), (c).end())
#define ROTATE_LEFT(arr, c) rotate((arr).begin(), (arr).begin()+(c), (arr).end())
#define ROTATE_RIGHT(arr, c) rotate((arr).rbegin(), (arr).rbegin() + (c), (arr).rend())
#define SUMI(arr) accumulate((arr).begin(), (arr).end(), 0)
#define SUMD(arr) accumulate((arr).begin(), (arr).end(), 0.)
#define SUMLL(arr) accumulate((arr).begin(), (arr).end(), 0LL)
#define MULD(arr) accumulate((arr).begin(), (arr).end(), 1., multiplies<double>())
#define UB(arr, n) upper_bound((arr).begin(), (arr).end(), n)
#define LB(arr, n) lower_bound((arr).begin(), (arr).end(), n)
#define PB push_back
#define MP make_pair
#define ft first
#define sd second


// input output
//------------------------------------------
#define GL(s) getline(cin, (s))
#define INIT() std::ios::sync_with_stdio(false);std::cin.tie(0)
#define OUT(d) std::cout<<(d)
#define OUT_L(d) std::cout<<(d)<<endl
#define FOUT(n, data) std::cout<<std::fixed<<std::setprecision(n)<<(data)
#define FOUT_L(n, data) std::cout<<std::fixed<<std::setprecision(n)<<(data)<<"\n"
#define EL() printf("\n")
#define SHOW_VECTOR(v) {std::cerr << #v << "\t:";for(const auto& xxx : v){std::cerr << xxx << " ";}std::cerr << "\n";}
#define SHOW_MAP(v) {std::cerr << #v << endl; for(const auto& xxx: v){std::cerr << xxx.first << " " << xxx.second << "\n";}}
#define Yes() printf("Yes\n")
#define No() printf("No\n")
#define YES() printf("YES\n")
#define NO() printf("NO\n")
#define Yay() printf("Yay!\n")
#define Nnn() printf(":(\n")
#define CE(x, y) ((x + y - 1) / (y))


template<typename T1, typename T2>
istream &operator>>(istream &in, pair<T1, T2> &p) {
    in >> p.first >> p.second;
    return in;
}

template<typename T>
istream &operator>>(istream &in, vector<T> &v) {
    for (auto &x: v)
        in >> x;
    return in;
}


template<typename T1, typename T2>
ostream &operator<<(ostream &out, const std::pair<T1, T2> &p) {
    out << "[" << p.first << ", " << p.second << "]" << "\n";
    return out;
}

template<class T1, class T2>
inline bool chmin(T1 &a, T2 b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}

template<class T1, class T2>
inline bool chmax(T1 &a, T2 b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}


//repetition
//------------------------------------------
#define FOR(i, a, b) for(int i=(a);i<(b);++i)
#define RFOR(i, a, b) for(int i=(b)-1;i>=(a);--i)
#define REP(i, n)  FOR(i,0,n)
#define RREP(i, n) for(int i = n-1;i >= 0;i--)
#define FORLL(i, a, b) for(LL i=LL(a);i<LL(b);++i)
#define RFORLL(i, a, b) for(LL i=LL(b)-1;i>=LL(a);--i)
#define REPLL(i, n) for(LL i=0;i<LL(n);++i)
#define RREPLL(i, n) for(LL i=LL(n)-1;i>=0;--i)
#define FOREACH(x, arr) for(auto &(x) : (arr))

// Option + Control + K  Terminal
// Command + Control + K    Run
// Command + b   EOF



//------------------------------------------
//------------------------------------------

template<int_fast64_t Modulas = 1000000007ul>
class ModInt {
    using u64 = int_fast64_t;
public:
    u64 x;

    constexpr ModInt() : x(0) {}
    constexpr ModInt(int_fast64_t v) : x((v % Modulas + Modulas) % Modulas) {}
    constexpr ModInt operator+(const ModInt rhs) const noexcept {
        return ModInt(*this) += rhs;
    }
    constexpr ModInt operator-(const ModInt rhs) const noexcept {
        return ModInt(*this) -= rhs;
    }
    constexpr ModInt operator*(const ModInt rhs) const noexcept {
        return ModInt(*this) *= rhs;
    }
    constexpr ModInt operator/(const ModInt rhs) const noexcept {
        return ModInt(*this) /= rhs;
    }
    constexpr ModInt operator/(const long long rhs) const noexcept {
        return ModInt(*this) /= rhs;
    }
    constexpr ModInt operator+=(const ModInt rhs) noexcept {
        x += rhs.x;
        if (x >= Modulas) x -= Modulas;
        return *this;
    }
    constexpr ModInt operator+=(const long long rhs) noexcept {
        auto hs = ModInt<Modulas>(rhs);
        (*this) += hs;
        return *this;
    }
    constexpr ModInt operator-=(const ModInt rhs) noexcept {
        if (x < rhs.x) x += Modulas;
        x -= rhs.x;
        return *this;
    }
    constexpr ModInt operator-=(const long long rhs) noexcept {
        auto hs = ModInt<Modulas>(rhs);
        (*this) -= hs;
        return *this;
    }
    constexpr ModInt operator*=(const ModInt rhs) noexcept {
        x = x * rhs.x % Modulas;
        return *this;
    }
    constexpr ModInt operator*=(const long long rhs) noexcept {
        auto hs = ModInt<Modulas>(rhs);
        (*this) *= hs;
        return *this;
    }
    constexpr ModInt &operator/=(ModInt rhs) noexcept {
        u64 exp = Modulas - 2;
        while (exp > 0) {
            if (exp & 1ul) *this *= rhs;
            rhs *= rhs;
            exp >>= 1ul;
        }
        return *this;
    }
    constexpr ModInt &operator/=(long long rhs) noexcept {
        auto hs = ModInt<Modulas>(rhs);
        (*this) /= hs;
        return *this;
    }

    constexpr ModInt &operator++() noexcept {
        x++;
        if (x >= Modulas) x -= Modulas;
        return *this;
    }
    constexpr ModInt &operator--() noexcept {
        if (x == 0) x += Modulas;
        x--;
        return *this;
    }
    constexpr bool operator<(const ModInt rhs) const noexcept {
        return x < rhs.x;
    }
    constexpr bool operator==(const ModInt rhs) const noexcept {
        return this->x == rhs.x;
    }
    constexpr bool operator!=(const ModInt rhs) const noexcept {
        return !(*this == rhs);
    }
    friend istream &operator>>(istream &in, ModInt &m) {
        in >> m.x;
        if (m.x < 0) m.x += Modulas;
        m.x %= Modulas;
        return in;
    }
    friend ostream &operator<<(ostream &out, const ModInt &p) {
        out << p.x;
        return out;
    }
    constexpr ModInt pow(u64 p) const {
        ModInt ret(1);
        ModInt mul(x);
        while (p > 0) {
            if (p & 1ul) ret *= mul;
            mul *= mul;
            p >>= 1ul;
        }
        return ret;
    }
    constexpr ModInt operator~() const noexcept {
        u64 exp = Modulas - 2;
        return pow(exp);
    }

    constexpr static ModInt arith_sum(ModInt<Modulas> a, ModInt<Modulas> d, ModInt<Modulas> n) noexcept {
        return (a * ModInt<Modulas>(2) + (n - 1) * d) * n / ModInt<Modulas>(2);
    }
};

template<class T>
class CulSum2D {
private:
    //original
    int H, W;

    //original+1
    vector<vector<T>> data;

public:

    /// @param _H original
    /// @param _W original
    CulSum2D(int _H, int _W) :
            H(_H), W(_W) {
        reset();
    }

    void reset() {
        data = vector<vector<T>>(H + 1, vector<T>(W + 1, 1));
    }

    /// 0-index
    /// @param y original
    /// @param x original
    /// @param d original
    void add(int y, int x, T d) {
        y++;
        x++;
        if (y >= H + 1 || x >= W + 1) return;
        data[y][x] *= d;
    }

    void build() {
        for (int i = 0; i < H; i++) {
            for (int j = 0; j < W; j++) {
                data[i + 1][j + 1] *= data[i + 1][j] * data[i][j + 1] / data[i][j];
            }
        }
    }

    /// 0-index
    /// @param sy original close
    /// @param sx original close
    /// @param ty original open
    /// @param tx original open
    T get(int sy, int sx, int ty, int tx) {
        return data[ty][tx] / data[ty][sx] / data[sy][tx] * data[sy][sx];
    }

};

template<class T>
class CulSum2D_Add {
private:
    //original
    int H, W;

    //original+1
    vector<vector<T>> data;

public:

    /// @param _H original
    /// @param _W original
    CulSum2D_Add(int _H, int _W) :
            H(_H), W(_W) {
        reset();
    }

    void reset() {
        data = vector<vector<T>>(H + 1, vector<T>(W + 1, T{}));
    }

    /// 0-index
    /// @param y original
    /// @param x original
    /// @param d original
    void add(T y, T x, T d) {
        y++;
        x++;
        if (y >= H + 1 || x >= W + 1) return;
        data[y][x] += d;
    }

    void build() {
        for (int i = 0; i < H; i++) {
            for (int j = 0; j < W; j++) {
                data[i + 1][j + 1] += data[i + 1][j] + data[i][j + 1] - data[i][j];
            }
        }
    }

    /// 0-index
    /// @param sy original close
    /// @param sx original close
    /// @param ty original open
    /// @param tx original open
    T get(int sy, int sx, int ty, int tx) {
        return data[ty][tx] - data[ty][sx] - data[sy][tx] + data[sy][sx];
    }

};

int main() {

    LL H, W;
    cin >> H >> W;

    VVLL A(H, VLL(W));
    cin >> A;
    VVLL B = A;

    REP(i, H) REP(j, W) if (A[i][j] == 0) A[i][j] = 1;
    REP(i, H) REP(j, W) B[i][j] = (B[i][j] == 0);

    CulSum2D<ModInt<>> cs(H, W);
    REP(i, H) REP(j, W) cs.add(i, j, A[i][j]);
    cs.build();

    CulSum2D_Add<int> zero(H, W);
    REP(i, H) REP(j, W) zero.add(i, j, B[i][j]);
    zero.build();

    int Q;
    cin >> Q;

    while (Q--) {
        LL r, c;
        cin >> r >> c;
        r--, c--;

        ModInt<> ans = cs.get(0, 0, r, c) * cs.get(r + 1, 0, H, c) * cs.get(0, c + 1, r, W) * cs.get(r + 1, c + 1, H, W);

        int z = zero.get(0, 0, r, c) + zero.get(r + 1, 0, H, c) + zero.get(0, c + 1, r, W) + zero.get(r + 1, c + 1, H, W);
        if (z) ans = 0;

        cout << ans << endl;

    }

}

yukicoder

結果

テストケース

ソースコード