結果

問題 No.440 2次元チワワ問題
ユーザー 🍡yurahuna
提出日時 2016-04-28 09:03:25
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,412 bytes
コンパイル時間 1,577 ms
コンパイル使用メモリ 168,716 KB
実行使用メモリ 12,524 KB
最終ジャッジ日時 2024-10-04 16:05:59
合計ジャッジ時間 15,576 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 )
// 入出力をprintf, scanfにした
// vector<string>を char[][]にした

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

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

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

const ll MAX_H = 505;
const ll MAX_W = 505;
ll H, W;
char s[MAX_H][MAX_W];

// 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() {
    scanf("%lld %lld\n", &H, &W);
    rep(i, H) scanf("%s\n", 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;
    scanf("%lld\n", &Q);
    rep(i, Q) {
        ll a, b, c, d;
        scanf("%lld %lld %lld %lld\n", &a, &b, &c, &d);
        a--; b--;
        printf("%lld\n", solve(a, b, c, d));
    }
}
0