結果

問題 No.440 2次元チワワ問題
ユーザー t8m8⛄️t8m8⛄️
提出日時 2016-11-05 20:07:21
言語 Nim
(2.0.2)
結果
AC  
実行時間 760 ms / 5,000 ms
コード長 2,239 bytes
コンパイル時間 3,745 ms
コンパイル使用メモリ 67,556 KB
実行使用メモリ 20,992 KB
最終ジャッジ日時 2024-06-29 19:59:51
合計ジャッジ時間 10,235 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 3 ms
6,944 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 31 ms
7,936 KB
testcase_08 AC 72 ms
6,944 KB
testcase_09 AC 138 ms
9,344 KB
testcase_10 AC 26 ms
6,944 KB
testcase_11 AC 28 ms
6,940 KB
testcase_12 AC 50 ms
14,080 KB
testcase_13 AC 174 ms
20,864 KB
testcase_14 AC 6 ms
6,940 KB
testcase_15 AC 172 ms
20,992 KB
testcase_16 AC 23 ms
6,944 KB
testcase_17 AC 260 ms
14,720 KB
testcase_18 AC 272 ms
14,848 KB
testcase_19 AC 264 ms
14,720 KB
testcase_20 AC 273 ms
14,720 KB
testcase_21 AC 754 ms
14,720 KB
testcase_22 AC 744 ms
14,720 KB
testcase_23 AC 733 ms
14,976 KB
testcase_24 AC 760 ms
14,720 KB
testcase_25 AC 760 ms
14,848 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import strutils, sequtils

type
  TernaryKind = enum
    isTrue, isFalse

  TernaryOp[T] = ref object
    case kind*: TernaryKind
    of isTrue:  val: T
    of isFalse: nil

proc `\`[T](f: bool, x: T): TernaryOp[T] =
  if f:
    result =  TernaryOp[T](kind: isTrue, val: x)
  else:
    result = TernaryOp[T](kind: isFalse)

proc `\`[T](t: TernaryOp[T], y: T): T =
  case t.kind
  of isTrue:   result = t.val
  of isFalse:  result = y

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] + ((s[h-1][w-1] == 'c') \ 1 \ 0)
      cw[h][w] = cw[h][w-1] + ((s[h-1][w-1] == 'w') \ c[h][w-1] \ 0)
      cww[h][w] = cww[h][w-1] + ((s[h-1][w-1] == 'w') \ cw[h][w-1] \ 0)

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

  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