atcoderアルゴリズム③

問題①

問題①

入力例

解答コード

N = int(input())
data_lst = []
cnt = 0
cnt_1 = 0
for _ in range(N):
    data = input()
    if data == 'For':
        cnt += 1
    else:
        cnt_1 += 1

if cnt > cnt_1:
    print('Yes')
else:
    print('No')

問題②

問題

入力例

解答コード

N, M = map(int, input().split())
n_lst = []
m_lst = []
ans = 0
index = 0
for _ in range(N):
    n = input()
    n_1 = n[-3:] # 末尾3文字を取得する
    n_lst.append(n_1)
for _ in range(M):
    m = input()
    m_lst.append(m)

for i in n_lst:
    if i in m_lst:
        ans += 1
print(ans)

問題③

問題

入力例

解答コード

N = int(input())
n_lst = list(map(int,input().split()))

while len(n_lst) > 1:
    a = [] # 初期化
    for i in range(1, len(n_lst)):
        a.append(abs(n_lst[i] - n_lst[i-1]))
    n_lst = a # 追加
    print(n_lst)
print(n_lst)
print(n_lst[0])

以上になります。