競技プログラミングの問題(Python)

Question

下記のプログラミングの問題を解きました。
プログラミングの問題
みなさんはPythonでこの問題をどのように解きますか？
僕は以下のように解きました。
ぜひみなさんの解答を参考にさせてください。

from __future__ import print_function
n = int(raw_input())
bilding1 = [[" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"]
           ]
bilding2 = [[" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"]
           ]
bilding3 = [[" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"]
           ]
bilding4 = [[" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"],
           [" 0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0 ","0"]
           ]
bilding = []
bilding.append(bilding1)
bilding.append(bilding2)
bilding.append(bilding3)
bilding.append(bilding4)

for i in range(0,n):
    inf = raw_input()
    inf= inf.split(" ")
    b = int(inf[0]) -1
    f = int(inf[1]) -1
    r = int(inf[2]) -1
    v = int(inf[3])
    bilding[b][f][r] = int(bilding[b][f][r]) + v
    if(r == 0):
        bilding[b][f][r] = " " + str(bilding[b][f][r]) + " "
    elif(r < 9):
        bilding[b][f][r] = str(bilding[b][f][r]) + " "
    else:
        bilding[b][f][r] = str(bilding[b][f][r])

count = 0
for j in range(0,4):
    for k in range(0,3):
        for l in range(0,10):
            print((bilding[j][k][l]),end="")
        print("")
    if(count < 3):
        print("####################")
        count += 1

Answer 1

私も試しに書いてみました。python3

room_list = [[[0]*10,[0]*10,[0]*10],[[0]*10,[0]*10,[0]*10],[[0]*10,[0]*10,[0]*10],[[0]*10,[0]*10,[0]*10]]
for i in range(int(input())):
    b,f,r,v = map(int, input().split())
    room_list[b-1][f-1][r-1] += v
for b_ix in range(4):
    if b_ix > 0: print("#"*20)
    for f_ix in range(3):
        print(""," ".join(map(str, room_list[b_ix][f_ix])))

Comments

[namnium1125]

#は20個必要なのではないでしょうか？

またlistは組み込み型を上書きしてしまっているのでできれば別な名称に変えるべきだと思います。

[退会済みユーザー]

今確認しました。確かに20でしたので、コード手直ししました。

[super_tomato]

個人的に一番分かりやすく簡潔なコードだと思いましたのでベストアンサーにさせていただきました！回答ありがとうございました。

[namnium1125]

> 入居者数の前には１つの空白を出力して下さい。

このコードのままだと各フロアの一つ目の部屋の前にスペースが入りませんが、これは問題ないのでしょうか？

[super_tomato]

厳密に言えば、必要ですね。

[退会済みユーザー]

ありがとうございます。条件漏れがありましたか。ちょっと直してみました。いかがでしょうか。

[super_tomato]

出力の仕方が厳密なルールからは反していますが、そこまでしてもらわなくても十分参考になりますので！わざわざありがとうございました。

Answer 2

リファレンスをみたりREPLで試しつつ書いてみました。
特にフォーマットするコードは明快とは言いがたいので「コードはいくらでもヘンテコにできる」という悪い例かも知れません。

import numpy as np
import re

num_b = 4
num_f = 3
num_r = 10

b_separator = '\n' + ''.join(['#'] * num_r * 2) + '\n '

rooms = np.zeros((num_b, num_f, num_r), dtype=np.uint8)

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

o = re.sub(r'[[]]', '', ' ' + b_separator.join(map(str, rooms)))
print(o)

---

python 3.6.0

Comments

[super_tomato]

正規表現をまだ慣れていないので参考になりました！ご回答ありがとうございました。

Answer 3

私ならこうしますかね…入力は標準入力とします。

A_college = [[[0 for _ in range(10)] for _ in range(3)] for _ in range(4)]

N = int(input())
moves = [[int(i)-1 for i in input().split()] for _ in range(N)]

for m in moves:
    b,f,r,v = m
    v += 1
    A_college[b][f][r] += v

for b,building in enumerate(A_college):
    if 0 != b: print('#'*20)
    for flr in building:
        print('',*flr)

for文のところでbuilding(棟)とかflr(floor、階)とかわかりやすい単語にすると意味のあるプログラムに見えて、個人的には好きです。

Python 3.6.1

Comments

[super_tomato]

