pontz_rwのブログ

プログラミング等の備忘録

ITP1_7_A: Grading

成績の判定 | プログラミング入門 | Aizu Online Judge
複数の条件分岐がある問題

条件分岐の順番に気を付ければ溶ける問題です。


解答

while True:
    m, f, r = map(int, input().split())
    score = m + f
    if m == f == r == -1:
        break
    elif m == -1 or f == -1:
        print('F')
    elif score >= 80:
        print('A')
    elif score >= 65:
        print('B')
    elif score >= 50 or (score >= 30 and r >= 50):
        print('C')
    elif score >= 30:
        print('D')
    else:
        print('F')

ITP1_6_D: Matrix Vector Multiplication

行列とベクトルの積 | プログラミング入門 | Aizu Online Judge
ベクトルと行列の積についての問題

 c_i = \sum_{j=1}^m a_{ij}b_j  = a_{i1}b_1 + a_{i2}b_2 + \dots + a_{im}b_m
この式に沿って問題を解いていきます。


解答1

n, m = map(int, input().split())

A = []
b = []

# 行列Aに要素を追加
for i in range(n):
    A.append([int(s) for s in input().split()])

# 列ベクトルbに要素を追加
for i in range(m):
    b.append([int(input())])

# Aとbの積を出力
for i in range(n):
    sum = 0
    # 列ベクトルc_iを求める
    for j in range(m):
        sum_i += A[i][j] * b[j][0]
    # 列ベクトルc_iを出力
    print(sum_i)

解答2
リスト内包表記を利用し、簡潔にしています。

n, m = list(map(int, input().split()))

A = [list(map(int, input().split())) for i in range(n)]
bt = [int(input()) for i in range(m)]

for i in range(n):
    print(sum([x * y for (x, y) in zip(A[i], bt)]))

ITP1_6_C: Official House

公舎 | プログラミング入門 | Aizu Online Judge
配列に関する問題

多次元配列を利用します。


解答

houses = [[[0 for r in range(10)] for f in range(3)] for b in range(4)]

n = int(input())

for i in range(n):
    b, f, r, v = map(int, input().split())
    houses[b - 1][f - 1][r - 1] += v

for b in range(4):
    for f in houses[b]:
        print('', *f)
    if b != 3:
        print('#' * 20)

ITP1_6_B: Finding Missing Cards

なくなったカードの発見 | プログラミング入門 | Aizu Online Judge
不足しているカードを指定された順序で表示する問題

トランプのリストを用意しておき、指定されたカードをフラグで管理するか、取り除くかします。


解答1

# トランプの生成
cards = [
    "{} {}".format(s, r) for s in ('S', 'H', 'C', 'D') for r in range(1, 13 + 1)
]

count = int(input())

# トランプの削除
for n in range(count):
    card = input()
    cards.remove(card)

# 不足しているトランプの表示
for n in cards:
    print(n)

解答2

from collections import OrderedDict

# トランプの生成
cards = OrderedDict()  # 順序付き辞書
for s in ['S', 'H', 'C', 'D']:
    for r in range(1, 13 + 1):
        key = "{} {}".format(s, r)
        cards[key] = False

count = int(input())

# 所持しているトランプの処理
for n in range(count):
    input_key = input()
    cards[input_key] = not(cards[input_key])  # 真理値を反転 (Trueへ)

# 不足しているトランプの表示
for k, v in cards.items():
    if not v:
        print(k)

ITP1_6_A: Reversing Numbers

数列の反転 | プログラミング入門 | Aizu Online Judge
リストを逆順にする問題


解答1
reverse()を使用してリストを逆順にしています。

input()
list = input().split()
list.reverse()
print(" ".join(list))


解答2
リストに対して[::-1]を指定することで逆順にできます。

input()
print(" ".join(input().split()[::-1]))

ITP1_5_D: Structured Programming

構造化プログラミング | プログラミング入門 | Aizu Online Judge
プログラムを書き直す問題


3の倍数であるか、剰余が3か0になるまで10で割り続け、3である場合に、その数字を出力します。


解答1
*演算子を使用し、リストやタプルから引数をアンパックします。

n = int(input())
list = []
for i in range(3, n + 1):
    if i % 3 == 0:
        list.append(i)
    else:
        x = i
        while x:
            if x % 10 == 3:
                list.append(i)
                break
            else:
                x = x // 10
print("", *list)


解答2
剰余を求めていた処理を、文字列に3を含むかという処理に変更しています。

n = int(input())
for i in range(3, n + 1):
    if i % 3 == 0 or '3' in str(i):
        print('', i, end='')
print()

ITP1_5_C: Print a Chessboard

チェック柄の描画 | プログラミング入門 | Aizu Online Judge
繰り返し構文を用いて2種類の文字列を出力する問題


現在の行と列の番号の合計が奇数か偶数かで出力する文字列が変わります。


解答1

while True:
    H, W = map(int, input().split())
    if H == W == 0:
        break
    for i in range(H):
        for j in range(W):
            print('#.'[(i + j) % 2], end='')
        print()
    print()


解答2

while True:
    H, W = map(int, input().split())
    if H == 0 and W == 0:
        break
    line = ['#.' * (W // 2) + '#' * (W % 2), '.#' * (W // 2) + '.' * (W % 2)]
    for i in range(H):
        print(line[i % 2])
    print()


解答3

while True:
    H, W = map(int, input().split())
    if H == W == 0:
        break
    line = '#.' * (W // 2 + 1)
    for i in range(H):
        if i % 2:
            print(line[1:W+1])
        else:
            print(line[:W])
    print()