結果

問題 No.440 2次元チワワ問題
ユーザー t8m8⛄️t8m8⛄️
提出日時 2016-11-05 20:07:21
言語 Nim
(2.0.2)
結果
AC  
実行時間 1,000 ms / 5,000 ms
コード長 2,239 bytes
コンパイル時間 3,692 ms
コンパイル使用メモリ 70,184 KB
実行使用メモリ 21,220 KB
最終ジャッジ日時 2023-09-12 07:00:38
合計ジャッジ時間 12,192 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,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 3 ms
4,380 KB
testcase_05 AC 2 ms
4,504 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 41 ms
7,904 KB
testcase_08 AC 97 ms
8,000 KB
testcase_09 AC 181 ms
12,044 KB
testcase_10 AC 35 ms
6,572 KB
testcase_11 AC 40 ms
4,708 KB
testcase_12 AC 64 ms
14,324 KB
testcase_13 AC 234 ms
20,936 KB
testcase_14 AC 7 ms
4,380 KB
testcase_15 AC 223 ms
21,220 KB
testcase_16 AC 26 ms
5,204 KB
testcase_17 AC 376 ms
20,008 KB
testcase_18 AC 415 ms
20,068 KB
testcase_19 AC 385 ms
20,068 KB
testcase_20 AC 381 ms
20,068 KB
testcase_21 AC 996 ms
19,848 KB
testcase_22 AC 997 ms
19,864 KB
testcase_23 AC 997 ms
19,844 KB
testcase_24 AC 1,000 ms
19,848 KB
testcase_25 AC 999 ms
19,920 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