結果

問題 No.459 C-VS for yukicoder
ユーザー compass19compass19
提出日時 2016-12-10 01:24:13
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 365 ms / 2,000 ms
コード長 1,490 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 26,216 KB
最終ジャッジ日時 2024-11-28 21:13:04
合計ジャッジ時間 7,335 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 58
権限があれば一括ダウンロードができます

ソースコード

diff # 

# データ入力
h, w, n = map(int, input().split())
s = [0 for _ in range(w)]
for i in range(h):
	for k, elm in enumerate(list(input())): 
		if elm == '#': 
			s[k] += 1

c_tmp = [int(input()) for _ in range(n)]


# パック情報
# pack_dic[pack]: dict.
pack_dic = { pack: {'left': c_tmp[pack], 'elm': [0, 0, 0], 'sum': 0} for pack in range(n)}


# 列の情報の入力
# column_list[column] = list(columnに影響するpack_number)
column_list = [list() for _ in range(w)]
for pack, left in enumerate(c_tmp):
	column_list[left].append(pack)
	column_list[left+1].append(pack)
	column_list[left+2].append(pack)


# main(詰め込み)
for column in range(w):
	sorted_temp_list = sorted(column_list[column], \
							  key=lambda x: (pack_dic[x]['sum'], pack_dic[x]['left']))

	# 列columnについて、そのcolumnに影響するpackを
	# 現時点でのpackの要素の合計が少ない順に#を詰める(要素が全くないpackをなくすため)
	stock = s[column]
	while stock != 0:
		for pack in sorted_temp_list:
			left = pack_dic[pack]['left']
			pack_dic[pack]['elm'][column-left] += 1
			pack_dic[pack]['sum'] += 1
			stock -= 1
			if stock == 0:
				break


# パックの表示
def pack_print(_list):
	left, center, right = _list
	for _ in range(3):
		s = str()
		s += '#' if left > 0 else '.'
		s += '#' if center > 0 else '.'
		s += '#' if right > 0 else '.'
		print(s)
		left -= 1
		center -= 1
		right -= 1

for pack in range(n):
	pack_print(pack_dic[pack]['elm'])
0