結果
問題 | No.346 チワワ数え上げ問題 |
ユーザー | yomo3 |
提出日時 | 2020-02-29 23:52:42 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 168 ms / 2,000 ms |
コード長 | 824 bytes |
コンパイル時間 | 327 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 14,848 KB |
最終ジャッジ日時 | 2024-10-13 20:22:25 |
合計ジャッジ時間 | 2,537 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
from functools import reduce def comb(n, r): if n < 0 or r < 0 or r > n: return 0 r = min(r, n - r) if r == 0: return 1 numer = [n - r + i + 1 for i in range(r)] denom = [i + 1 for i in range(r)] for i in range(1, r): divisor = denom[i] if divisor > 1: offset = (n - r) % (i + 1) for j in range(i, r, i+1): numer[j - offset] //= divisor denom[j] //= divisor return reduce(lambda x, y: x*y if y > 1 else x, numer) def main(): S = input() N = len(S) cnt_w = [0] * N for i in range(N-1, -1, -1): cnt_w[i] = (0 if i == N-1 else cnt_w[i + 1]) + (1 if S[i] == 'w' else 0) ans = 0 for i in range(N-1): if S[i] == 'c': ans += comb(cnt_w[i+1], 2) print(ans) main()