ABC434 B – Bird Watching

B - Bird Watching

AtCoder is a programming contest site for anyone from beginners to experts. We hold weekly programming contests online.

atcoder.jp

M種類の鳥が合わせてN羽飛んでいる。

鳥の種類ごとに平均の重さを求める問題。

鳥の種類ごとの出現回数と種類ごとの重さの合計をそれぞれ配列で持つことにしました。

重さから種類ごとの出現回数を割ると平均が出せます。

以下でACできました。

def i_map(): return map(int, input().split())
def i_list(): return list(map(int, input().split()))

n, m = i_map()

a_arr = [0]*m
b_arr = [0]*m

for i in range(n):
  a, b = i_map()
  a-=1
  a_arr[a] += 1
  b_arr[a] += b

for i in range(m):
  print(b_arr[i]/a_arr[i])Code language: Python (python)