ABC_359_B

[ishihara0507]

もう少し短く書ける

• 自分のコード

  
  def main():
  N = int(input())
  A = [int(s) for s in input().split()]
  
  count = 0
  for i in range(2*N-2):
  if A[i] == A[i+2]:
    count += 1
  
  print(count)

if name == "main": main()

- 短縮したコード（他の参加者のコード参照）

def main(): N = int(input()) A = list(map(int, input().split())) ans = 0 for i in range(len(A) - 2): ans += A[i] == A[i + 2] print(ans)

if name == "main": main()

## 条件を満たした回数を論理演算の結果で数える

for i in range(len(A) - 2): ans += A[i] == A[i + 2]

上のコードのように、if文を使わずに直接論理演算の結果で条件を満たした回数を数える。