結果

問題 No.440 2次元チワワ問題
ユーザー t8m8⛄️t8m8⛄️
提出日時 2016-11-05 18:25:30
言語 Nim
(2.0.0)
結果
AC  
実行時間 943 ms / 5,000 ms
コード長 1,874 bytes
コンパイル時間 4,131 ms
コンパイル使用メモリ 68,964 KB
実行使用メモリ 21,064 KB
最終ジャッジ日時 2023-09-12 07:00:24
合計ジャッジ時間 12,527 ms
ジャッジサーバーID
(参考情報)
judge12 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 31 ms
8,060 KB
testcase_08 AC 88 ms
8,120 KB
testcase_09 AC 154 ms
11,772 KB
testcase_10 AC 26 ms
6,628 KB
testcase_11 AC 37 ms
4,752 KB
testcase_12 AC 45 ms
14,112 KB
testcase_13 AC 196 ms
20,948 KB
testcase_14 AC 7 ms
4,380 KB
testcase_15 AC 187 ms
21,064 KB
testcase_16 AC 27 ms
5,348 KB
testcase_17 AC 330 ms
20,064 KB
testcase_18 AC 319 ms
19,908 KB
testcase_19 AC 319 ms
19,920 KB
testcase_20 AC 307 ms
19,876 KB
testcase_21 AC 941 ms
19,856 KB
testcase_22 AC 932 ms
20,068 KB
testcase_23 AC 920 ms
19,864 KB
testcase_24 AC 943 ms
19,856 KB
testcase_25 AC 924 ms
19,920 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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