atcoder問題集
atcoderの問題を前解いていたのでそのコードを共有したいと思います。
勉強記録として残します。
from unittest import result
def Sample1():
"""
配列の操作
入力した数値を何回2で割る事が出来るのか？
"""
# 入力するデータの個数
N = int(input())
# サイズNの配列Aへの入力
A = list(map(int,input().split()))
counter = 0 # 操作回数
while True: # Trueの間ループする。
can_do = True
for i in range(N):
if A[i] % 2 == 1:
can_do = False
if not can_do: # Falseの場合
break
for i in range(N):
A[i] //= 2
counter += 1
print(counter)
# Sample1()
def Sample2():
"""
A以上B以下の整数の中にKの倍数があるかどうか
"""
K = int(input())
A, B = map(int, input().split())
ext = False
for i in range(A, B + 1): # A からBをループする。
if i % K == 0: # 倍数があった場合フラグをTrueにする。
ext = True
if ext == True:
print('OK')
else:
print('NG')
# Sample2()
def Sample3():
# 整数nの各桁の和を求める関数
def sum(n):
sum_1 = 0
while n > 0: # nが0以下になるまでループする。
sum_1 += n % 10 # 1の位を求める
n //= 10
return sum_1
N, A, B = map(int, input().split())
result = 0
for i in range(1, N + 1):
if A <= sum(i) <= B: # A以上B以下
result += i
print(result)
# Sample3()
def Sample4():
"""
お金の枚数を計算する。
"""
N, Y = map(int, input().split()) # 入力する。Nが枚数Yが金額
a, b, c = -1, -1, -1
# 全探索
for i in range(N + 1):
for j in range(N + 1):
n = N - i - j
if n < 0 or n > N: # nが0以上N以下でない場合はスキップ
continue
if 10000 * i + 5000 * j + 1000 * n == Y: # 条件を満たすのか確認する。
a, b, c = i, j, n
print(f'1万円札{a}枚5千円札{b}枚千円札{c}枚')
# Sample4()
def prime_number():
num = int(input('数値入力'))
if num <= 1: # 素数ではない
print(False)
else:
for i in range(2, int(num**0.5)+1):
if num % i == 0: # 0になったらFalseを返す。
print(False)
break
else:
print(True)
# prime_number()
def Sample6():
num = int(input('数値入力'))
cnt = 0
for i in range(2, int(num**0.5)+1):
cnt += i
print(cnt)
# Sample6()
def Sample7():
"""
周期性
入力例 A B C K
操作をK回行う
そしてA-Bをした整数を出力する。
"""
A, B, C, K = map(int, input().split())
print(A - B if K % 2 == 0 else B - A)
# Sample7()
def Sample8():
"""
旅行プラン
T秒後に(x,y)に移動できるのかどうか
入力例:
2
3 1 2
6 1 1
出力例:
yes
"""
def solve():
N = int(input('プラン数'))
pt, px, py = 0, 0, 0 # 前回の時刻と座標を表す変数
for i in range(N):
t, x, y = map(int, input().split()) # 移動先の情報を受け取る
T, X, Y = t - pt, abs(x - px), abs(y - py) # 前回の状態と差分を取るabs→(絶対値)
if T < X + Y: # 間に合わない時は不可能
return False
if T % 2 != (X + Y) % 2:
return False
pt, px, py = t, x, y
return True # 最後まで到達した時
print('Yes' if solve() else 'NO') # returnでTrueの場合はYes,Falseの場合はNo
# Sample8()
def Sample9():
"""
N個の整数が並んでいてN個の整数の総和として考えられる最大値を求める
入力例
3
-10 5 4
出力例
19
"""
N = int(input())
A = list(map(int, input().split())) # 入力はリストにする。
num_minus = sum(v < 0 for v in A) # マイナスの個数を求める。sumを使用して合計を求める。
sum_abs = sum(map(abs, A)) # N個の絶対値の総和と最小値
min_abs = min(map(abs, A))
if num_minus % 2 == 0: # マイナスが偶数の場合
print(sum_abs)
else: # マイナスが奇数の場合
num = sum_abs - min_abs
print(num)
# Sample9()
def Sample10():
"""
入力した数値にいくつ素数がいくつ出るのか？
"""
num = int(input())
prime = []
for i in range(2, num + 1):
for j in range(2, i):
if i % j == 0:
break # 0で割れた場合はjのfor文を抜ける。
else:
prime.append(i) # 0にならなかった場合にリストに追加する。
print(prime)
# Sample10()
def Sample11():
"""
九九の表示
"""
for i in range(1, 10):
for j in range(1, 10):
num = i * j
if num >= 10:
print(num, end=" ")
else:
print(num, end=" ")
print()
Sample11()