結果

問題 No.440 2次元チワワ問題
ユーザー t8m8⛄️t8m8⛄️
提出日時 2016-11-05 20:08:56
言語 Nim
(2.0.2)
結果
AC  
実行時間 707 ms / 5,000 ms
コード長 1,897 bytes
コンパイル時間 4,251 ms
コンパイル使用メモリ 67,456 KB
実行使用メモリ 20,992 KB
最終ジャッジ日時 2024-06-29 20:02:39
合計ジャッジ時間 9,769 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 25 ms
7,936 KB
testcase_08 AC 69 ms
6,144 KB
testcase_09 AC 110 ms
9,344 KB
testcase_10 AC 20 ms
5,376 KB
testcase_11 AC 29 ms
5,376 KB
testcase_12 AC 35 ms
14,080 KB
testcase_13 AC 143 ms
20,864 KB
testcase_14 AC 6 ms
5,376 KB
testcase_15 AC 135 ms
20,992 KB
testcase_16 AC 24 ms
5,376 KB
testcase_17 AC 230 ms
14,720 KB
testcase_18 AC 218 ms
14,720 KB
testcase_19 AC 227 ms
14,720 KB
testcase_20 AC 228 ms
14,848 KB
testcase_21 AC 703 ms
14,720 KB
testcase_22 AC 700 ms
14,720 KB
testcase_23 AC 705 ms
14,720 KB
testcase_24 AC 705 ms
14,720 KB
testcase_25 AC 707 ms
14,720 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

{.optimization: speed.}
import strutils, sequtils

proc calc(H, W, Q: int, s: seq[seq[char]], q: seq[(int, int, int, int)], ans: var seq[int64]) =
  var
    c = newSeqWith(H+1, newSeq[int64](W+2))
    cw = newSeqWith(H+1, newSeq[int64](W+2))
    cww = newSeqWith(H+1, newSeq[int64](W+2))
    wc = newSeqWith(H+1, newSeq[int64](W+2))
    wwc = newSeqWith(H+1, newSeq[int64](W+2))

  for h in 1..H:
    for w in 1..W:
      c[h][w] += c[h][w-1]
      cw[h][w] += cw[h][w-1]
      cww[h][w] += cww[h][w-1]

      if s[h-1][w-1] == 'c':
        c[h][w] += 1
      if s[h-1][w-1] == 'w':
        cw[h][w] += c[h][w-1]
        cww[h][w] += cw[h][w-1]

    var rc = 0
    for w in countDown(W, 1):
      wc[h][w] += wc[h][w+1]
      wwc[h][w] += wwc[h][w+1]

      if s[h-1][w-1] == 'w':
        wc[h][w] += rc
        wwc[h][w] += wc[h][w+1]
      if s[h-1][w-1] == 'c':
        rc += 1

  for k in 0..Q-1:
    var (y1, x1, y2, x2) = q[k]
    for i in y1..y2:
      var
        ccnt = c[i][x2] - c[i][x1-1]
        wcnt = x2 - x1 + 1 - ccnt
      ans[k] += cww[i][x2] - cww[i][x1-1]
      ans[k] += wwc[i][x1] - wwc[i][x2+1]
      ans[k] -= (c[i][W] - ccnt)*wcnt*(wcnt-1) div 2
      ans[k] -= wcnt*(cw[i][x1-1] + wc[i][x2+1])


when isMainModule:
  var
    tmp = stdin.readline.split.map(parseint)
    (H, W) = (tmp[0], tmp[1])
    s = newSeqWith(H, newSeq[char](W))
    s2 = newSeqWith(W, newSeq[char](H))

  for i in 0..H-1:
    var t = stdin.readline
    for j in 0..W-1:
      s[i][j] = t[j]
      s2[j][i] = t[j]

  var
    Q = stdin.readline.parseint
    q: seq[(int, int, int, int)] = @[]
    q2: seq[(int, int, int, int)] = @[]

  for i in 0..Q-1:
    var tmp = stdin.readline.split.map(parseint)
    q.add((tmp[0], tmp[1], tmp[2], tmp[3]))
    q2.add((tmp[1], tmp[0], tmp[3], tmp[2]))

  var ans = newSeq[int64](Q)
  calc(H, W, Q, s, q, ans)
  calc(W, H, Q, s2, q2, ans)
  for i in ans: i.echo
0