結果

問題 No.440 2次元チワワ問題
ユーザー pekempey
提出日時 2016-10-29 03:48:34
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 90 ms / 5,000 ms
コード長 1,760 bytes
コンパイル時間 1,329 ms
コンパイル使用メモリ 163,332 KB
実行使用メモリ 7,004 KB
最終ジャッジ日時 2024-11-24 07:07:43
合計ジャッジ時間 2,792 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:57:44: warning: format ‘%lld’ expects argument of type ‘long long int’, but argument 2 has type ‘__gnu_cxx::__alloc_traits<std::allocator<long int>, long int>::value_type’ {aka ‘long int’} [-Wformat=]
   57 |     for (int i = 0; i < q; i++) printf("%lld\n", ans[i]);
      |                                         ~~~^
      |                                            |
      |                                            long long int
      |                                         %ld
main.cpp:25:38: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   25 |     for (int i = 0; i < h; i++) scanf("%s", s[i]);
      |                                 ~~~~~^~~~~~~~~~~~
main.cpp:30:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   30 |         scanf("%hd %hd %hd %hd", &y1[i], &x1[i], &y2[i], &x2[i]);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

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

const int N = 500;

int h, w, q;
char s[N][N + 1];

struct DP { int c__, cw_, cww; };
DP dp[N][N + 1];

void calc() {
    for (int i = 0; i < h; i++) {
        memset(&dp[i][0], 0, sizeof(dp[i][0]));
        for (int j = 0; j < w; j++) {
            dp[i][j + 1].c__ = dp[i][j].c__ + (s[i][j] == 'c');
            dp[i][j + 1].cw_ = dp[i][j].cw_ + (s[i][j] == 'w') * dp[i][j].c__;
            dp[i][j + 1].cww = dp[i][j].cww + (s[i][j] == 'w') * dp[i][j].cw_;
        }
    }
}

int main() {
    cin >> h >> w;
    for (int i = 0; i < h; i++) scanf("%s", s[i]);
    cin >> q;

    vector<short> y1(q), x1(q), y2(q), x2(q);
    for (int i = 0; i < q; i++) {
        scanf("%hd %hd %hd %hd", &y1[i], &x1[i], &y2[i], &x2[i]);
        y1[i]--; x1[i]--; x2[i]--; y2[i]--;
    }

    vector<int64_t> ans(q);
    for (int yy = 0; yy < 2; yy++) {
        for (int xx = 0; xx < 2; xx++) {
            calc();
            for (int k = 0; k < q; k++) {
                short L = x1[k], R = x2[k] + 1;
                for (short i = y1[k]; i <= y2[k]; i++) {
                    int W = (R - L) - (dp[i][R].c__ - dp[i][L].c__);
                    ans[k] += dp[i][R].cww - dp[i][L].cww - dp[i][L].c__ * W * (W - 1) / 2 - dp[i][L].cw_ * W;
                }
            }
            for (int i = 0; i < h; i++) reverse(s[i], s[i] + w);
            for (int i = 0; i < q; i++) {
                x1[i] = w - 1 - x1[i];
                x2[i] = w - 1 - x2[i];
            }
            swap(x1, x2);
        }
        for (int i = 0; i < N; i++) for (int j = 0; j < i; j++) swap(s[i][j], s[j][i]);
        swap(h, w);
        swap(y1, x1);
        swap(y2, x2); 
    }
    for (int i = 0; i < q; i++) printf("%lld\n", ans[i]);
}
0