とても分かりやすい（僕にとって）コードで参考になりました！ご回答ありがとうございました。

Answer 4

こんにちは。

いろいろな書き方が出来ると思いますが、考え方の一例として回答します。
なお入力データは、標準入力から与えるものとしました。

import re
from collections import defaultdict

# データ行から、部屋の特定部分と人数との2つをキャプチャする正規表現
pattern = r"^(.+) (-?\d+)$"

# データ行を読み込み、人数をカウントしておく辞書(デフォルト値: 0)
counts = defaultdict(lambda: 0)

# 標準入力からの読み込み
for _ in range(int(input())):
    m = re.match(pattern, input())
    if m:
        counts[m.group(1)] += int(m.group(2))

# 結果の出力
for building in range(4):
    for floor in range(3):
        counts_of_residents = []
        for room in range(10):
            key = "{0} {1} {2}".format(building+1, floor+1, room+1)
            counts_of_residents.append(str(counts[key]))

        print(" ".join(counts_of_residents))

    if building < 3:
        print("#" * 20)

　
考え方としては、入力のデータ行、たとえば

3 2 2 7

を、部屋を特定する文字列"3 2 2"と、人数の7とに分けて、
前者をキーとして、後者を値とする辞書 counts に格納する
ことにしました。
辞書にまだないキーに対しては、0 をデフォルト値としたいために、
defaultdict を初期値が 0になるようにして作っています。

以下、実行例です。（上記のコードを count.py とし、データをinput.txtに入れました。）

$ cat input.txt
3
1 1 3 8
3 2 2 7
4 3 8 1
$ python count.py < input.txt
0 0 8 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
####################
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
####################
0 0 0 0 0 0 0 0 0 0
0 7 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
####################
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 1 0 0
$ 

以上参考になれば幸いです。

Comments

[super_tomato]

詳しい解説ありがとうございました。参考になります。ご回答ありがとうございました。

[jun68ykt]

defaultdict 使って、楽してみました(笑) でも、Pythonのリスト操作の腕を上げるには、他の皆様の回答のほうがよいと思います。私も勉強になりました。

Answer 5

最後の出力に苦労しますが、部屋を120室、直列に用意すればデータ構造がスッキリします。

from io import StringIO
from textwrap import dedent


_INPUT = iter(dedent('''\
    3
    1 1 3 8
    3 2 2 7
    4 3 8 1
''').splitlines())

def input():
    return next(_INPUT)


rooms = [0] * 10 * 3 * 4
for _ in range(int(input())):
    b, f, r, v = map(int, input().split())
    b, f, r = (x - 1 for x in (b, f, r))
    rooms[30 * b + 10 * f + r] += v

with StringIO() as fp:
    for n in range(0, len(rooms), 10):
        if n % 30 == 0:
            print('#' * 20, file=fp)
        print(*rooms[n:n+10], file=fp)
    for line in fp.getvalue().splitlines()[1:]:
        print(line)

Comments

[super_tomato]

ありがとうございます！参考になります！

Answer 6

みなさんのコードを参考に以下のように改良しました。ありがとうございました。

from __future__ import print_function
n = int(raw_input())
bilding = [[[0]*10,[0]*10,[0]*10],[[0]*10,[0]*10,[0]*10],[[0]*10,[0]*10,[0]*10],[[0]*10,[0]*10,[0]*10]]
for i in range(0,n):
    b,f,r,v = map(int,raw_input().split(" "))
    bilding[b-1][f-1][r-1] += v

lcnt = 0
for j in range(0,4):
    for k in range(0,3):
        for l in range(0,10):
            print(" ",end="")
            print(bilding[j][k][l],end="")
        if(lcnt < 11):
            print("")
            lcnt += 1
    if(j < 3):
        print("#"*20)

• lcntは出力の最後に改行が入ってしまうのでそれをなくすためです。

Comments

[namnium1125]

pythonのif文の条件式は、条件が複数でないならば()で囲む必要はないですよ。また0始まりならばrange(4)でいいと思います。

最後の改行を出さないというのは面白いですね！勉強になりました。

[super_tomato]

ありがとうございます。if文の条件式についてですがちょうど今それを発見したところでした笑。
0からならrange(数値)でいいのですね。勉強になります。

最後の改行についてですが、最後に改行を入れてしまうとサイトの方でエラーが出てしまうのであのように行いました。