AtCoder Beginner Contest Edit

【pythonでABC203を解説】B - AtCoder Condominium

問題概要

問題ページ

問題文

AtCoder マンションは \(1\) 階から \(N\) 階までの \(N\) 階建てのマンションです。
各階には \(K\) 室の部屋があり、\(1\) 号室から \(K\) 号室まで番号が振られています。

ここで \(N,K\) は \(1\) 桁の整数であり、 \(i\) 階の \(j\) 号室の部屋番号は i0j で表されます。
例えば、\(1\) 階の \(2\) 号室の部屋番号は \(102\) です。

マンションの管理人である高橋君は各部屋番号を \(3\) 桁の整数とみなし、
AtCoder マンションに存在するすべての部屋について足しあわせたらいくつになるのか興味を持ちました。
その値を求めてください。

制約

  • \(1 \leq N,K \leq 9\)
  • \(N,K\) は整数である。

問題の考察

ACコード

import sys


def solve():
    input = sys.stdin.readline
    mod = 10 ** 9 + 7
    n, k = list(map(int, input().rstrip('\n').split()))
    print(100 * (1 + n) * n // 2 * k + n * (1 + k) * k // 2)


if __name__ == '__main__':
    solve()

プログラミング

-AtCoder Beginner Contest, Edit
-