AtCoder Beginner Contest 149 過去問(A~C)

参照

atcoder.jp

A問題

文字列を繋げろとしか書かれてなかったので、
サンプル例を参考に解きました。

s,t = input().split()
print(t+s)

B問題

Cを解いて安心したせいか、
気が抜けてBでWAを出してしまいました。
日々のルーティーンは大事です。

a,b,k=map(int, input().split())
if a<k: 
  if a+b<k:
    b=0
  else:b=a+b-k
  a=0
else:
  a=a-k
print('%d %d'%(a,b))

C問題

与えられたnより大きい最小の素数を求めるもの。
nから10^6までの数字を探索すればいいだけで、
素数を見つければ計算終了となる為、
1つずつ調べて行ってもTLEにはなりませんでした。

n = int(input())

if n==2: print(n)
else: 
  for i in range(n,999999,1):
    for j in range(2,i):
      if i%j==0: break
      elif j==i-1:
        print(i)
        exit(0)