結果

問題 No.440 2次元チワワ問題
ユーザー 🍡yurahuna
提出日時 2016-04-26 17:05:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,843 bytes
コンパイル時間 1,720 ms
コンパイル使用メモリ 173,392 KB
実行使用メモリ 13,632 KB
最終ジャッジ日時 2024-10-11 00:04:29
合計ジャッジ時間 16,133 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define rep(i,n) for (int i=0;i<(n);i++)
#define rep2(i,a,b) for (int i=(a);i<(b);i++)
#define rrep(i,n) for (int i=(n)-1;i>=0;i--)
#define rrep2(i,a,b) for (int i=(b)-1;i>=(a);i--)
#define all(a) (a).begin(),(a).end()

typedef long long ll;
typedef pair<int, int> P;

// 累積和解
// O( Q H W )
// ！！！modはさぼってます！！！

const ll mod = 1e9 + 7;

const ll MAX_H = 505;
const ll MAX_W = 505;

/************ 補助関数 ****************/

// // mod演算の関数
// ll ADD(ll a, ll b) { return (a + b) % mod; }
// ll SUB(ll a, ll b) { return (mod + a - b) % mod; }
// ll MUL(ll a, ll b) { return (a * b) % mod; }

ll nC2(ll n) {
    return n * (n - 1) / 2;
}

// // 点{x, y}が 長方形領域 [ {a, b}, {c, d} ] に含まれるならtrueを返す
// // ！！！長方形は閉区間！！！
// bool inside(ll x, ll y, ll a, ll b, ll c, ll d) {
//     return a <= x && x <= c && b <= y && y <= d;
// }

/*********** 解答ここから *************/

ll H, W;
vector<string> s;

// wの個数の2次元累積和
ll w[MAX_H][MAX_W];

int dx[4] = {1, 0, -1, 0};
int dy[4] = {0, 1, 0, -1};

// 長方形領域 [ {a, b}, {c, d} ) における w の個数を返す
// 実際は 'c' から見た各方向へ伸びる線上の w の個数を求めるのに使われている
ll getW(ll a, ll b, ll c, ll d) {
    return w[a][b] - w[a][d] - w[c][b] + w[c][d];
}

// 点 {x, y} を起点としたときの、長方形領域 [ {a, b}, {c, d} ) におけるcwwの数を返す
ll CWW4(ll x, ll y, ll a, ll b, ll c, ll d) {
    ll ret = 0;
    ret += nC2(getW(x, y, x + 1, d));  // wwの個数 = wから2個選ぶ場合の数
    ret += nC2(getW(x, y, c, y + 1));
    ret += nC2(getW(a, y, x + 1, y + 1));
    ret += nC2(getW(x, b, x + 1, y + 1));
    return ret;
}

// 長方形領域 [ {a, b}, {c, d} ) におけるcwwの数を返す
ll solve(ll a, ll b, ll c, ll d) {
    ll ret = 0;
    rep2(i, a, c) {
        rep2(j, b, d) {
            if (s[i][j] == 'c') {
                // この'c'を起点とした4方向のcwwの数を足し上げる
                ret += CWW4(i, j, a, b, c, d);
            }
        }
    }
    return ret;
}


signed main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);

    cin >> H >> W;
    s.resize(H);
    rep(i, H) cin >> s[i];

    /************** 前処理 ****************/

    rrep(i, H) {
        rrep(j, W) {
            w[i][j] += w[i][j + 1];
            w[i][j] += (s[i][j] == 'w');
        }
    }

    rrep(j, W) {
        rrep(i, H) {
            w[i][j] += w[i + 1][j];
        }
    }
    
    /*********** クエリへの応答 *************/

    ll Q;
    cin >> Q;
    rep(i, Q) {
        ll a, b, c, d;
        cin >> a >> b >> c >> d;
        a--; b--;
        cout << solve(a, b, c, d) << endl;
    }
}
0