はじめに
AtCoder の問題を Python で解いています。問題の復習と自分の技術力を分析するために、問題を解いて得られた知見をメモとして残しています。周りの方から「知見を共有して欲しい」「公開することでコードレビューされて、技術力がより高まるよ」などの意見を頂いたため、本記事を執筆する運びとなりました。
本記事について
本記事について説明します。
構成
本記事では 分類観点 を定義し、分類観点ごとに問題を分類しています。各分類観点では、サンプルコード（入力 / 実行結果）と 解答例 を記載しています。サンプルコード（入力 / 実行結果）と 解答例 は下記の通りです。
| 項目 | 項目内容 |
|---|---|
| サンプルコード | 問題から得られた知見と検証によって得られた知見を整理したコード |
| 入力 | サンプルコードの入力例 |
| 実行結果 | サンプルコードの実行結果 |
| 解答例 | AtCoder で AC になったコード |
| 解答例(TLE) | AtCoder で TLE になったコード |
分類観点の定義
分類観点の定義は下記の通りです。1つの問題から複数の観点で分類できる問題は、複数の観点に分類します。
| 章 | 観点 | 観点内容 |
|---|---|---|
| 第1章 | 入出力観点 | 入出力を知っていれば解ける問題 |
| 第2章 | 演算観点 | 演算を知っていれば解ける問題 |
| 第3章 | 制御フロー観点 | 制御フローを知っていれば解ける問題 |
| 第4章 | 組み込み型観点 | 組み込み型を知っていれば解ける問題 |
| 第5章 | 組み込み関数観点 | 組み込み関数を知っていれば解ける問題 |
| 第6章 | モジュール観点 | モジュールを知っていれば解ける問題 |
| 第7章 | データ構造観点 | データ構造を知っていれば解ける問題 |
| 第8章 | アルゴリズム観点 | アルゴリズムを知っていれば解ける問題 |
| 第9章 | 計算量観点 | 計算量を工夫すれば解ける問題 |
| 第10章 | 数学観点 | 数学的知見を知っていれば解ける問題 |
| 第11章 | カテゴリ観点 | カテゴリ特有の解き方を知っていれば解ける問題 |
| - | 参考 | - |
分類する問題の範囲
分類する問題の範囲は下記の通りです。ABC-C問題 と ARC-A問題 は一部の問題が同じです。同じ問題の場合、ABC-C問題 として分類します。
- ABC-A問題（ABC001 - ABC158）
- ABC-B問題（ABC001 - ABC158）
- ABC-C問題（ABC001 - ABC158）
- ARC-A問題（ARC001 - ARC103）
分類する対象の問題
本記事では、上記分類範囲で示したすべての問題が記載されているわけではありません。私が上記分類観点で知見が得られた問題のみを記載しています。
バージョン / 言語
サンプルコードは Python の下記バージョンで動作することを確認しています。
Python 3.6.4
解答例は AtCoder の下記言語で AC になることを確認しています。
Python3 (3.4.3)
コーディング方針
コーティングは下記方針で行っています。また、Python のコードスタイルガイドとして PEP 8 があります。
- インデントは空白4つを使う
- コード1行の長さは定義しない
- 演算子の前後、コンマの後、括弧類のすぐ内側に空白を入れない
- import は行を分けて定義する
- クォーテーションはダブルクォーテーションで統一する
- 複数の変数を1行で定義できる場合は1行で定義する
- 変数名は AtCoder の問題文を参考にする
- 命名規則はない
- 自分の直感に合うコーディングをする
更新方針
本記事は下記方針で記事を更新する予定です。更新情報は 編集履歴 より確認をお願いします。
- 新規問題の分類
- 分類の詳細化
- AtCoder言語アップデートの対応（AtCoderの言語アップデートに関して - AtCoder）
知的財産権
本記事に記載しているコードはすべて私が作成したコードです。また、AtCoder における知的財産権は下記の通りです。
知的財産権
1.本サービスに対して投稿されたプログラムの所有権と著作権は、そのプログラムを作成したユーザに帰属します
2.本サービスを構成する文章、画像、プログラムその他のデータ等についての一切の権利（所有権、知的財産権、肖像権、パブリシティー権等）は、ユーザ自身が作成したものを除き、弊社又は当該権利を有する第三者に帰属しています
3.ユーザ自身が作成した著作物を本サービスを通じて掲載した場合、弊社が宣伝告知等に利用することを許諾するものとします。また、かかる使用に際して、当該ユーザは著作者人格権を行使しないものとします
本記事に記載しているコードを私的利用する場合、自由に活用して頂いて構いません。また、作成したコードは GitHub で公開しています。コードを転用する / 商業利用する場合は、メールで事前にご相談をお願い致します。
学習方法の特性上「他人のコードに酷似している」ことがあります。気分を害される方がいましたら、相談の上で引用元の記載 / 記事の更新 / 文章の削除等に対応しますので、ご連絡をお願い致します。
最後に
サービスを提供している AtCoder株式会社 と、交流している 競技プログラマ の方々に感謝を致します。
編集履歴
- 初版公開
1.1版 編集内容
1.2版 編集内容
1.3版 編集内容
1.4版 編集内容
1.5版 編集内容
1.6版 編集内容
2.0版 編集内容
第1章 入出力観点
1.1 出力
整数（int）
整数を出力する
サンプルコード
print(10)
実行結果
10
整数を変数に格納して出力する
サンプルコード
x=20
print(x)
実行結果
20
int() は整数を返す
サンプルコード
print(int(2.9))
実行結果
2
浮動小数点数（float）
浮動小数点数を出力する
サンプルコード
print(2.9)
実行結果
2.9
浮動小数を変数に格納して出力する
サンプルコード
x=3.5
print(x)
実行結果
3.5
float() は浮動小数点数を返す
サンプルコード
print(float(2))
実行結果
2.0
文字列（str）
- テキストシーケンス型
- str
- str()：文字列を返す
文字列を出力する
サンプルコード
print("Hello World!")
実行結果
Hello World!
文字列を変数に格納して出力する
サンプルコード
s="Hello World!"
print(s)
実行結果
Hello World!
文字列の変数のインデックスを指定して文字を出力する
サンプルコード
s="Hello World!"
print(s[0])
print(s[11])
実行結果
H
!
解答例
s=input()
print(s[0]+str((len(s)-2))+s[-1])
解答例
a,*b,c=input()
print(a+str(len(b))+c)
解答例
print(input()[int(input())-1])
解答例
s=input()
i=int(input())
print(s[i-1])
ABC048 A - AtCoder *** Contest
解答例
print("A%sC"%input()[8])
解答例
print("A"+input()[8]+"C")
リスト（list）
1次元リストを出力する
サンプルコード
Lists=[1,2,3,4,5]
print(Lists)
実行結果
[1, 2, 3, 4, 5]
1次元リストの各要素を出力する
サンプルコード
Lists=[1,2,3,4,5]
for List in Lists:
print(List)
実行結果
1
2
3
4
5
1次元リストの各要素を改行しない（アンパック）で出力する
サンプルコード
Lists=[1,2,3,4,5]
print(*Lists)
実行結果
1 2 3 4 5
1次元リストのインデックス指定して各要素を出力する
サンプルコード
List=[1,2,3,4,5]
for i in range(len(List)):
print(List[i])
実行結果
1
2
3
4
5
2次元リストを出力する
サンプルコード
List=[[1,2,3],[4,5,6]]
print(List)
実行結果
[[1, 2, 3], [4, 5, 6]]
2次元リストの各行を出力する
サンプルコード
Lists=[[1,2,3],[4,5,6]]
for List in Lists:
print(List)
実行結果
[1, 2, 3]
[4, 5, 6]
2次元リストの各要素を出力する
サンプルコード
List=[[1,2,3],[4,5,6]]
for Row in List:
for Col in Row:
print(Col)
実行結果
1
2
3
4
5
6
2次元リストの各要素をインデックス指定して出力する
サンプルコード
List=[[1,2,3],[4,5,6]]
for i in range(len(List)):
for j in range(len(List[i])):
print(List[i][j])
実行結果
1
2
3
4
5
6
ABC042 B - 文字列大好きいろはちゃんイージー / Iroha Loves Strings (ABC Edition)
解答例
n,l=map(int,input().split())
s=sorted([input() for i in range(n)])
print(*s,sep="")
解答例
n,l=map(int,input().split())
s=sorted([input() for i in range(n)])
print("".join(s))
解答例
K,X=map(int,input().split())
print(*[x for x in range(X-K+1,X+K)])
解答例
K,X=map(int,input().split())
print(*list(range(X-K+1,X+K)))
解答例
K,X=map(int,input().split())
print(*range(X-K+1,X+K))
ABC148 B - Strings with the Same Length
解答例
N=input()
S,T=input().split()
print(*[s+t for s,t in zip(S,T)],sep="")
タプル（tuple）
サンプルコード
x=1,2,3
print(x)
実行結果
(1, 2, 3)
解答例
N,Y=map(int,input().split())
answer=-1,-1,-1
for x in range(N+1):
for y in range(N-x+1):
z=N-x-y
if 0<=z<=2000 and 10000*x+5000*y+1000*z==Y:
answer=x,y,z
break
else:
continue
break
print(*answer)
区切り文字の指定
- print()：キーワード引数 sep を指定する
指定がないと 半角スペース で区切る
サンプルコード
a=1
b=2
c=3
print(a,b,c)
実行結果
1 2 3
解答例
print(*input().split(","))
改行 を指定する
サンプルコード
a=1
b=2
c=3
print(a,b,c,sep="\n")
実行結果
1
2
3
カンマ を指定する
サンプルコード
a=1
b=2
c=3
print(a,b,c,sep=",")
実行結果
1,2,3
区切り文字なし を指定する
サンプルコード
a=1
b=2
c=3
print(a,b,c,sep="")
実行結果
123
ABC046 B - 文字列大好きいろはちゃんイージー / Iroha Loves Strings (ABC Edition)
解答例
n,l=map(int,input().split())
s=sorted([input() for i in range(n)])
print(*s,sep="")
ABC148 B - Strings with the Same Length
解答例
N=input()
S,T=input().split()
print(*[s+t for s,t in zip(S,T)],sep="")
Tenka1 Programmer Beginner Contest 2019 B - e*** *****e *ee* **e**
解答例
input()
S=input()
k=int(input())
for s in S:print(s if s==S[k-1] else "*",end="")
末尾文字の指定
- print()：キーワード引数 end を指定する
指定がないと 改行する
サンプルコード
print("a")
print("b")
実行結果
a
b
末尾文字なし を指定する
サンプルコード
print("a",end="")
print("b")
実行結果
ab
解答例
o=list(input())
e=list(input())+[""]
for x,y in zip(o,e):print(x+y,end="")
1.2 入力
- input()：入力から1行を読み込み、文字列に変換して（末尾の改行を除いて）返す
- map()：function の結果を返しながら、全ての要素に適用する iterator を返す
- str.split()：文字列を区切った単語のリストで返す
- リスト内包表記
1行 / 1列
整数を変数に格納する
入力
2
サンプルコード
N=int(input())
print(N)
実行結果
2
解答例
print(int(input())-1)
文字列を1つの変数に格納する
入力
Hello world!
サンプルコード
s=input()
print(s)
実行結果
Hello world!
文字列を複数の変数に格納する
入力
1234
サンプルコード
a,b,c,d=input() # 1234を整数(int)型ではなく、文字列(str)型で変数に格納する
print(a,b,c,d)
実行結果
1 2 3 4
入力
abcd
サンプルコード
a,b,c,d=input()
print(a,b,c,d)
実行結果
a b c d
解答例
print("ABC" if input()=="1" else "chokudai")
解答例
a,b,c=input()
print("Yes" if a==c else "No")
解答例
n=input()
print("Yes" if n[0]==n[2] else "No")
解答例
a,b,c,d=input()
print("Yes" if a==b==c or b==c==d else "No")
解答例
n=input()
print("Yes" if n[0]==n[1]==n[2] or n[1]==n[2]==n[3] else "No")
解答例
a,b,c,d=input()
print("Good" if a!=b and b!=c and c!=d else "Bad")
解答例
a,b,c,d=input()
print("Good" if a!=b!=c!=d else "Bad")
1行 / 複数列
整数を複数の変数に格納する
入力
1 2
サンプルコード
a,b=map(int,input().split())
print(a)
print(b)
実行結果
1
2
解答例
a,b=map(int,input().split())
print(b,a)
解答例
S,T=input().split()
print(T+S)
文字列を複数の変数に格納する
入力
Hello world!
サンプルコード
a,b=input().split()
print(a)
print(b)
実行結果
Hello
world!
解答例
a,b=input().split()
print(b,a)
a,b,c=input().split(",")
print(a,b,c)
解答例
a,_,b=input()
print("DH"[a==b])
解答例
a,b,c,d=input()
print("Yes" if a==b==c or b==c==d else "No")
ABC110 A - Maximize the Formula
解答例
A,B,C=sorted(map(int,input().split()))
print(10*C+B+A)
解答例
a,b,c=map(int,input().split())
print(a*b//2)
リストに整数を格納する
入力
1 2 3 4 5
サンプルコード
List=list(map(int,input().split()))
print(List)
実行結果
[1, 2, 3, 4, 5]
入力
1 2 3 4 5
サンプルコード
List=[int(i) for i in input().split()]
print(List)
実行結果
[1, 2, 3, 4, 5]
リストに文字列を格納する
入力
Hello world !
サンプルコード
List=list(input().split())
print(List)
実行結果
['Hello', 'world', '!']
入力
Hello world !
サンプルコード
List=[i for i in input().split()]
print(List)
実行結果
['Hello', 'world', '!']
アンパックを使ってリストに格納する
解答例
a,*b,c=input()
print(a+str(len(b))+c)
解答例
print(*input().split(","))
ABC103 A - Task Scheduling Problem
解答例
*a,=map(int,input().split())
print(max(a)-min(a))
複数行 / 1列
整数を複数の変数に格納する
入力
1
2
サンプルコード
a=int(input())
b=int(input())
print(a,b)
実行結果
1 2
入力
1
2
サンプルコード
a,b=[int(input()) for i in range(2)]
print(a,b)
実行結果
1 2
ABC044 A - 高橋君とホテルイージー / Tak and Hotels (ABC Edit)
解答例
n,k,x,y=(int(input()) for i in [0]*4)
print(n*x-(x-y)*max(n-k,0))
解答例
x,a,b=[int(input()) for _ in range(3)]
print((x-a)%b)
解答例
a,b,c,d=[int(input()) for _ in range(4)]
print(min(a,b)+min(c,d))
リストに格納する
入力
1
2
3
4
5
サンプルコード
List=[int(input()) for i in range(5)]
print(List)
実行結果
[1, 2, 3, 4, 5]
解答例
print(len(set([int(input()) for _ in range(int(input()))])))
ABC091 B - Two Colors Card Game
解答例
n=[input() for _ in range(int(input()))]
m=[input() for _ in range(int(input()))]
l=list(set(n))
print(max(0,max(n.count(l[i])-m.count(l[i]) for i in range(len(l)))))
解答例
N=int(input())
p=[int(input()) for i in range(N)]
print(sum(p)-max(p)//2)
入力行数(Row=5)が指定され、リストに整数を格納する
入力
5
0
1
2
3
4
サンプルコード
Row=int(input())
List=[int(input()) for i in range(Row)]
print(List)
実行結果
[0, 1, 2, 3, 4]
解答例
l,w=map(int,input().split())
n=int(input())
for i in range(n):
a=int(input())
print(l-a if l>a else "0" if w>=a else "-1")
ABC112 B - Time Limit Exceeded
解答例
N,T=map(int,input().split())
m=1001
for i in range(N):
c,t=map(int,input().split())
if t<=T:m=min(m,c)
print("TLE" if m==1001 else m)
入力行数(Row=5)が指定され、リストに文字列を格納する
入力
5
a
b
c
d
e
サンプルコード
Row=int(input())
List=[input() for i in range(Row)]
print(List)
実行結果
['a', 'b', 'c', 'd', 'e']
複数行 / 複数列
複数のリストに格納する（入力時に行数(Row)を指定する）
入力
3
0 a
1 b
2 c
サンプルコード
Row=int(input())
Number=[]
Alphabet=[]
for i in range(Row):
n,a=input().split()
Number.append(int(n))
Alphabet.append(a)
print(Number)
print(Alphabet)
実行結果
[0, 1, 2]
['a', 'b', 'c']
解答例
n=int(input())
S=[]
P=[]
for i in range(n):
s,p=input().split()
S.append(s)
P.append(int(p))
if max(P)>sum(P)/2:print(S[P.index(max(P))])
else:print("atcoder")
2次元リストに格納する（入力時に行数(Row)を指定する）
入力
3
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
サンプルコード
Row=int(input())
List=[list(map(int,input().split())) for i in range(Row)]
print(List)
実行結果
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15]]
入力
3
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
サンプルコード
Row=int(input())
List=[[int(j) for j in input().split()] for i in range(Row)]
print(List)
実行結果
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15]]
入力
3
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
サンプルコード
Row=int(input())
List=[]
for i in range(Row):
List.append(list(map(int,input().split())))
print(List)
実行結果
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15]]
2次元リストに格納する（入力時に行数が指定されない）
入力
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
サンプルコード
List=[list(map(int,input().split())) for i in range(3)]
print(List)
実行結果
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15]]
1次元リストに格納する
入力
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
サンプルコード
List=[]
for i in range(3):
List+=list(map(int,input().split()))
print(List)
実行結果
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
変数に格納する
入力
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
サンプルコード
a=" ".join([input() for i in [0]*3])
print(a)
実行結果
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
解答例
n,q=map(int,input().split())
a=[0]*n
for i in range(q):
l,r,t=map(int,input().split())
for i in range(l,r+1):a[i-1]=t
print(*a,sep="\n")
解答例
n,m,x=map(int,input().split())
s=sum(int(i)<x for i in input().split())
print(min(s,m-s))
ABC118 B - Foods Loved by Everyone
解答例
n,m=map(int,input().split())
S=set(range(1,m+1))
for i in range(n):
K,*A=map(int,input().split())
S&=set(A)
print(len(S))
ABC121 B - Can you solve this?
解答例
N,M,C=map(int,input().split())
B=list(map(int,input().split()))
count=0
for i in range(N):
A=list(map(int,input().split()))
p=C
for j in range(M):p+=A[j]*B[j]
if p>0:count+=1
print(count)
第2章 演算観点
2.1 算術演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| + | a+b | 加算 |
| - | a-b | 減算 |
| * | a*b | 乗算 |
| ** | a**b | べき乗 / べき根 |
| / | a/b | 除算 |
| // | a//b | 小数点切り捨て除算 |
| % | a%b | 剰余 |
加算
サンプルコード
print(1+2)
実行結果
3
解答例
a,b=map(int,input().split())
print(a+b if a+b<10 else "error")
ABC110 A - Maximize the Formula
解答例
A,B,C=sorted(map(int,input().split()))
print(10*C+B+A)
解答例
l=list(map(int,input().split()))
print(sum(l)+max(l)*9)
減算
サンプルコード
print(2-1)
実行結果
1
解答例
x=abs(int(input())-int(input()))
print(min(x,10-x))
解答例
print(sum(1-eval(input().replace(" ","-")) for _ in range(int(input()))))
解答例
print(int(input())-int(input()))
解答例
print(int(input())-1)
解答例
s,t=map(int,input().split())
print(t-s+1)
解答例
a,b,c=map(int,input().split())
print("YES" if b-a==c-b else "NO")
解答例
x,t=map(int,input().split())
print("0" if x-t<0 else x-t)
解答例
print(int(input())**2-int(input()))
解答例
print(-int(input())+2*int(input()))
解答例
print(48-int(input()))
ABC103 A - Task Scheduling Problem
解答例
A=list(map(int,input().split()))
print(max(A)-min(A))
解答例
a,b,c=sorted(map(int,input().split()))
print(c-a)
解答例
a,b=map(int,input().split())
print(a*b-(a+b-1))
解答例
n,i=map(int,input().split())
print(n-i+1)
解答例
a,b,c,d,e,k=[int(input()) for i in range(6)]
print("Yay!" if abs(a-e)<=k else ":(")
解答例
A,B,C=map(int,input().split())
d=C-(A-B)
print(d if d>0 else 0)
解答例
A,B,C=map(int,input().split())
print(max(0,C-A+B))
乗算
サンプルコード
print(3*2)
実行結果
6
解答例
print(int(input())*2)
解答例
s1,e1=map(int,input().split())
s2,e2=map(int,input().split())
s3,e3=map(int,input().split())
print((s1*e1+s2*e2+s3*e3)//10)
解答例
print((["Bad"]*6+["Good"]*3+["Great"]+["Perfect"])[int(input())//10])
解答例
a,d=map(int,input().split())
print(a*d+max(a,d))
解答例
a,d=map(int,input().split())
print(max(a,d)*(min(a,d)+1))
解答例
a,b,c,d=map(int,input().split())
print(max(a*b,c*d))
解答例
n,m=map(int,input().split())
print((n-1)*(m-1))
解答例
print(-int(input())+2*int(input()))
解答例
a,b=map(int,input().split())
print(a*b-(a+b-1))
サンプルコード
H,W=map(int,input().split())
h,w=map(int,input().split())
print((H-h)*(W-w))
サンプルコード
K,X=map(int,input().split())
print("Yes" if 500*K>=X else "No")
階乗
- math.factorial()：階乗を整数で返す
サンプルコード
import math
print(math.factorial(5))
実行結果
120
解答例
import math
print(math.factorial(int(input()))%(10**9+7))
解答例
ans=1
n=int(input())
for i in range(1,n+1):
ans*=i
ans=ans%(10**9+7)
print(ans)
べき乗
サンプルコード
print(2**3)
実行結果
8
サンプルコード
print(10**9+7)
実行結果
1000000007
解答例
print(2**(len(bin(int(input())))-3))
解答例
n=int(input())
ans=1
for i in range(7):
if 2**i<=n:ans=2**i
print(ans)
解答例
print(10**int(input())+7)
解答例
x=int(input())
c=1
for b in range(1,x):
for p in range(2,x):
if b**p<=x:c=max(c,b**p)
else:break
print(c)
解答例
print(int(input())**2-int(input()))
解答例
print(3*int(input())**2)
解答例
print(int(input())**3)
解答例
print(int(input())**2)
べき根
サンプルコード
print(4**0.5)
実行結果
2.0
サンプルコード
print(4**.5)
実行結果
2.0
サンプルコード
print(4**(1/2))
実行結果
2.0
サンプルコード
print(8**(1/3))
実行結果
2.0
解答例
print(int(int(input())**.25))
解答例
print(int(int(input())**.5)**2)
解答例
print("No" if int(input().replace(" ",""))**.5%1 else "Yes")
解答例
c=int(input().replace(" ",""))
print("Yes" if c==int(c**.5)**2 else "No")
除算
サンプルコード
print(6/3)
実行結果
2.0
解答例
N=int(input())
v=sorted(map(int,input().split()))
value=v[0]
for i in range(1,N):
value=(value+v[i])/2
print(value)
ABC138 B - Resistors in Parallel
解答例
input()
A=map(int,input().split())
print(1/sum(1/a for a in A))
小数点切り捨て除算
サンプルコード
print(5//3)
実行結果
1
サンプルコード
print(6//3)
実行結果
2
解答例
x,y,z=map(int,input().split())
print((x-z)//(y+z))
解答例
n,x=map(int,input().split())
l=[int(input()) for _ in range(n)]
print(n+((x-sum(l))//min(l)))
解答例
N,A,B=map(int,input().split())
div=N//(A+B)
mod=N%(A+B)
print(div*A+min(mod,A))
解答例
x,y=map(int,input().split())
print(y//x)
解答例
a,b,c,d=map(int,input().split())
T,A=b/a,d/c
print("TAKAHASHI" if T>A else "AOKI" if T<A else "DRAW")
解答例
n=int(input())
print(n*800-n//15*200)
解答例
print(int(input())//3)
解答例
print(int(input())**2//4)
解答例
k=int(input())
print((k//2)*((k+1)//2))
解答例
x,y=map(int,input().split())
print(x+y//2)
解答例
a,b,c=map(int,input().split())
print(a*b//2)
ABC117 A - Entrance Examination
解答例
print(eval(input().replace(" ","/")))
解答例
a,p=map(int,input().split())
print((3*a+p)//2)
解答例
N=int(input())
print("0.5" if N%2==0 else ((N//2)+1)/N)
解答例
N=int(input())
print((N-N//2)/N)
解答例
H,A=map(int,input().split())
print(H//A if H%A==0 else H//A+1)
解答例
H,A=map(int,input().split())
print((H+A-1)//A)
小数点切り上げ除算
- math.ceil()：引数以上の最小の整数を返す
サンプルコード
import math
print(math.ceil(5/3))
実行結果
2
ABC123 C - Five Transportations
解答例
import math
n=int(input())
l=[int(input()) for i in range(5)]
print(math.ceil(n/min(l))+4)
解答例
import math
N=int(input())
A=list(map(int,input().split()))
print(math.ceil(sum(A)/(N-A.count(0))))
解答例
from itertools import permutations
import math
menu=5
times=[int(input()) for i in range(menu)]
orders=list(permutations(times,menu))
minimum=sum(math.ceil(time/10)*10 for time in times)
for order in orders:
total=order[0]
for i in range(1,menu):
total+=math.ceil(order[i]/10)*10
minimum=min(minimum,total)
print(minimum)
解答例
import math
N,D=map(int,input().split())
print(math.ceil(N/(D*2+1)))
解答例
import math
A,B=map(int,input().split())
print(1+math.ceil((B-A)/(A-1)))
解答例
import math
A,B=map(int,input().split())
print(math.ceil(B/A))
解答例
import math
a,b=map(int,input().split())
print(math.ceil((a+b)/2))
解答例
import math
N=int(input())
print(math.ceil(N/2))
解答例
N=int(input())
print(N//2+N%2)
解答例
N=int(input())
print((N+1)//2)
剰余
サンプルコード
print(7%3)
実行結果
1
解答例
N=int(input())
ans=N
for i in range(N+1):
cnt=0
t=i
while t>0:
cnt+=t%6
t//=6
j=N-i
while j>0:
cnt+=j%9
j//=9
ans = min(ans,cnt)
print(ans)
ABC133 C - Remainder Minimization 2019
解答例
L,R=map(int,input().split())
if R-L>=2019:
minimum=0
else:
left,right=L%2019,R%2019
minimum=2018**2
for i in range(left,right+1):
for j in range(i+1,right+1):
minimum=min(minimum,i*j%2019)
print(minimum)
解答例
L,R=map(int,input().split())
left,right=L%2019,R%2019
print(0 if R-L>=2019 else min(i*j%2019 for i in range(left,right+1) for j in range(i+1,right+1)))
解答例
n=int(input())
print(n+1 if n%2 else n-1)
解答例
print(eval(input().replace(" ","*"))%(10**9+7))
解答例
a,b,c=map(int,input().split())
print("YES" if any((a*i)%b==c for i in range(1,b+1)) else "NO")
解答例
N=int(input())
S=input()
print("".join(chr(65+(ord(s)-65+N)%26) for s in S))
解答例
N=int(input())
print(N//10*100+min(100,N%10*15)
解答例
n=int(input())
print(n//2+n%2)
解答例
print(int(input())%12+1)
解答例
print(-int(input())%int(input()))
解答例
a=int(input())
b=int(input())
print((((a//b)+1)*b-a) if a%b!=0 else "0")
解答例
m,d=map(int,input().split())
print("NO" if m%d else "YES")
解答例
w,h=map(int,input().split())
print("16:9" if w*h%144==0 else "4:3")
解答例
a,b=map(int,input().split())
print("Draw" if a==b else "Bob" if (a+13)%15<(b+13)%15 else "Alice")
解答例
print(sum(map(int,input().split()))%24)
解答例
print("NO" if int(input()[::2])%4 else "YES")
解答例
a,b=map(int,input().split())
print("Impossible" if a*b*(a+b)%3 else "Possible")
解答例
print(int(input())%9)
解答例
a,b=map(int,input().split())
print("Odd" if a*b%2 else "Even")
解答例
x,a,b=[int(input()) for _ in range(3)]
print((x-a)%b)
解答例
N=int(input())
A=int(input())
print("Yes" if N%500<=A else "No")
ABC102 A - Multiple of 2 and N
解答例
n=int(input())
print(n+n%2*n)
解答例
N=int(input())
print(N if N%2==0 else N*2)
解答例
print((eval(input().replace(" ","%"))>0)+0)
解答例
N,K=map(int,input().split())
print(0 if N%K==0 else 1)
解答例
a,b=map(int,input().split())
print(b-a if b%a>0 else a+b)
解答例
A,B=map(int,input().split())
print((A+B)//2 if (A+B)%2==0 else "IMPOSSIBLE")
解答例
H,A=map(int,input().split())
print(H//A if H%A==0 else H//A+1)
解答例
H,A=map(int,input().split())
print((H+A-1)//A)
解答例
N=int(input())
print(N//2+N%2)
解答例
N=int(input())
print((N+1)//2)
解答例
N,A,B=map(int,input().split())
div=N//(A+B)
mod=N%(A+B)
print(div*A+min(mod,A))
2.2 代入文
サンプルコード
a,b=1,2
print(a,b)
a,b=b,a
print(a,b)
実行結果
1 2
2 1
解答例
n=int(input())
a,b,c=0,0,1
for i in range(n-1):
a,b,c=b,c,(a+b+c)%10007
print(a)
解答例
N=int(input())
a,b=2,1
for i in range(N):
a,b=b,a+b
print(a)
2.3 累算代入文
| 演算子 | 記述例 | 等式 |
|---|---|---|
| += | a+=b | a=a+b |
| -= | a-=b | a=a-b |
| *= | a*=b | a=a*b |
| **= | a**=b | a=a**b |
| /= | a/=b | a=a/b |
| //= | a//=b | a=a//b |
| %= | a%=b | a=a%b |
2.4 論理演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| and | a and b | a も b も真であれば真 |
| or | a or b | a または b が真であれば真 |
| not | not a | a が偽であれば真 |
解答例
Y=int(input())
print("YES" if Y%4==0 and Y%100!=0 or Y%400==0 else "NO")
解答例
a,b=map(int,input().split())
c,d=map(int,input().split())
print("YES" if a==c or a==d or b==c or b==d else "NO")
解答例
N=int(input())
A=list(map(int,input().split()))
flag=True
for i in range(N):
if A[i]%2==0 and A[i]%3!=0 and A[i]%5!=0:flag=False
print("APPROVED" if flag else "DENIED")
解答例
input()
A=list(map(int,input().split()))
print("APPROVED" if all([a%2!=0 or a%3==0 or a%5==0 for a in A]) else "DENIED")
解答例
input()
A=list(map(int,input().split()))
print("APPROVED" if all(a&1 or a%3==0 or a%5==0 for a in A) else "DENIED")
解答例
a,b,x=map(int,input().split())
print("YES" if a<=x and x<=a+b else "NO")
ABC097 A - Colorful Transceivers
解答例
a,b,c,d=map(int,input().split())
print("Yes" if abs(c-a)<=d or abs(b-a)<=d and abs(c-b)<=d else "No")
解答例
a,b=map(int,input().split())
print("Yay!" if a<9 and b<9 else ":(")
Tenka1 Programmer Beginner Contest 2019 A - On the Way
解答例
a,b,c=map(int,input().split())
print("Yes" if a<c<b or b<c<a else "No")
2.5 比較演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| == | a==b | a が b と等しい |
| != | a!=b | a が b と等しくない |
| > | a>b | a が b より大きい |
| >= | a>=b | a が b 以上 |
| < | a<b | a が b より小さい |
| <= | a<=b | a が b 以下 |
解答例
s=sorted(input())
t=sorted(input())[::-1]
print("Yes" if s<t else "No")
解答例
S=input()
print("Heisei" if S<="2019/04/30" else "TBD")
2.6 所属検査演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| in | a in b | a が b に含まれる |
| not in | a not in b | a が b に含まれない |
2.7 ブール演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| or | x or y | x が偽なら y、そうでなければ x |
| and | x and y | x が偽なら x、そうでなければ y |
| not | not x | x が偽なら True、そうでなければ False |
ブール値を数値に変換
サンプルコード
print(int(True))
実行結果
1
サンプルコード
print(int(False))
実行結果
0
サンプルコード
print(-(3>2))
実行結果
-1
サンプルコード
print(-(3<2))
実行結果
0
サンプルコード
print(True+0)
実行結果
1
サンプルコード
print(False+0)
実行結果
0
解答例
W,H,x,y=map(int,input().split())
print(W*H/2,int(W==2*x and H==2*y))
解答例
print(sum("r" in input() for i in range(12)))
解答例
n,s,t=map(int,input().split())
w=c=0
for i in range(n):
w+=int(input())
c+=s<=w<=t
print(c)
解答例
a,b,c,k=map(int,input().split())
s,t=map(int,input().split())
print(a*s+b*t-(s+t)*c*(s+t>=k))
解答例
a,b=map(int,input().split())
print(a-(a>b))
解答例
print((eval(input().replace(" ","%"))>0)+0)
解答例
X,A=map(int,input().split())
print(10*(X>=A))
2.8 ビット演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| & | a&b | a と b のビット単位の AND を与える |
| ^ | a^b | a と b のビット単位の XOR（排他的OR）を与える |
| | | a|b | a と b のビット単位の OR を与える |
| ~ | ~a | a（整数）のビット単位反転を与える |
| 演算 | 結果 |
|---|---|
| bin(0b1010&0b0110) | 0b10 |
| bin(0b1010^0b0110) | 0b1100 |
| bin(0b1010|0b0110) | 0b1110 |
サンプルコード
print(bin(0b1010&0b0110))
print(bin(0b1010^0b0110))
print(bin(0b1010|0b0110))
実行結果
0b10
0b1100
0b1110
解答例
input()
A=list(map(int,input().split()))
print("APPROVED" if all(a&1 or a%3==0 or a%5==0 for a in A) else "DENIED")
ビット単位反転（bitwise invert）：xのビット単位反転は-(x+1)として定義されている
サンプルコード
print(~2)
実行結果
-3
サンプルコード
print(~-4)
実行結果
3
サンプルコード
print(~(5-3))
実行結果
-3
解答例
print(~eval(input().replace(" ","-"))+2)
解答例
a,b=map(int,input().split())
print(~-a*~-b)
解答例
a,b=map(int,input().split())
print(~-a*~-b)
2.9 シフト演算
| 演算子 | 記述例 | 意味 |
|---|---|---|
| << | a<<b | a を左に b ビットシフト |
| >> | a>>b | a を右に b ビットシフト |
左にビットシフト
| 演算 | 結果 | 10進数 |
|---|---|---|
| bin(2) | 0b10 | 2 |
| bin(2<<1) | 0b100 | 4 |
| bin(2<<2) | 0b1000 | 8 |
| bin(2<<3) | 0b10000 | 16 |
| bin(2<<4) | 0b100000 | 32 |
サンプルコード
print(2)
print(2<<1)
print(2<<2)
print(2<<3)
print(2<<4)
print(bin(2))
print(bin(2<<1))
print(bin(2<<2))
print(bin(2<<3))
print(bin(2<<4))
実行結果
2
4
8
16
32
0b10
0b100
0b1000
0b10000
0b100000
右にビットシフト
| 演算 | 結果 | 10進数 |
|---|---|---|
| bin(10) | 0b1010 | 10 |
| bin(10>>1) | 0b101 | 5 |
| bin(10>>2) | 0b10 | 2 |
| bin(10>>3) | 0b1 | 1 |
| bin(10>>4) | 0b0 | 0 |
サンプルコード
print(10)
print(10>>1)
print(10>>2)
print(10>>3)
print(10>>4)
print(bin(10))
print(bin(10>>1))
print(bin(10>>2))
print(bin(10>>3))
print(bin(10>>4))
実行結果
10
5
2
1
0
0b1010
0b101
0b10
0b1
0b0
解答例
print(["ABC","ARC","AGC"][int(input())//50+8>>5])
解答例
print(int(input())**2>>2)
第3章 制御フロー観点
3.1 条件式
if/else文
解答例
Y=int(input())
print("YES" if Y%4==0 and Y%100!=0 or Y%400==0 else "NO")
解答例
W,H,x,y=map(int,input().split())
print(W*H/2,1) if W==2*x and H==2*y else print(W*H/2,0)
解答例
W,H,x,y=map(int,input().split())
print(W*H/2,int(W==2*x and H==2*y))
解答例
a,b,c=map(int,input().split())
print("?" if a+b==c and a-b==c else "+" if a+b==c else "-" if a-b==c else "!")
解答例
l,w=map(int,input().split())
n=int(input())
for i in range(n):
a=int(input())
print(l-a if l>a else "0" if w>=a else "-1")
解答例
a=int(input())
b=int(input())
print("GREATER" if a>b else "LESS" if a<b else "EQUAL")
解答例
A,B,K=map(int,input().split())
print(A-K,B) if K<=A else print(0,B-(K-A)) if K<=A+B else print(0,0)
解答例
print("ABC" if input()=="1" else "chokudai")
解答例
n=int(input())
print("Bad" if n<60 else "Good" if n<90 else "Great" if n<100 else "Perfect")
解答例
a,b,c,d=map(int,input().split())
T,A=b/a,d/c
print("TAKAHASHI" if T>A else "AOKI" if T<A else "DRAW")
解答例
x,y=map(int,input().split())
print("Better" if x<y else "Worse")
解答例
print("ABC" if int(input()) < 1200 else "ARC")
解答例
a,b=input().split()
print("H" if a==b else "D")
解答例
x,a,b=map(int,input().split())
print("dangerous" if x<b-a else "safe" if a<b else "delicious")
解答例
x,t=map(int,input().split())
print("0" if x-t<0 else x-t)
解答例
a,b,c=map(int,input().split())
print(a if b==c else b if a==c else c)
解答例
x,y=input().split()
print("<" if x<y else "=" if x==y else ">")
解答例
a,b,c,d=map(int,input().split())
print("Left" if a+b>c+d else "Balanced" if a+b==c+d else "Right")
解答例
a,b,c=map(int,input().split())
print("Yes" if a+b>=c else "No")
解答例
a,b=map(int,input().split())
print(a-1 if a>b else a)
解答例
print("ABC" if int(input())<1000 else "ABD")
解答例
r=int(input())
print("ABC" if r<1200 else "ARC" if r<2800 else "AGC")
ABC112 A - Programming Education
解答例
n=int(input())
print("Hello World" if n%2!=0 else int(input())+int(input()))
解答例
b=input()
print("A" if b=="T" else "T" if b=="A" else "G" if b=="C" else "C")
解答例
X,A=map(int,input().split())
print("0" if X<A else "10")
解答例
a=int(input())
s=input()
print(s if a>=3200 else "red")
解答例
S=input()
print("Sunny" if S=="Rainy" else "Cloudy" if S=="Sunny" else "Rainy")
サンプルコード
K,X=map(int,input().split())
print("Yes" if 500*K>=X else "No")
解答例
N,M=input().split()
print("Yes" if N==M else "No")
解答例
S,T=input().split()
A,B=map(int,input().split())
U=input()
print(A-1,B) if S==U else print(A,B-1)
解答例
N,R=map(int,input().split())
print(R if N>=10 else R+100*(10-N))
解答例
N,R=map(int,input().split())
print(R+100*max(0,10-N))
解答例
N=int(input())
print(N//2 if N%2==0 else N//2+1)
Tenka1 Programmer Beginner Contest 2019 A - On the Way
解答例
a,b,c=map(int,input().split())
print("Yes" if a<c<b or b<c<a else "No")
if/in文
サンプルコード
String="abc"
print("a" in String)
実行結果
True
サンプルコード
String="abc"
print("d" in String)
実行結果
False
サンプルコード
List=["a","b","c"]
print("a" in List)
実行結果
True
サンプルコード
List=["a","b","c"]
print("d" in List)
実行結果
False
サンプルコード
List=["a","b","c"]
print("a" not in List)
実行結果
False
サンプルコード
List=["a","b","c"]
print("d" not in List)
実行結果
True
解答例
S=map(str,input().split("+"))
ans=0
for s in S:
if "0" not in s:ans+=1
print(ans)
解答例
print(sum("r" in input() for i in range(12)))
解答例
input()
l=map(str,input().split())
print("Three" if len(set(l))==3 else "Four")
解答例
input()
print("Four" if "Y" in input() else "Three")
解答例
print("Yes" if input() in input()*2 else "No")
解答例
S=input()
m=0
count=0
for s in S:
if s in "A":count+=1
elif s in "C":count+=1
elif s in "G":count+=1
elif s in "T":count+=1
else:count=0
m=max(m,count)
print(m)
解答例
print("YES" if input() in "369" else "NO")
ABC049 A - 居合を終え、青い絵を覆う / UOIAUAI
解答例
print("vowel" if input() in "aiueo" else "consonant")
解答例
print("Yes" if "9" in input() else "No")
解答例
print("No" if "2" in input() else "Yes")
解答例
print("YES" if input() in "753" else "NO")
複数条件式
ABC061 A - Between Two Integers
解答例
a,b,c=map(int,input().split())
print("Yes" if a<=c<=b else "No")
解答例
a,b,c,d=input()
print("Yes" if a==b==c or b==c==d else "No")
解答例
a,b,c,d=input()
print("Yes" if a==b==c or b==c==d else "No")
解答例
a,b,x=map(int,input().split())
print("YES" if a<=x<=a+b else "NO")
解答例
a,b,x=map(int,input().split())
print("YES" if 0<=x-a<=b else "NO")
解答例
S=input()
print("Good" if S[0]!=S[1]!=S[2]!=S[3] else "Bad")
解答例
a,b,c,d=input()
print("Good" if a!=b!=c!=d else "Bad")
解答例
A,B=map(int,input().split())
print(A*B if 1<=A<=9 and 1<=B<=9 else "-1")
3.2 繰り返し文
for文
解答例
N,M,A,B=map(int,input().split())
for m in range(M):
if N<=A:
N+=B
c=int(input())
N-=c
if N<0:
print(m+1)
break
else:
print("complete")
解答例
N,K=map(int,input().split())
t=[int(input()) for i in range(N)]
for i in range(3,N):
if sum(t[i-3:i])<K:
print(i)
break
else:
print(-1)
解答例
N,Y=map(int,input().split())
answer=-1,-1,-1
for x in range(N+1):
for y in range(N-x+1):
z=N-x-y
if 0<=z<=2000 and 10000*x+5000*y+1000*z==Y:
answer=x,y,z
break
else:
continue
break
print(*answer)
解答例
N,Y=map(int,input().split())
for x in range(N+1):
for y in range(N-x+1):
z=N-x-y
if 0<=z<=2000 and 10000*x+5000*y+1000*z==Y:
print(x,y,z)
exit()
print(-1,-1,-1)
N=int(input())
H=list(map(int,input().split()))
base=0
for h in H:
if base>h:
print("No")
break
base=max(base,h-1)
else:
print("Yes")
ABC144 C - Walk on Multiplication Table
解答例
N=int(input())
move=10**12
for n in range(1,int(N**.5)+1):
if N%n==0:
move=min(move,N//n+n-2)
print(move)
解答例
N,M=map(int,input().split())
sc=[list(map(int,input().split())) for m in range(M)]
for i in range(10**N):
if len(str(i))==N and all(str(i)[s-1]==str(c) for s,c in sc):
print(i)
break
else:
print(-1)
解答例
A,B=map(int,input().split())
for money in range(1,1001):
if int(money*0.08)==A and int(money*0.1)==B:
print(money)
break
else:
print(-1)
N=int(input())
for i in range(1,10):
m=N/i
if m.is_integer() and 1<=m<=9:
print("Yes")
break
else:
print("No")
while文
ABC144 C - Walk on Multiplication Table
解答例
N=int(input())
move=int(N**.5)
while N%move!=0:
move-=1
print(N//move+move-2)
解答例
X=int(input())
while True:
for x in range(2,X):
if X%x==0:
break
else:
print(X)
break
X+=1
解答例
X=int(input())
num=2
while num<X:
if X%num==0:
X+=1
num=2
else:
num+=1
print(X)
解答例
X=int(input())
for x in range(X,2*X):
for i in range(2,int(x**.5)+1):
if x%i==0:
break
else:
print(x)
break
解答例
X=int(input())
for x in range(X,2*X):
flag=True
for i in range(2,int(x**.5)+1):
if x%i==0:
flag=False
break
if flag:
print(x)
exit()
ABC045 B - 3人でカードゲームイージー / Card Game for Three (ABC Edit)
解答例
S={c:list(input()) for c in "abc"}
s="a"
while S[s]:s=S[s].pop(0)
print(s.upper())
解答例
n=int(input())
a=[int(input()) for i in range(n)]
c,s=1,a[0]
while s!=2 and c<n:c,s=c+1,a[s-1]
print(c if c<n else -1)
解答例
s=list(input())
while len(s)!=0:
s.pop()
if len(s)%2==0 and s[:len(s)//2]==s[len(s)//2:]:
print(len(s))
exit()
解答例
input()
A=list(map(int,input().split()))
ans=0
while all(a%2==0 for a in A):
A=[a/2 for a in A]
ans+=1
print(ans)
解答例
S=int(input())
l=[]
while (S not in l):
l.append(S)
if S%2==0:S//=2
else:S=3*S+1
print(len(l)+1)
解答例
N,K=map(int,input().split())
count=0
while(N>0):
N//=K
count+=1
print(count)
解答例
a,b,n=[int(input()) for i in range(3)]
while n%a!=0 or n%b!=0:n+=1
print(n)
第4章 組み込み型観点
4.1 int
サンプルコード
print(int(2.9))
実行結果
2
4.2 float
サンプルコード
print(5e10)
実行結果
50000000000.0
サンプルコード
print(1.1e-3)
実行結果
0.0011
解答例
N=int(input())
ans=0
for i in range(N):
x,u=input().split()
if u=="BTC":ans+=float(x)*380000
else:ans+=int(x)
print(ans)
解答例
H,B=map(float,input().split())
print(H**2*B/1e4)
整数判定
-
float.is_integer()：浮動小数点数が整数値なら
Trueを、そうでなければFalseを返す
サンプルコード
x=1.0
print(x.is_integer())
実行結果
True
サンプルコード
x=1.1
print(x.is_integer())
実行結果
False
N=int(input())
for i in range(1,10):
m=N/i
if m.is_integer() and 1<=m<=9:
print("Yes")
break
else:
print("No")
4.3 str
- テキストシーケンス型
- str：イミュータブル（変更不可能）なシーケンス
- 文字列メソッド
文字列の結合
サンプルコード
print("foo"+"bar")
実行結果
foobar
解答例
print(input()+"pp")
解答例
print(input()+"s")
解答例
print("ABC"+input())
print(input()[0]+input()[1]+input()[2])
a=input()[0]
b=input()[1]
c=input()[2]
print(a+b+c)
解答例
S,T=input().split()
print(T+S)
文字列の繰り返し
解答例
a,b=input().split()
print(2*int(a+b))
解答例
a,b=input().split()
print(a*int(b) if a<b else b*int(a))
解答例
a,b=input().split()
print(min(a*int(b),b*int(a)))
解答例
print("x"*len(input()))
ABC115 A - Christmas Eve Eve Eve
解答例
print("Christmas"+" Eve"*(25-int(input())))
文字列の参照
サンプルコード
String="abc"
print(String[0],String[1],String[2])
実行結果
a b c
サンプルコード
String="abc"
print(String[-1],String[-2],String[-3])
実行結果
c b a
解答例
s=input()
print(s[0]+str((len(s)-2))+s[-1])
解答例
s=input()
n=int(input())-1
print(s[n//5]+s[n%5])
解答例
print("YES" if input()[-1]=="T" else "NO")
ABC048 A - AtCoder *** Contest
解答例
print("A%sC"%input()[8])
解答例
print("A"+input()[8]+"C")
ABC059 A - Three-letter acronym
解答例
a,b,c=input().split()
print((a[0]+b[0]+c[0]).upper())
解答例
a,b,c=input().split()
print("YES" if a[-1]==b[0] and b[-1]==c[0] else "NO")
解答例
S="XACABABAABABA"
x,y=map(int,input().split())
print("Yes" if S[x]==S[y] else "No")
解答例
n=input()
print("Yes" if n[0]==n[2] else "No")
解答例
n=input()
print("Yes" if n[0]==n[1]==n[2] or n[1]==n[2]==n[3] else "No")
print(input()[0]+input()[1]+input()[2])
a=input()[0]
b=input()[1]
c=input()[2]
print(a+b+c)
S=input()
print("Good" if S[0]!=S[1] and S[1]!=S[2] and S[2]!=S[3] else "Bad")
S=input()
print("Good" if S[0]!=S[1]!=S[2]!=S[3] else "Bad")
S=input()
T=input()
count=0
if S[0]==T[0]:count+=1
if S[1]==T[1]:count+=1
if S[2]==T[2]:count+=1
print(count)
文字列の逆順
サンプルコード
String="abc"
print(String[::-1])
実行結果
cba
ABC090 B - Palindromic Numbers
解答例
a,b=map(int,input().split())
print(sum(i==i[::-1] for i in map(str,range(a,b+1))))
解答例
a,b=map(int,input().split())
print(len([i for i in map(str,range(a,b+1)) if i==i[::-1]]))
解答例
print(*input().split()[::-1])
解答例
n=input()
print("Yes" if n==n[::-1] else "No")
解答例
print("YES" if input()==input()[::-1] else "NO")
文字列の比較
解答例
print("Yes" if sorted(input())<sorted(input())[::-1] else "No")
解答例
S=input()
print(sum(S[i]!=S[-i-1] for i in range(len(S)//2)))
解答例
N=int(input())
S=input()
print(sum(S[i:i+3]=="ABC" for i in range(N-2)))
解答例
print("Heisei" if input()<="2019/04/30" else "TBD")
解答例
S=input()
T=input()
print(sum(s==t for s,t in zip(S,T)))
文字列の置換
- str.replace()：文字列の置換
サンプルコード
String="aabbccbbaabbaacc"
print(String.replace("bb","xx"))
実行結果
aaxxccxxaaxxaacc
解答例
S=input()
for b,a in zip("ODIZSB","001258"):
S=S.replace(b,a)
print(S)
解答例
S=input()
b="ODIZSB"
a="001258"
for i in range(6):
S=S.replace(b[i],a[i])
print(S)
解答例
print(input().replace("O","0").replace("D","0").replace("I","1").replace("Z","2").replace("S","5").replace("B","8"))
解答例
print(input().replace("Left","<").replace("Right",">").replace("AtCoder","A"))
解答例
print("YES" if input().replace("ch","").replace("o","").replace("k","").replace("u","")=="" else "NO")
解答例
print(2*int(input().replace(" ","")))
解答例
print(eval(input().replace(" ","*"))%(10**9+7))
解答例
s=input()
print("AC" if "C" in s[2:-1] and "A" in s and s[1:].replace("C","",1).islower() else "WA")
解答例
print(sum(eval(input().replace(" ","*")) for i in range(3))//10)
解答例
print(eval(input().replace(" ","^")))
解答例
print("2018"+input()[4:])
解答例
print(eval(input().replace(" ","-"))+1)
解答例
print("No" if eval(input().replace(" ","*"))%2==0 else "Yes")
ABC111 A - AtCoder Beginner Contest 999
解答例
print(input().replace("1","x").replace("9","1").replace("x","9"))
ABC117 A - Entrance Examination
解答例
print(eval(input().replace(" ","/")))
文字列のコピー
解答例
s=input()
for i in range(int(input())):
l,r=map(int,input().split())
s=s[:l-1]+s[l-1:r][::-1]+s[r:]
print(s)
ABC043 B - バイナリハックイージー / Unhappy Hacking (ABC Edit)
解答例
t=""
for x in input():
if x=="B":t=t[:-1]
else:t+=x
print(t)
解答例
s=input()[:-1]
while s[:len(s)//2]!=s[len(s)//2:]:s=s[:-1]
print(len(s))
文字列の判定
サンプルコード
String="abc"
[print(chr(i),end=" ") for i in range(ord("a"),ord("z")+1) if chr(i) in String]
実行結果
a b c
サンプルコード
String="abc"
[print(chr(i),end=" ") for i in range(ord("a"),ord("z")+1) if chr(i) not in String]
実行結果
d e f g h i j k l m n o p q r s t u v w x y z
解答例
print(min(set(map(chr,range(97,123)))-set(input())or["None"]))
解答例
letters="abcdefghijklmnopqrstuvwxyz"
ans=sorted(set(letters)^set(input()))
print("None" if len(ans)==0 else ans[0])
ABC090 B - Palindromic Numbers
解答例
a,b=map(int,input().split())
print(sum(i==i[::-1] for i in map(str,range(a,b+1))))
解答例
a,b=map(int,input().split())
print(len([i for i in map(str,range(a,b+1)) if i==i[::-1]]))
解答例
x,y=input().split()
print("<" if x<y else "=" if x==y else ">")
- str.isdecimal()：全ての文字が十進数字なら真、そうでなければ偽（半角・全角のアラビア数字が真）
- str.isdigit()：全ての文字が数字なら真、そうでなければ偽（半角・全角のアラビア数字、特殊数字が真）
- str.isnumeric()：全ての文字が数を表す文字なら真、そうでなければ偽（半角・全角のアラビア数字、特殊数字、漢数字が真）
- str.isalpha()：全ての文字が英字なら真、そうでなければ偽（便宜上「英字」と書いているが、平仮名やカタカナ、漢字なども真）
- str.isalnum()：全ての文字が英数字なら真、そうでなければ偽（各文字が上のメソッドで真となれば真）
isdecimal
サンプルコード
s="1234567890"
print(s.isdecimal())
実行結果
True
解答例
a,b=map(int,input().split())
s=input()
if 1<=a<=5 and 1<=b<=5:
if s[a]=="-":
if s[0:a].isdecimal() and s[a+1:a+b+1].isdecimal():
print("Yes")
exit()
print("No")
isdigit
サンプルコード
s="\u00B2" # 2乗
print(s.isdigit())
実行結果
True
解答例
print("".join(i for i in input() if i.isdigit()))
isnumeric
サンプルコード
s="一二三四五六七八九〇壱億参阡萬"
print(s.isnumeric())
実行結果
True
isalpha
サンプルコード
s="abcあいうアイウ漢字"
print(s.isalpha())
実行結果
True
isalnum
サンプルコード
s="abc100"
print(s.isalnum())
実行結果
True
文字列の検索
- str.find() / str.rfind() ：文字列の検索
解答例
S=input()[0:12]
key="WBWBWWBWBWBW"*2
ans=["Do","","Re","","Mi","Fa","","So","","La","","Si"]
print(ans[(key.find(S))])
解答例
s=input()
print(s.rfind("Z")-s.find("A")+1)
大文字小文字変換
- str.upper()：すべての文字列を大文字に変換する
- str.lower()：すべての文字列を小文字に変換する
- str.capitalize()：文字列の先頭文字を大文字に変換する
- str.title()：各単語の先頭文字を大文字に変換する
upper
サンプルコード
Text="this is a pen."
print(Text.upper())
実行結果
THIS IS A PEN.
解答例
s=input()
print(s[0].upper()+s[1:].lower())
ABC059 A - Three-letter acronym
解答例
for a in input().upper().split():print(a[0],end="")
解答例
a,b,c=input().split()
print((a[0]+b[0]+c[0]).upper())
lower
サンプルコード
Text="THIS IS A PEN."
print(Text.lower())
実行結果
this is a pen.
解答例
s=input()
print(s[0].upper()+s[1:].lower())
capitalize
サンプルコード
Text="this is a pen."
print(Text.capitalize())
実行結果
This is a pen.
解答例
print("abcd".capitalize())
title
サンプルコード
Text="this is a pen."
print(Text.title())
実行結果
This Is A Pen.
大文字小文字判定
- str.isupper()：大文字判定
- str.islower()：小文字判定
解答例
s=input()
print("AC" if "C" in s[2:-1] and "A" in s and s[1:].replace("C","",1).islower() else "WA")
文字列の変換
- str.maketrans()：変換テーブルを作成する
- str.translate()：変換テーブルに基づいてマッピングする
解答例
print(input().translate(str.maketrans("ODIZSB","001258")))
4.4 list
- シーケンス型
- list：ミュータブル（変更可能）なシーケンス
- ミュータブルなシーケンス型
要素の追加
- list.append()：要素の追加
- ミュータブルなシーケンス型
サンプルコード
List=["a","b","c"]
List.append("d")
print(List)
実行結果
['a', 'b', 'c', 'd']
サンプルコード
List=["a","b","c"]
List.append(["d","e"])
print(List)
実行結果
['a', 'b', 'c', ['d', 'e']]
要素の拡張
- list.extend()：要素の拡張
- ミュータブルなシーケンス型
サンプルコード
List=["a","b","c"]
List.extend(["d","e"])
print(List)
実行結果
['a', 'b', 'c', 'd', 'e']
要素の挿入
- list.insert()：要素の挿入
- ミュータブルなシーケンス型
サンプルコード
List=["a","b","c"]
List.insert(1,"z")
print(List)
実行結果
['a', 'z', 'b', 'c']
サンプルコード
List=["a","b","c"]
List.insert(1,["x","y"])
print(List)
実行結果
['a', ['x', 'y'], 'b', 'c']
解答例
S=list(input())
A,B,C,D=map(int,input().split())
S.insert(D,"\"")
S.insert(C,"\"")
S.insert(B,"\"")
S.insert(A,"\"")
print(*S,sep="")
要素の探索
- list.index()：要素の探索
- 共通のシーケンス演算
サンプルコード
List=["a","b","c","d","e","f"]
print(List.index("c"))
実行結果
2
サンプルコード
List=["a","b","c","d","c","c"]
print(List.index("c"))
実行結果
2
解答例
print((1/int(input()))*sum("FDCBA".index(r) for r in input()))
解答例
N=int(input())
print(sum(map(int,input().translate(str.maketrans("FDCBA","01234"))))/N)
解答例
s=input()
print(len(s)-s[::-1].index("Z")-s.index("A"))
解答例
n=int(input())
S=[]
P=[]
for i in range(n):
s,p=input().split()
S.append(s)
P.append(int(p))
if max(P)>sum(P)/2:print(S[P.index(max(P))])
else:print("atcoder")
解答例
n=int(input())
t,a=map(int,input().split())
h=list(map(int,input().split()))
z=[abs(t-x*0.006-a) for x in h]
print(z.index(min(z))+1)
解答例
print("0ABCDE".index(input()))
解答例
print(["A","B","C","D","E"].index(input())+1)
解答例
X=[int(input()) for i in range(3)]
for x in X:print(3-sorted(X).index(x))
解答例
l=[int(input()) for _ in range(3)]
s=sorted(l)[::-1]
for i in l:
print(s.index(i)+1)
解答例
weather=["Sunny","Cloudy","Rainy"]
S=input()
print(weather[(weather.index(S)+1)%3])
解答例
weather=["Sunny","Cloudy","Rainy"]
S=input()
print(weather[weather.index(S)-2])
ABC146 A - Can't Wait for Holiday
解答例
week=["SUN","MON","TUE","WED","THU","FRI","SAT"]
print(7-week.index(input()))
[ABC151 A - Next Alphabet](https://atcoder.jp/contests/abc151/tasks/abc151_a
解答例
s="abcdefghijklmnopqrstuvwxyz"
print(s[s.index(input())+1])
要素の取り出し
- list.pop()：要素の取り出し
- ミュータブルなシーケンス型
サンプルコード
List=["a","b","c","d","e","f"]
print(List.pop(1))
print(List)
実行結果
b
['a', 'c', 'd', 'e', 'f']
サンプルコード
List=["a","b","c","d","e","f"]
print(List.pop())
print(List)
実行結果
f
['a', 'b', 'c', 'd', 'e']
ABC045 B - 3人でカードゲームイージー / Card Game for Three (ABC Edit)
解答例
S={c:list(input()) for c in "abc"}
s="a"
while S[s]:s=S[s].pop(0)
print(s.upper())
要素の取り除き
- list.remove()：要素の取り除き
- ミュータブルなシーケンス型
サンプルコード
List=["a","b","c","d","e","f"]
List.remove("d")
print(List)
実行結果
['a', 'b', 'c', 'e', 'f']
サンプルコード
List=["a","a","b","b","c","c"]
List.remove("b")
print(List)
実行結果
['a', 'a', 'b', 'c', 'c']
要素の削除
- del()：要素の削除
- ミュータブルなシーケンス型
サンプルコード
List=["a","b","c","d","e","f"]
del List[1]
print(List)
実行結果
['a', 'c', 'd', 'e', 'f']
サンプルコード
List=["a","b","c","d","e","f"]
del List[2:4]
print(List)
実行結果
['a', 'b', 'e', 'f']
サンプルコード
List=["a","b","c","d","e","f"]
del List[:]
print(List)
実行結果
[]
要素の出現回数
- list.count()：要素の出現回数
- 共通のシーケンス演算
サンプルコード
List=["a","b","b","c","c","c"]
print(List.count("c"))
実行結果
3
サンプルコード
List=["a","b","b","c","c","c"]
print(List.count("d"))
実行結果
0
解答例
input()
C=input()
print(*sorted(C.count(c) for c in "1234")[::-3])
解答例
input()
C=input()
l=[C.count(c) for c in "1234"]
print(max(l),min(l))
解答例
input()
C=input()
a=C.count("1")
b=C.count("2")
c=C.count("3")
d=C.count("4")
print(max(a,b,c,d),min(a,b,c,d))
ABC124 C - Coloring Colorfully
解答例
s=input()
n=s[::2].count("0")+s[1::2].count("1")
print(min(n,len(s)-n))
解答例
s=input()
print(min(s[::2].count("0")+s[1::2].count("1"),s[::2].count("1")+s[1::2].count("0")))
解答例
import math
N=int(input())
A=list(map(int,input().split()))
print(math.ceil(sum(A)/(N-A.count(0))))
解答例
s=input()
print(s.count("A"),s.count("B"),s.count("C"),s.count("D"),s.count("E"),s.count("F"))
ABC044 B - 美しい文字列 / Beautiful Strings
解答例
s=input()
print("Yes" if all([s.count(i)%2==0 for i in set(s)]) else "No")
ABC091 B - Two Colors Card Game
解答例
n=[input() for _ in range(int(input()))]
m=[input() for _ in range(int(input()))]
l=list(set(n))
print(max(0,max(n.count(l[i])-m.count(l[i]) for i in range(len(l)))))
解答例
input()
print(input().count("ABC"))
ABC042 A - 和風いろはちゃんイージー / Iroha and Haiku (ABC Edition)
解答例
n=input().split()
print("YES" if n.count("5")==2 and n.count("7")==1 else "NO")
解答例
a,b=map(int,input().split())
s=input()
print("Yes" if s[a]=="-" and s.count("-")==1 else "No")
解答例
print(input().count("1"))
解答例
print(700+100*input().count("o"))
ABC101 A - Eating Symbols Easy
解答例
print(2*input().count("+")-4)
解答例
s=input()
print(s.count("+")-s.count("-"))
解答例
S=input()
s=list(set(S))
print("Yes" if len(s)==2 and S.count(s[0])==S.count(s[1])==2 else "No")
要素の連結
- str.join()：要素の連結
サンプルコード
List=["ab","cd","ef"]
print("".join(List))
実行結果
abcdef
解答例
S=list(input().split())
a=[]
for s in S:
if s=="Left":a.append("<")
elif s=="Right":a.append(">")
else:a.append("A")
print(" ".join(a))
ABC148 B - Strings with the Same Length
解答例
N=input()
S,T=input().split()
print("".join(s+t for s,t in zip(S,T)))
解答例
N=input()
S,T=input().split()
print(*[s+t for s,t in zip(S,T)],sep="")
ABC111 A - AtCoder Beginner Contest 999
解答例
print("".join(["9" if x=="1" else "1" for x in input()]))
要素のユニーク化
サンプルコード
List=[2,3,1,2]
print(list(set(List)))
実行結果
[1, 2, 3]
サンプルコード
List=[[1,0],[0,0],[1,1],[1,0],[0,1],[0,0]]
print(list(map(list,set(map(tuple,List)))))
実行結果
[[0, 1], [1, 0], [0, 0], [1, 1]]
ABC009 B - 心配性な富豪、ファミリーレストランに行く。
解答例
n=int(input())
print(sorted(set([int(input()) for i in range(n)]))[-2])
解答例
s=input()
k=int(input())
print(len(set(s[i:i+k] for i in range(len(s)-k+1))))
解答例
s=input()
print("no" if len(s)-len(set(s)) else "yes")
解答例
print(len(set([int(input()) for _ in range(int(input()))])))
解答例
print("DIFFERENT" if len(set(input()))!=1 else "SAME")
ABC046 A - AtCoDeerくんとペンキ / AtCoDeer and Paint Cans
解答例
print(len(set(input().split())))
解答例
print("Yes" if len(set(input()))==3 else "No")
解答例
S=input()
s=list(set(S))
print("Yes" if len(s)==2 and S.count(s[0])==S.count(s[1])==2 else "No")
解答例
NM=input().split()
print("Yes" if len(set(NM))==1 else "No")
解答例
ABC=input().split()
print("Yes" if len(set(ABC))==2 else "No")
解答例
S=input()
print("Yes" if len(set(S))!=1 else "No")
スライス
サンプルコード
List=["a","b","c","d","e"]
print(List[1:3])
実行結果
['b', 'c']
サンプルコード
List=["a","b","c","d","e"]
print(List[1:4:2])
実行結果
['b', 'd']
サンプルコード
List=["a","b","c","d","e"]
print(List[:3])
実行結果
['a', 'b', 'c']
サンプルコード
List=["a","b","c","d","e"]
print(List[3:])
実行結果
['d', 'e']
サンプルコード
List=["a","b","c","d","e"]
print(List[:-3])
実行結果
['a', 'b']
サンプルコード
List=["a","b","c","d","e"]
print(List[-3:])
実行結果
['c', 'd', 'e']
サンプルコード
List=["a","b","c","d","e"]
print(List[::-1])
実行結果
['e', 'd', 'c', 'b', 'a']
サンプルコード
List=["a","b","c","d","e"]
print(List[::-2])
実行結果
['e', 'c', 'a']
解答例
input()
print(input()[:-1].lower().split().count("takahashikun"))
解答例
S=input()
A,B,C,D=map(int,input().split())
print(S[:A]+"\""+S[A:B]+"\""+S[B:C]+"\""+S[C:D]+"\""+S[D:])
解答例
S=input()
A,B,C,D=map(int,input().split())
print(S[:A]+'"'+S[A:B]+'"'+S[B:C]+'"'+S[C:D]+'"'+S[D:])
解答例
N,Q=map(int,input().split())
S=input()
count=[0]*N
for i in range(1,N):
count[i]+=count[i-1]
if S[i-1:i+1]=="AC":count[i]+=1
for i in range(Q):
l,r=map(int,input().split())
print(count[r-1]-count[l-1])
解答例
N,K=map(int,input().split())
H=sorted(map(int,input().split()))[::-1]
print(sum(H[K:]))
解答例
s=input()
print(s[0].upper()+s[1:].lower())
解答例
r=sorted(int(input())**2 for i in range(int(input())))[::-1]
print((sum(r[::2])-sum(r[1::2]))*355/113)
解答例
s=input()
k=int(input())
print(len(set(s[i:i+k] for i in range(len(s)-k+1))))
解答例
s=input()[:-1]
while s[:len(s)//2]!=s[len(s)//2:]:s=s[:-1]
print(len(s))
解答例
s=list(input())
while len(s)!=0:
s.pop()
if len(s)%2==0 and s[:len(s)//2]==s[len(s)//2:]:
print(len(s))
exit()
解答例
n,k=map(int,input().split())
print(sum(sorted(map(int,input().split()))[-k:]))
解答例
n,k=map(int,input().split())
print(sum(sorted(map(int,input().split()))[::-1][:k]))
解答例
print(input()[::2])
解答例
input()
l=sorted(map(int,input().split()))[::-1]
print(sum(l[::2])-sum(l[1::2]))
ABC093 B - Small and Large Integers
解答例
a,b,k=map(int,input().split())
r=range(a,b+1)
for i in sorted(set(r[:k])|set(r[-k:])):print(i)
解答例
n=int(input())
s=input()
print(max(len(set(s[:i])&set(s[i:])) for i in range(n)))
解答例
s=input()
print(min(abs(int(s[i:i+3])-753) for i in range(len(s))))
解答例
N=int(input())
p=sorted([int(input()) for i in range(N)])
print(sum(p[:-1])+p[-1]//2)
解答例
input()
L=sorted(map(int,input().split()))[::-1]
print("Yes" if L[0]<sum(L[1:]) else "No")
解答例
n=int(input())
w=list(map(int,input().split()))
m=max(w)
for i in range(n):
m=min(m,abs(sum(w[:i+1])-sum(w[i+1:])))
print(m)
解答例
S=input()
odd=S[::2]
even=S[1::2]
print("No" if "L" in odd or "R" in even else "Yes")
解答例
S=input()
print("No" if "L" in S[::2] or "R" in S[1::2] else "Yes")
解答例
N=int(input())
S=input()
print("Yes" if S[:N//2]==S[N//2:] else "No")
解答例
N=int(input())
S=input()
print(sum(S[i:i+3]=="ABC" for i in range(N-2)))
解答例
print("NO" if int(input()[::2])%4 else "YES")
解答例
print(sum(sorted(map(int,input().split()))[:2]))
解答例
print("2018"+input()[4:])
リスト内包表記
サンプルコード
List=[i for i in range(5)]
print(List)
実行結果
[0, 1, 2, 3, 4]
サンプルコード
A=[2,4,6,8,10]
A=[a//2 for a in A]
print(A)
実行結果
[1, 2, 3, 4, 5]
解答例
input()
A=list(map(int,input().split()))
ans=0
while all(a%2==0 for a in A):
A=[a/2 for a in A]
ans+=1
print(ans)
- ネスト
サンプルコード
List=[1*i + 10*j + 100*k for k in range(2) for j in range(3) for i in range(4)]
print(List)
実行結果
[0, 1, 2, 3, 10, 11, 12, 13, 20, 21, 22, 23, 100, 101, 102, 103, 110, 111, 112, 113, 120, 121, 122, 123]
ABC051 B - Sum of Three Integers
解答例
k,s=map(int,input().split())
print(len([1 for z in range(k+1) for y in range(k+1) if 0<=s-y-z<=k]))
解答例(TLE)
k,s=map(int,input().split())
print([x+y+z for z in range(k+1) for y in range(k+1) for x in range(k+1)].count(s))
解答例
a,b,c,x=[int(input()) for _ in range(4)]
print([500*x+100*y+50*z for z in range(c+1) for y in range(b+1) for x in range(a+1)].count(x))
解答例
print("Yes" if [4*x+7*y for y in range(101) for x in range(101)].count(int(input())) else "No")
要素のソート
サンプルコード
List=[[0,1],[1,1],[1,0],[0,0]]
List.sort(key=lambda List:(List[0],List[1]))
for i in List:
print(i)
実行結果
[0, 0]
[0, 1]
[1, 0]
[1, 1]
サンプルコード
List=[[0,1],[1,1],[1,0],[0,0]]
List.sort(key=lambda List:(List[0],-List[1]))
for i in List:
print(i)
実行結果
[0, 1]
[0, 0]
[1, 1]
[1, 0]
サンプルコード
List=[[0,1],[1,1],[1,0],[0,0]]
List.sort(key=lambda List:(-List[0],-List[1]))
for i in List:
print(i)
実行結果
[1, 1]
[1, 0]
[0, 1]
[0, 0]
サンプルコード
List=[[0,1],[1,1],[1,0],[0,0]]
List.sort(key=lambda List:(List[0],List[1]),reverse=True)
for i in List:
print(i)
実行結果
[1, 1]
[1, 0]
[0, 1]
[0, 0]
解答例
n=int(input())
sp=[]
for i in range(n):
s,p=input().split()
sp.append([s,int(p),i+1])
sp.sort(key=lambda sp:(sp[0],-sp[1]))
for i in sp:print(i[2])
4.5 tuple
要素の探索
解答例
import itertools
N,M=map(int,input().split())
edges={tuple(sorted(map(int,input().split()))) for i in range(M)}
ans=0
for i in itertools.permutations(range(2,N+1),N-1):
l=[1]+list(i)
ans+=sum(1 for edge in zip(l,l[1:]) if tuple(sorted(edge)) in edges)==N-1
print(ans)
4.6 dict
要素の探索
解答例
print({"Mo":5,"Tu":4,"We":3,"Th":2,"Fr":1,"Sa":0,"Su":0}[input()[:2]])
解答例
day=({"Saturday":0,"Sunday":0,"Monday":5,"Tuesday":4,"Wednesday":3,"Thursday":2,"Friday":1})
print(day[input()])
解答例
N=int(input())
A=[int(input()) for i in range(N)]
B={a:i for (i,a) in enumerate(sorted(set(A)))}
for a in A:
print(B[a])
解答例
import bisect
N=int(input())
A=[int(input()) for i in range(N)]
B=sorted(list(set(A)))
for a in A:
print(bisect.bisect_left(B,a))
解答例
N=int(input())
dictionary={}
count=0
for i in range(N):
s="".join(sorted(input()))
if s in dictionary:
dictionary[s]+=1
count+=dictionary[s]
else:
dictionary[s]=0
print(count)
解答例
N=int(input())
dictionary={}
count=0
for i in range(N):
s=str(sorted(input()))
if s in dictionary:
dictionary[s]+=1
count+=dictionary[s]
else:
dictionary[s]=0
print(count)
解答例
N=int(input())
dictionary={}
count=0
for i in range(N):
s=tuple(sorted(input()))
if s in dictionary:
dictionary[s]+=1
count+=dictionary[s]
else:
dictionary[s]=0
print(count)
解答例
import collections
N=int(input())
s=[str(sorted(input())) for i in range(N)]
c=collections.Counter(s)
print(sum((v*(v-1))//2 for v in c.values()))
解答例(TLE)
N=int(input())
S=[input() for i in range(N)]
dictionary={}
for word in S:
if word in dictionary:
dictionary[word]+=1
else:
dictionary[word]=1
for key,value in sorted(dictionary.items()):
if value==max(dictionary.values()):
print(key)
解答例(TLE)
import collections
N=int(input())
S=collections.Counter(input() for i in range(N))
dictionary=sorted(key for key,value in S.items() if value==max(S.values()))
for d in dictionary:
print(d)
解答例
import collections
N=int(input())
S=collections.Counter(input() for i in range(N))
maximum=max(S.values())
sort=sorted(key for key,value in S.items() if value==maximum)
for s in sort:
print(s)
解答例
N=int(input())
S=[input() for i in range(N)]
dictionary={}
for word in S:
if word in dictionary:
dictionary[word]+=1
else:
dictionary[word]=1
maximum=max(dictionary.values())
for key,value in sorted(dictionary.items()):
if value==maximum:
print(key)
ABC045 B - 3人でカードゲームイージー / Card Game for Three (ABC Edit)
解答例
S={c:list(input()) for c in "abc"}
s="a"
while S[s]:s=S[s].pop(0)
print(s.upper())
解答例
XU=[input().split() for i in range(int(input()))]
print(sum([float(x)*{"JPY":1,"BTC":380000}[u] for x,u in XU]))
解答例
weather={"Sunny":"Cloudy","Cloudy":"Rainy","Rainy":"Sunny"}
print(weather[input()])
ABC146 A - Can't Wait for Holiday
解答例
week={"SUN":7,"MON":6,"TUE":5,"WED":4,"THU":3,"FRI":2,"SAT":1}
print(week[input()])
解答例
week=dict(SUN=7,MON=6,TUE=5,WED=4,THU=3,FRI=2,SAT=1)
print(week[input()])
解答例
S,T=input().split()
A,B=map(int,input().split())
U=input()
ST={S:A,T:B}
ST[U]-=1
print(ST[S],ST[T])
4.7 set
和
A \cup B
サンプルコード
print({"a","b","c"}|{"c","d","e"})
実行結果
{'a', 'b', 'd', 'e', 'c'}
ABC093 B - Small and Large Integers
解答例
a,b,k=map(int,input().split())
r=range(a,b+1)
for i in sorted(set(r[:k])|set(r[-k:])):print(i)
差
A － B
サンプルコード
print({"a","b","c"}-{"c","d","e"})
実行結果
{'a', 'b'}
解答例
print(min(set(map(chr,range(97,123)))-set(input())or["None"]))
解答例
u={1,2,3}
AB=[int(input()) for i in range(2)]
print(list(u-set(AB))[0])
積
A \cap B
サンプルコード
print({"a","b","c"}&{"c","d","e"})
実行結果
{'c'}
ABC079 C - Cat Snuke and a Voyage
解答例
N,M=map(int,input().split())
sa=set()
sb=set()
for i in range(M):
a,b=map(int,input().split())
if a==1:sb.add(b)
if b==N:sa.add(a)
print("IMPOSSIBLE" if len(sa&sb)==0 else "POSSIBLE")
解答例
n=int(input())
s=input()
print(max(len(set(s[:i])&set(s[i:])) for i in range(n)))
ABC118 B - Foods Loved by Everyone
解答例
n,m=map(int,input().split())
S=set(range(1,m+1))
for i in range(n):
K,*A=map(int,input().split())
S&=set(A)
print(len(S))
対称差
A ? B
サンプルコード
print({"a","b","c"}^{"c","d","e"})
実行結果
{'a', 'b', 'e', 'd'}
解答例
N=int(input())
s=set()
for i in range(N):s^={input()}
print(len(s))
解答例
letters="abcdefghijklmnopqrstuvwxyz"
ans=sorted(set(letters)^set(input()))
print("None" if len(ans)==0 else ans[0])
解答例
print(eval(input().replace(" ","^")))
解答例
print(eval(input().replace(" ","^")))
部分集合
A \subseteq B
サンプルコード
print({"a","b"}<={"a","b","c"})
実行結果
True
サンプルコード
print({"a","d"}<={"a","b","c"})
実行結果
False
解答例
x,y=input().split()
a={"1","3","5","7","8","10","12"}
b={"4","6","9","11"}
print("Yes" if {x,y}<=a or {x,y}<=b else "No")
文字の集合
サンプルコード
import string
letters=string.ascii_lowercase
print(letters)
実行結果
abcdefghijklmnopqrstuvwxyz
サンプルコード
print(sorted(map(chr,range(97,123))))
実行結果
['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
サンプルコード
print(sorted(map(chr,range(65,91))))
実行結果
['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
解答例
print(min(set(map(chr,range(97,123)))-set(input())or["None"]))
ABC093 B - Small and Large Integers
解答例
a,b,k=map(int,input().split())
r=range(a,b+1)
for i in sorted(set(r[:3])|set(r[-3:])):
print(i)
4.8 range
サンプルコード
r=range(2,10)
print(r)
実行結果
range(2, 10)
サンプルコード
r=range(2,10)
print(r[:3])
実行結果
range(2, 5)
サンプルコード
r=range(2,10)
print(set(r[:3]))
実行結果
{2, 3, 4}
サンプルコード
r=range(2,10)
print(r[3:])
実行結果
range(5, 10)
サンプルコード
r=range(2,10)
print(set(r[3:]))
実行結果
{5, 6, 7, 8, 9}
サンプルコード
r=range(2,10)
print(r[-3:])
実行結果
range(7, 10)
サンプルコード
r=range(2,10)
print(set(r[-3:]))
実行結果
{8, 9, 7}
サンプルコード
r=range(2,10)
print(r[:-3])
実行結果
range(2, 7)
サンプルコード
r=range(2,10)
print(set(r[:-3]))
実行結果
{2, 3, 4, 5, 6}
解答例
N,L=map(int,input().split())
taste=range(L,L+N)
print(sum(taste)-min(taste,key=abs))
解答例
K,X=map(int,input().split())
print(*list(range(X-K+1,X+K)))
解答例
K,X=map(int,input().split())
print(*range(X-K+1,X+K))
第5章 組み込み関数観点
5.1 abs
- abs()：絶対値を返す
サンプルコード
print(abs(-1))
実行結果
1
ABC056 B - NarrowRectanglesEasy
解答例
w,a,b=map(int,input().split())
print(max(abs(a-b)-w,0))
解答例
s=input()
print(min(abs(int(s[i:i+3])-753) for i in range(len(s))))
解答例
x,a,b=map(int,input().split())
print("A" if abs(a-x)<abs(b-x) else "B")
5.2 all
-
all()：iterable の全ての要素が真ならば（もしくは iterable が空ならば）
True、そうでない場合はFalseを返す
サンプルコード
print(all([True,True]))
実行結果
True
サンプルコード
print(all([True,False]))
実行結果
False
サンプルコード
print(all([False,False]))
実行結果
False
ABC044 B - 美しい文字列 / Beautiful Strings
解答例
s=input()
print("Yes" if all([s.count(i)%2==0 for i in set(s)]) else "No")
解答例
input()
A=list(map(int,input().split()))
ans=0
while all(a%2==0 for a in A):
A=[a/2 for a in A]
ans+=1
print(ans)
解答例
n=int(input())
A=list(map(int,input().split()))
ans=0
while all(a%2==0 for a in A):
for i in range(n):
A[i]=A[i]/2
ans+=1
print(ans)
解答例
n=int(input())
w=[input() for i in range(n)]
flag=False
if n==len(set(w)):
if all([w[i-1][-1]==w[i][0] for i in range(1,n)]):
flag=True
print("Yes" if flag else "No")
解答例
input()
A=list(map(int,input().split()))
print("APPROVED" if all([a%2!=0 or a%3==0 or a%5==0 for a in A]) else "DENIED")
解答例
input()
A=list(map(int,input().split()))
print("APPROVED" if all(a&1 or a%3==0 or a%5==0 for a in A) else "DENIED")
S=input()
print("Good" if all(S[i]!=S[i+1] for i in range(3)) else "Bad")
5.3 any
-
any()：iterable のいずれかの要素が真ならば
True、空ならFalseを返す
サンプルコード
print(any([True,True]))
実行結果
True
サンプルコード
print(any([True,False]))
実行結果
True
サンプルコード
print(any([False,False]))
実行結果
False
S=input()
print("Bad" if any(S[i]==S[i+1] for i in range(3)) else "Good")
解答例
a,b,c=map(int,input().split())
print("YES" if any((a*i)%b==c for i in range(1,b+1)) else "NO")
5.4 bin/oct/hex
2進数⇒10進数
サンプルコード
print(int("1101",2))
実行結果
13
8進数⇒10進数
サンプルコード
print(int("700",8))
実行結果
448
16進数⇒10進数
サンプルコード
print(int("FE",16))
実行結果
254
10進数⇒2進数
サンプルコード
print(bin(23))
実行結果
0b10111
10進数⇒8進数
サンプルコード
print(oct(23))
実行結果
0o27
10進数⇒16進数
サンプルコード
print(hex(23))
実行結果
0x17
2進数表示
サンプルコード
print(0b10111)
実行結果
23
8進数表示
サンプルコード
print(0o27)
実行結果
23
16進数表示
サンプルコード
print(0x20)
実行結果
32
解答例
n=int(input())
x=""
while n!=0:
x=str(n%2)+x
n=-(n//2)
print(0 if x=="" else x)
解答例
print(2**(len(bin(int(input())))-3))
5.5 divmod
- divmod()：複素数でない2つの数を引数として取り、除算を行ったときの商と剰余からなる対を返す
サンプルコード
a,b=divmod(5,2)
print(a)
print(b)
実行結果
2
1
5.6 enumerate
- enumerate()：カウントと iterable 上の iteration によって得られた値を含むタプルを返す
サンプルコード
List=["a","b","c"]
for number,alphabet in enumerate(List):
print(number,alphabet)
実行結果
0 a
1 b
2 c
解答例
N=int(input())
A=[(int(a), i) for i,a in enumerate(input().split(),1)]
for a in sorted(A,reverse=True):
print(a[1])
ABC102 C - Linear Approximation
解答例
N=int(input())
A=sorted(a-i-1 for i,a in enumerate(map(int,input().split())))
print(sum(abs(a-A[N//2]) for a in A))
5.7 eval
サンプルコード
print(eval("1+2"))
実行結果
3
解答例
print(eval(input().replace(" ","**2*"))/1e4)
解答例
print(eval(input().replace(" ","*"))%(10**9+7))
解答例
print(sum(1-eval(input().replace(" ","-")) for _ in range(int(input()))))
解答例
print("NO" if eval(input().replace(" ","%")) else "YES")
解答例
print(sum(eval(input().replace(" ","*")) for i in range(3))//10)
解答例
print("4:3" if eval(input().replace(" ","*"))%144 else "16:9")
ABC050 A - Addition and Subtraction Easy
解答例
print(eval(input()))
5.8 len
- len()：オブジェクトの長さ（要素の数）を返す
サンプルコード
print(len("aiueo"))
実行結果
5
サンプルコード
print(max(["abc","d","ef","ghij"],key=len))
実行結果
ghij
サンプルコード
print(min(["abc","d","ef","ghij"],key=len))
実行結果
d
解答例
N=int(input())
A=set(map(int,input().split()))
print("YES" if N==len(A) else "NO")
解答例
s=input()
print("no" if len(s)-len(set(s)) else "yes")
解答例
s=input()
print(s[0]+str((len(s)-2))+s[-1])
解答例
print("x"*len(input()))
解答例
print(max(input(),input(),key=len))
解答例
print("DIFFERENT" if len(set(input()))!=1 else "SAME")
ABC046 A - AtCoDeerくんとペンキ / AtCoDeer and Paint Cans
解答例
print(len(set(input().split())))
解答例
print("Yes" if len(set(input()))==3 else "No")
解答例
S=input()
s=list(set(S))
print("Yes" if len(s)==2 and S.count(s[0])==S.count(s[1])==2 else "No")
解答例
S=input()
print("Yes" if len(set(S))!=1 else "No")
5.9 map
- map()：function の結果を返しながら、全ての要素に適用する iterator を返す
ABC110 C - String Transformation
解答例
s=input()
t=input()
S=sorted(map(s.count,set(s)))
T=sorted(map(t.count,set(t)))
print("Yes" if S==T else "No")
5.10 max
- max()：iterable の中で最大の要素、または2つ以上の引数の中で最大のものを返す
サンプルコード
print(max(1,2))
実行結果
2
サンプルコード
print(max(2,1,3))
実行結果
3
サンプルコード
print(max([2,1,3]))
実行結果
3
サンプルコード
print(max(["b","a","c","d"]))
実行結果
d
サンプルコード
print(max(["ab","aa","ca","bd"]))
実行結果
ca
サンプルコード
print(max(["a","bcde","fg","hij"],key=len))
実行結果
bcde
ABC052 B - Increment Decrement
解答例
input()
s=input()
c,m=0,0
for i in range(len(s)):
if s[i]=="I":
c+=1
m=max(c,m)
if s[i]=="D":
c=c-1
m=max(c,m)
print(m)
ABC056 B - NarrowRectanglesEasy
解答例
w,a,b=map(int,input().split())
print(max(abs(a-b)-w,0))
ABC064 B - Traveling AtCoDeer Problem
解答例
input()
a=list(map(int,input().split()))
print(max(a)-min(a))
解答例
N=int(input())
p=[int(input()) for i in range(N)]
print(sum(p)-max(p)//2)
解答例
A,B,K=map(int,input().split())
print(max(0,A-K),max(0,B-max(0,K-A)))
解答例
N,K,M=map(int,input().split())
A=list(map(int,input().split()))
x=max(0,M*N-sum(A))
print(x if x<=K else -1)
解答例
N,K,M=map(int,input().split())
A=sum(map(int,input().split()))
x=max(0,M*N-A)
print(x if x<=K else -1)
解答例
print(max(map(int,input().split())))
解答例
print(max(input(),input(),key=len))
解答例
a,b,c=map(int,input().split())
print(max(c//a,c//b))
ABC044 A - 高橋君とホテルイージー / Tak and Hotels (ABC Edit)
解答例
n,k,x,y=[int(input()) for i in range(4)]
print(n*x-(x-y)*max(n-k,0))
解答例
a,b,c,d=map(int,input().split())
print(max(a*b,c*d))
解答例
a,b=map(int,input().split())
print(max(a+b,a-b,a*b))
解答例
print("Yay!" if max(map(int,input().split()))<9 else ":(")
ABC103 A - Task Scheduling Problem
解答例
A=list(map(int,input().split()))
print(max(A)-min(A))
解答例
input()
L=list(map(int,input().split()))
print("Yes" if sum(L)>2*max(L) else "No")
解答例
a,b=map(int,input().split())
print(max(2*a-1,2*b-1,a+b))
解答例
A,B,C=map(int,input().split())
print(max(0,C-A+B))
解答例
A,B=map(int,input().split())
print(max(A+B,A-B,A*B))
解答例
A,B=map(int,input().split())
print(max(0,A-2*B))
解答例
A,B=map(int,input().split())
print("-1" if max(A,B)>9 else A*B)
5.11 min
- min()：iterable の中で最小の要素、または2つ以上の引数の中で最小のものを返す
サンプルコード
print(min(1,2))
実行結果
1
サンプルコード
print(min(2,1,3))
実行結果
1
サンプルコード
print(min([2,1,3]))
実行結果
1
サンプルコード
print(min(["b","a","c","d"]))
実行結果
a
サンプルコード
print(min(["ab","aa","ca","bd"]))
実行結果
aa
サンプルコード
print(min(["a","bcde","fg","hij"],key=len))
実行結果
a
解答例
N=int(input())
B=list(map(int,input().split()))
B+=[max(B)]
A=[0]*N
A[0]=B[0]
for i in range(1,N):
A[i]=min(B[i-1],B[i])
print(sum(A))
解答例
N=int(input())
B=list(map(int,input().split()))
B+=[max(B)]
print(sum(min(B[i-1],B[i]) for i in range(N)))
解答例
N=int(input())
P=list(map(int,input().split()))
count,minimum=0,max(P)
for i in range(N):
if P[i]<=minimum:
count+=1
minimum=P[i]
print(count)
解答例
n=int(input())
print(min(int(input()) for i in range(n)))
解答例
x=abs(int(input())-int(input()))
print(min(x,10-x))
ABC064 B - Traveling AtCoDeer Problem
解答例
input()
a=list(map(int,input().split()))
print(max(a)-min(a))
ABC076 B - Addition and Multiplication
解答例
n=int(input())
k=int(input())
s=1
for i in range(n):s=min(s*2,s+k)
print(s)
解答例
n,m,x=map(int,input().split())
s=sum(int(i)<x for i in input().split())
print(min(s,m-s))
解答例
n=int(input())
t,a=map(int,input().split())
h=list(map(int,input().split()))
z=[abs(t-x*0.006-a) for x in h]
print(z.index(min(z))+1)
解答例
N,L=map(int,input().split())
taste=[L+i for i in range(N)]
print(sum(taste)-min(taste,key=abs))
解答例
N,L=map(int,input().split())
taste=range(L,L+N)
print(sum(taste)-min(taste,key=abs))
解答例
a,b=input().split()
print(min(a*int(b),b*int(a)))
解答例
N,A,B=map(int,input().split())
div=N//(A+B)
mod=N%(A+B)
print(div*A+min(mod,A))
解答例
a,b,c=map(int,input().split())
print(c//min(a,b))
解答例
n,x=map(int,input().split())
print(min(x-1,n-x))
解答例
a,b,c=map(int,input().split())
print(min(a+b,b+c,c+a))
解答例
n,a,b=map(int,input().split())
print(min(n*a,b))
解答例
a,b,c,d=[int(input()) for _ in range(4)]
print(min(a,b)+min(c,d))
ABC103 A - Task Scheduling Problem
解答例
A=list(map(int,input().split()))
print(max(A)-min(A))
解答例
a,b,c=map(int,input().split())
print(min(c,b//a))
解答例
P,Q,R=map(int,input().split())
print(min(P+Q,Q+R,R+P))
解答例
N,A,B=map(int,input().split())
print(min(N*A,B))
5.12 ord/chr
- ord()：1 文字の Unicode 文字を表す文字列に対し、その文字の Unicode コードポイントを表す整数を返す
- chr()：Unicode コードポイントが整数である文字を表す文字列を返す
ord
サンプルコード
print([chr(i) for i in range(ord("a"),ord("z")+1)])
実行結果
['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
解答例
N=int(input())
S=input()
print("".join(chr(65+(ord(s)-65+N)%26) for s in S))
解答例
print(ord(input())-64)
解答例
print(chr(ord(input())+1))
chr
サンプルコード
print([chr(i) for i in range(97, 97+26)])
実行結果
['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
[ABC151 A - Next Alphabet](https://atcoder.jp/contests/abc151/tasks/abc151_a
解答例
print(chr(ord(input())+1))
5.13 pow
- pow()：べき乗を返す
解答例
from math import factorial
W,H=map(int,input().split())
m=10**9+7
print(factorial(W+H-2)*pow(factorial(H-1)*factorial(W-1)%m,m-2,m)%m)
解答例
print(3*pow(int(input()),2))
解答例
print(pow(int(input()),3))
解答例
print(pow(int(input()),2))
5.14 sorted
- sorted()：iterable の要素を並べ替えた新たなリストを返す
解答例
N,M=map(int,input().split())
A=[int(input()) for i in range(M)][::-1]
ans=[]
s=set()
for a in A:
if a not in s:ans.append(a)
s.add(a)
for i in range(1,N+1):
if i not in s:ans.append(i)
print(*ans,sep="\n")
解答例
N=int(input())
A=list(map(int,input().split()))
sort=[0]*N
for i in range(N):
sort[A[i]-1]+=i+1
print(*sort)
解答例
import numpy
N=int(input())
A=list(map(int,input().split()))
print(*numpy.argsort(A)+1)
解答例
import collections
import bisect
n,m=map(int,input().split())
p=[[int(j) for j in input().split()] for i in range(m)]
a=collections.defaultdict(list)
for x,y in sorted(p):
a[x]+=[y]
for x,y in p:
z=bisect.bisect(a[x],y)
print("%06d%06d"%(x,z))
解答例
N=int(input())
A=[int(input()) for i in range(N)]
sort=sorted(A)
x,y=sort[-1],sort[-2]
for a in A:
print(x if a!=x else y)
解答例
N=int(input())
A=[int(input()) for i in range(N)]
y,x=sorted(A)[-2:]
for a in A:
print(x if a!=x else y)
解答例
N,K=map(int,input().split())
H=sorted(map(int,input().split()))[::-1]
print(sum(H[K:]))
ABC009 B - 心配性な富豪、ファミリーレストランに行く。
解答例
n=int(input())
print(sorted(set([int(input()) for i in range(n)]))[-2])
解答例
r=sorted(int(input())**2 for i in range(int(input())))[::-1]
print((sum(r[::2])-sum(r[1::2]))*355/113)
ABC046 B - 文字列大好きいろはちゃんイージー / Iroha Loves Strings (ABC Edition)
n,l=map(int,input().split())
s=sorted([input() for i in range(n)])
print(*s,sep="")
n,l=map(int,input().split())
s=sorted([input() for i in range(n)])
print("".join(s))
ABC064 B - Traveling AtCoDeer Problem
解答例
input()
l=sorted(map(int,input().split()))
print(l[-1]-l[0])
解答例
n,k=map(int,input().split())
print(sum(sorted(map(int,input().split()))[-k:]))
解答例
n,k=map(int,input().split())
print(sum(sorted(map(int,input().split()))[::-1][:k]))
解答例
input()
a=sorted(map(int,input().split()))
print(a[-1]-a[0])
解答例
input()
L=sorted(map(int,input().split()))[::-1]
print("Yes" if L[0]<sum(L[1:]) else "No")
解答例
X=[int(input()) for i in range(3)]
for x in X:print(3-sorted(X).index(x))
解答例
l=[int(input()) for _ in range(3)]
s=sorted(l)[::-1]
for i in l:
print(s.index(i)+1)
解答例
print(sorted(map(int,input().split()))[1])
ABC047 A - キャンディーと2人の子供 / Fighting over Candies
解答例
a,b,c=sorted(map(int,input().split()))
print("Yes" if a+b==c else "No")
解答例
l=sorted(map(int,input().split()))
print("Yes" if sum(l[:2])==l[2] else "No")
解答例
a,b,c=sorted(map(int,input().split()))
print(a+b)
ABC103 A - Task Scheduling Problem
解答例
a,b,c=sorted(map(int,input().split()))
print(c-a)
ABC110 A - Maximize the Formula
解答例
print(eval("+".join(sorted(input()))+"*10"))
解答例
A,B,C=sorted(map(int,input().split()))
print(10*C+B+A)
5.15 sum
- sum()：合計を返す
解答例
N=int(input())
B=list(map(int,input().split()))
B+=[max(B)]
A=[0]*N
A[0]=B[0]
for i in range(1,N):
A[i]=min(B[i-1],B[i])
print(sum(A))
解答例
N=int(input())
B=list(map(int,input().split()))
B+=[max(B)]
print(sum(min(B[i-1],B[i]) for i in range(N)))
解答例
N=int(input())
S=input()
print(sum(S[n]!=S[n-1] for n in range(1,N))+1)
解答例
import itertools
N=int(input())
S=input()
print(len(list(itertools.groupby(S))))
解答例
import math
N=int(input())
A=list(map(int,input().split()))
print(math.ceil(sum(A)/(N-A.count(0))))
ABC051 B - Sum of Three Integers
解答例
k,s=map(int,input().split())
print(sum(0<=s-y-z<=k for z in range(k+1) for y in range(k+1)))
解答例
n,k=map(int,input().split())
print(sum(sorted(map(int,input().split()))[-k:]))
解答例
n,k=map(int,input().split())
print(sum(sorted(map(int,input().split()))[::-1][:k]))
解答例
N=int(input())
p=[int(input()) for i in range(N)]
print(sum(p)-max(p)//2)
解答例
n=int(input())
p=sorted([int(input()) for i in range(n)])
print(p[-1]//2+sum(p[:-1]))
解答例
input()
L=list(map(int,input().split()))
print("Yes" if sum(L)>2*max(L) else "No")
解答例
input()
L=sorted(map(int,input().split()))[::-1]
print("Yes" if L[0]<sum(L[1:]) else "No")
解答例
N,L=map(int,input().split())
taste=[L+i for i in range(N)]
print(sum(taste)-min(taste,key=abs))
解答例
N,K=map(int,input().split())
H=list(map(int,input().split()))
print(sum(1 for h in H if h>=K))
解答例
N,K=map(int,input().split())
H=list(map(int,input().split()))
print(sum(h>=K for h in H))
解答例
S=input()
print(sum(S[i]!=S[-i-1] for i in range(len(S)//2)))
ABC153 B - Common Raccoon vs Monster
解答例
H,N=map(int,input().split())
A=sum(map(int,input().split()))
print("Yes" if H<=A else "No")
解答例
print(sum(eval(input().replace(" ","*")) for i in range(3))//10)
解答例
x=sum(map(int,input()[::2]))
print(x if x<10 else "error")
解答例
print(sum(sorted(map(int,input().split()))[:2]))
解答例
S=input()
T=input()
print(sum(s==t for s,t in zip(S,T)))
解答例
A=map(int,input().split())
print("bust" if sum(A)>=22 else "win")
数字和
- 数字和：digit sum
- 数字和 - Wikipedia
サンプルコード
string="1234"
print(sum(map(int,string)))
実行結果
10 # 1+2+3+4
- 各位の数字和が元の数の約数にある自然数
解答例
n=input()
print("No" if int(n)%sum(map(int,n)) else "Yes")
解答例
n,a,b=map(int,input().split())
print(sum(i for i in range(n+1) if a<=sum(map(int,str(i)))<=b))
解答例
n,a,b=map(int,input().split())
ans=0
for i in range(n+1):
if a<=sum(map(int,str(i)))<=b:ans+=i
print(ans)
解答例
n,a,b=map(int,input().split())
s=0
for i in range(n+1):
if a<=sum(int(j) for j in str(i))<=b:
s+=i
print(s)
解答例
n,a,b=map(int,input().split())
s=0
for i in range(n+1):
c=0
d=str(i)
for j in range(len(d)):
c+=int(d[j])
if a<=c<=b:
s+=i
print(s)
解答例
N=input()
print("Yes" if int(N)%sum(map(int,list(N)))==0 else "No")
解答例
N=input()
print("Yes" if int(N)%sum(int(_) for _ in N)==0 else "No")
解答例
N=input()
s=0
for n in N:
s+=int(n)
print("Yes" if int(N)%s==0 else "No")
解答例
print(sum(map(int,input())))
5.16 zip
- zip()：それぞれの iterable から要素を集めた iterator を作り、タプルを返す
サンプルコード
Matrix=[[1,2,3],[4,5,6],[7,8,9]]
print(Matrix)
print(list(map(list,zip(*Matrix))))
実行結果
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
解答例
print(sum(map(max,zip(*[list(map(int,input().split())) for i in range(2)]))))
解答例
D=map(int,input().split())
J=map(int,input().split())
print(sum(max(d,j) for d,j in zip(D,J)))
解答例
N,M=map(int,input().split())
S,C=[],[]
for m in range(M):
s,c=map(int,input().split())
S.append(s)
C.append(c)
for i in range(10**N):
if len(str(i))==N and all(str(i)[s-1]==str(c) for s,c in zip(S,C)):
print(i)
break
else:
print(-1)
解答例
N,M=map(int,input().split())
sc=[tuple(map(int,input().split())) for m in range(M)]
for i in range(10**N):
if len(str(i))==N and all(str(i)[s-1]==str(c) for s,c in sc):
print(i)
break
else:
print(-1)
解答例
N,M=map(int,input().split())
sc=[list(map(int,input().split())) for m in range(M)]
for i in range(10**N):
if len(str(i))==N and all(str(i)[s-1]==str(c) for s,c in sc):
print(i)
break
else:
print(-1)
解答例
o=list(input())
e=list(input())+[""]
for x,y in zip(o,e):print(x+y,end="")
ABC121 B - Can you solve this?
解答例
N,M,C=map(int,input().split())
B=list(map(int,input().split()))
count=0
for i in range(N):
A=list(map(int,input().split()))
if sum(a*b for a,b in zip(A,B))+C>0:count+=1
print(count)
解答例
S=input()
print(sum(x!=y for x,y in zip(S,S[::-1]))//2)
ABC148 B - Strings with the Same Length
解答例
N=input()
S,T=input().split()
print("".join(s+t for s,t in zip(S,T)))
解答例
N=input()
S,T=input().split()
print(*[s+t for s,t in zip(S,T)],sep="")
解答例
S=input()
T=input()
print(sum(s==t for s,t in zip(S,T)))
第6章 モジュール観点
6.1 bisect
- bisect：配列二分法アルゴリズム
bisect_left / bisect_right
サンプルコード
import bisect
List=[ 1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
print(bisect.bisect_left(List,5))
# 2
"""
List=[ 1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
1 2 3 4 5 6 7 8 9 10
^
"""
print(bisect.bisect_right(List,5))
# 3
"""
List=[ 1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
1 2 3 4 5 6 7 8 9 10
^
"""
print(bisect.bisect_left(List,15))
# 7
"""
List=[ 1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
1 2 3 4 5 6 7 8 9 10
^
"""
print(bisect.bisect_right(List,15))
# 8
"""
List=[ 1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
1 2 3 4 5 6 7 8 9 10
^
"""
List=[1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3]
print(bisect.bisect_left(List,2))
# 4
"""
List=[1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3]
1 2 3 4 5 6 7 8 9 10 11 12
^
"""
print(bisect.bisect_right(List,2))
# 8
"""
List=[1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3]
1 2 3 4 5 6 7 8 9 10 11 12
^
"""
print(bisect.bisect_right(List,2))
# 8
"""
List=[1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3]
1 2 3 4 5 6 7 8 9 10 11 12
^
"""
print(bisect.bisect_right(List,3))
# 12
"""
List=[1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3]
1 2 3 4 5 6 7 8 9 10 11 12
^
"""
解答例
import bisect
A,B,Q=map(int,input().split())
INF=10**18
S=[-INF]+[int(input()) for i in range(A)]+[INF]
T=[-INF]+[int(input()) for i in range(B)]+[INF]
for q in range(Q):
x=int(input())
i=bisect.bisect_right(S,x)
j=bisect.bisect_right(T,x)
d=INF
for s in [S[i-1],S[i]]:
for t in [T[j-1],T[j]]:
d1=abs(s-x)+abs(s-t)
d2=abs(t-x)+abs(s-t)
d=min(d,d1,d2)
print(d)
解答例
import bisect
N=int(input())
A=sorted(list(map(int,input().split())))
B=sorted(list(map(int,input().split())))
C=sorted(list(map(int,input().split())))
print(sum(bisect.bisect_left(A,b)*(N-bisect.bisect_right(C,b)) for b in B))
解答例
from collections import Counter
N=int(input())
A=Counter(input().split())
B=Counter(input().split())
C=Counter(input().split())
ans=0
for a_key,a_count in A.most_common():
for b_key,b_count in B.most_common():
for c_key,c_count in C.most_common():
a_key,b_key,c_key=int(a_key),int(b_key),int(c_key)
if a_key<b_key and b_key<c_key:
ans+=a_count*b_count*c_count
print(ans)
解答例
import bisect
N,X=map(int,input().split())
L=list(map(int,input().split()))
for i in range(N-1):L[i+1]+=L[i]
print(bisect.bisect_left(L,X+1)+1)
6.2 calendar
- calendar：一般的なカレンダーに関する関数群
isleap
うるう年：Leap year
解答例
import calendar
print("YES" if calendar.isleap(int(input())) else "NO")
6.3 collections
- collections：コンテナデータ型
Counter
サンプルコード
import collections
List=["a","a","a","a","b","c","c"]
c=collections.Counter(List)
print(c)
# Counter({'a': 4, 'c': 2, 'b': 1})
print(type(c))
# <class 'collections.Counter'>
print(issubclass(type(c), dict))
# True
print(c.keys())
# dict_keys(['a', 'b', 'c'])
print(c.values())
# dict_values([4, 1, 2])
print(c.items())
# dict_items([('a', 4), ('b', 1), ('c', 2)])
print(c.most_common())
# [('a', 4), ('c', 2), ('b', 1)]
print(c.most_common()[0])
# ('a', 4)
print(c.most_common()[-1])
# ('b', 1)
print(c.most_common()[0][0])
# a
print(c.most_common()[0][1])
# 4
print(c.most_common()[::-1])
# [('b', 1), ('c', 2), ('a', 4)]
print(c.most_common(2))
# [('a', 4), ('c', 2)]
values,counts=zip(*c.most_common())
print(values)
# ('a', 'c', 'b')
print(counts)
# (4, 2, 1)
print(c["a"])
# 4
print(c["b"])
# 1
print(c["c"])
# 2
print(c["d"])
# 0
解答例
from collections import Counter
input()
L=Counter(input().split())
R=Counter(input().split())
print(sum([min(value,L[key]) for key,value in R.items()]))
解答例
from collections import Counter
N=int(input())
A=Counter(list(map(int,input().split())))
x=[0,0]
for a in A:
if A[a]>1:x.append(a)
if A[a]>3:x.append(a)
x.sort()
print(x[-1]*x[-2])
ABC071 C - 怪文書 / Dubious Document
解答例
from collections import Counter
n=int(input())
s=Counter(input())
for i in range(n-1):
s&=Counter(input())
print("".join(sorted(s.elements())))
解答例
from collections import Counter
N=int(input())
A=Counter([int(input()) for i in range(N)])
print(sum(1 if count%2 else 0 for count in A.values()))
解答例
from collections import Counter
n,k=map(int,input().split())
a=Counter(input().split())
print(sum(sorted(a.values(),reverse=True)[k:]))
解答例
from collections import Counter
n,k=map(int,input().split())
a=Counter(input().split())
ans=0
keys,counts=zip(*a.most_common())
for num,(key,count) in enumerate(zip(keys,counts)):
if int(num)>k-1:ans+=count
print(ans)
解答例
from collections import Counter
n=int(input())
a=Counter(input().split())
ans=0
for i,j in a.items():
i=int(i)
if i>j:
ans+=j
elif i<j:
ans+=j-i
print(ans)
解答例
from collections import Counter
n=int(input())
v=list(map(int,input().split()))
a=Counter(v[0::2]).most_common()
b=Counter(v[1::2]).most_common()
a.append([0,0])
b.append([0,0])
if a[0][0]!=b[0][0]:
print(n-(a[0][1]+b[0][1]))
else:
print(min(n-(a[1][1]+b[0][1]),n-(a[0][1]+b[1][1])))
ABC110 C - String Transformation
解答例
from collections import Counter
s=Counter(list(input()))
t=Counter(list(input()))
s,t=list(s.values()),list(t.values())
print("Yes" if sorted(s)==sorted(t) else "No")
解答例
import collections
N=int(input())
s=[str(sorted(input())) for i in range(N)]
c=collections.Counter(s)
print(sum((v*(v-1))//2 for v in c.values()))
解答例
from collections import Counter
print(Counter([input() for i in range(int(input()))]).most_common()[0][0])
解答例
from collections import Counter
S=Counter(input())
print("Yes" if len(S)==2 and len(set(list(S.values())))==1 else "No")
deque
サンプルコード
import collections
queue=collections.deque(["a","b","c"])
queue.append("d")
print(queue)
queue.popleft()
print(queue)
実行結果
deque(['a', 'b', 'c', 'd'])
deque(['b', 'c', 'd'])
6.4 fractions
- fractions：有理数
gcd
ABC118 C - Monsters Battle Royale
解答例
import functools
import fractions
N=int(input())
A=list(map(int,input().split()))
print(functools.reduce(fractions.gcd,A))
6.5 functools
- functools：高階関数と呼び出し可能オブジェクトの操作
reduce
サンプル
from functools import reduce
from operator import add
from operator import sub
from operator import mul
List=[20,1,2,3,4,5]
print(reduce(add,List))
# 35(=20+1+2+3+4+5)
print(reduce(sub,List))
# 5(=20-1-2-3-4-5)
print(reduce(mul,List))
# 2300(=20*1*2*3*4*5)
# Lambda式に変換が可能
print(reduce(lambda a,b:a+b,List)) # 35
print(reduce(lambda a,b:a-b,List)) # 5
print(reduce(lambda a,b:a*b,List)) # 2400
解答例
from functools import reduce
from fractions import gcd
N,X=map(int,input().split())
x=[abs(X-int(i)) for i in input().split()]
print(reduce(gcd,x))
6.6 heapq
- heapq：ヒープキューアルゴリズム
heappush / heappop
サンプルコード
import heapq
hq=[2,-20,5,0,-1,100,27]
heapq.heappush(hq,-30)
print(hq)
heapq.heappop(hq)
print(hq)
実行結果
[-30, 2, 5, -20, -1, 100, 27, 0]
[2, -20, 5, 0, -1, 100, 27]
nlargest / nsmallest
サンプルコード
import heapq
hq=[2,-20,5,0,-1,100,27]
print(heapq.nlargest(3,hq))
print(heapq.nsmallest(3,hq))
実行結果
[100, 27, 5]
[-20, -1, 0]
heapreplace
解答例
import heapq
N=int(input())
s=set()
q=[0,0]
for a in map(int,input().split()):
if a>q[0]:
try:
s.remove(a)
heapq.heapreplace(q,a)
except:
s.add(a)
print(q[0]*q[1])
6.7 itertools
- itertools：効率的なループ実行のための iterator 生成関数
accumulate
ABC143 B - TAKOYAKI FESTIVAL 2019
解答例
import itertools
N=int(input())
D=list(map(int,input().split()))
print(sum(d*cumulate for d,cumulate in zip(D[1:],itertools.accumulate(D))))
combinations
_4 C _2 = \frac{_4 P _2}{2!} = 6通り
サンプル
from itertools import combinations
List=["a","b","c","d"]
print(list(combinations(List,2)))
# [('a', 'b'), ('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd'), ('c', 'd')]
print(len(list(combinations(List,2))))
# 6
解答例
from itertools import combinations
S=map(int,input().split())
print(sorted(map(sum,combinations(S,3)))[-3])
サンプルコード
from itertools import combinations
for a,b,c in combinations("MARCH",3):
print(a,b,c)
実行結果
M A R
M A C
M A H
M R C
M R H
M C H
A R C
A R H
A C H
R C H
解答例
from itertools import combinations
from collections import Counter
N=int(input())
S=Counter()
for i in range(N):
S[input()[0]]+=1
print(sum([S[a]*S[b]*S[c] for a,b,c in combinations("MARCH",3)]))
combinations_with_replacement
サンプルコード
from itertools import combinations_with_replacement
List=["a","b","c","d"]
print(list(combinations_with_replacement(List,3)))
実行結果
[('a', 'a', 'a'), ('a', 'a', 'b'), ('a', 'a', 'c'), ('a', 'a', 'd'), ('a', 'b', 'b'), ('a', 'b', 'c'), ('a', 'b', 'd'), ('a', 'c', 'c'), ('a', 'c', 'd'), ('a', 'd', 'd'), ('b', 'b', 'b'), ('b', 'b', 'c'), ('b', 'b', 'd'), ('b', 'c', 'c'), ('b', 'c', 'd'), ('b', 'd', 'd'), ('c', 'c', 'c'), ('c', 'c', 'd'), ('c', 'd', 'd'), ('d', 'd', 'd')]
サンプルコード
from itertools import combinations_with_replacement
List=["a","b","c","d"]
print(len(list(combinations_with_replacement(List,3))))
実行結果
20
groupby
ABC063 C - 一次元リバーシ / 1D Reversi
解答例
from itertools import groupby
S=input()
G=groupby(S)
print(len(list(G))-1)
解答例
import itertools
N=int(input())
S=input()
print(len(list(itertools.groupby(S))))
解答例
import itertools
print("".join([i+str(len(list(j))) for i,j in itertools.groupby(list(input()))]))
解答例
import itertools
s=""
for i,j in itertools.groupby(list(input())):
s+=i+str(len(list(j)))
print(s)
permutations
4! = 4 × 3 × 2 × 1 = 24通り
サンプルコード
from itertools import permutations
List=["a","b","c","d","e"]
print(list(permutations(List)))
print(len(list(permutations(List))))
実行結果
[('a', 'b', 'c', 'd'), ('a', 'b', 'd', 'c'), ('a', 'c', 'b', 'd'), ('a', 'c', 'd', 'b'), ('a', 'd', 'b', 'c'), ('a', 'd', 'c', 'b'), ('b', 'a', 'c', 'd'), ('b', 'a', 'd', 'c'), ('b', 'c', 'a', 'd'), ('b', 'c', 'd', 'a'), ('b', 'd', 'a', 'c'), ('b', 'd', 'c', 'a'), ('c', 'a', 'b', 'd'), ('c', 'a', 'd', 'b'), ('c', 'b', 'a', 'd'), ('c', 'b', 'd', 'a'), ('c', 'd', 'a', 'b'), ('c', 'd', 'b', 'a'), ('d', 'a', 'b', 'c'), ('d', 'a', 'c', 'b'), ('d', 'b', 'a', 'c'), ('d', 'b', 'c', 'a'), ('d', 'c', 'a', 'b'), ('d', 'c', 'b', 'a')]
24
_4 P _2 = 4 \times 3 = 12通り
サンプルコード
from itertools import permutations
List=["a","b","c"]
print(list(permutations(List,2)))
print(len(list(permutations(List,2))))
実行結果
[('a', 'b'), ('a', 'c'), ('a', 'd'), ('b', 'a'), ('b', 'c'), ('b', 'd'), ('c', 'a'), ('c', 'b'), ('c', 'd'), ('d', 'a'), ('d', 'b'), ('d', 'c')]
12
解答例
from itertools import permutations
n,m,l=map(int,input().split())
P=list(map(int,input().split()))
v=0
for p,q,r in permutations(P):
v=max(v,(n//p)*(m//q)*(l//r))
print(v)
解答例
import itertools
N=int(input())
P,Q=[tuple(map(int,input().split())) for i in range(2)]
dictionary=sorted(itertools.permutations(P,N))
print(abs(dictionary.index(P)-dictionary.index(Q)))
解答例
import itertools
N=int(input())
P,Q=[list(map(int,input().split())) for i in range(2)]
dictionary=sorted(itertools.permutations(P,N))
print(abs(dictionary.index(tuple(P))-dictionary.index(tuple(Q))))
解答例
from itertools import permutations
import math
menu=5
times=[int(input()) for i in range(menu)]
orders=list(permutations(times,menu))
minimum=sum(math.ceil(time/10)*10 for time in times)
for order in orders:
total=order[0]
for i in range(1,menu):
total+=math.ceil(order[i]/10)*10
minimum=min(minimum,total)
print(minimum)
product
解答例
from itertools import product
[print("".join(i)) for i in product("abc",repeat=int(input()))]
解答例
import itertools
N=int(input())
ans=0
S=[]
for i in range(10):
S+=list(itertools.product("357",repeat=i))
for s in S:
if len(set(s))>2 and int("".join(s))<=N:
ans+=1
print(ans)
6.8 math
- math：数学関数
factorial
ABC067 C - Factors of Factorial
解答例
import math
N=math.factorial(int(input()))
i=2
ans=1
M=10**9+7
while i*i<=N:
cnt=1
while N%i==0:
cnt+=1
N//=i
ans*=cnt
i+=1
if N!=1:ans*=2
print(int(ans%M))
解答例
import math
print(math.factorial(int(input()))%(10**9+7))
解答例
import math
N,D=map(int,input().split())
print(math.ceil(N/(D*2+1)))
floor
- 床関数（切り捨て）：floor
- math.floor()
サンプルコード
import math
print(math.floor(5/3))
実行結果
1
サンプルコード
import math
print(math.floor(6/3))
実行結果
2
log
解答例
import math
N,K=map(int,input().split())
print(int(math.log(N,K))+1)
sqrt
- 平方根：square root
- math.sqrt()
サンプルコード
import math
print(math.sqrt(4))
実行結果
2.0
6.9 numpy
- numpy：数値計算
argsort
解答例
import numpy
N=int(input())
A=list(map(int,input().split()))
print(*numpy.argsort(A)+1)
array
解答例
import numpy as np
from itertools import combinations
N,D=map(int,input().split())
X=[list(map(int,input().split())) for i in range(N)]
count=0
for y,z in combinations(X,2):
count+=(sum(diff**2 for diff in np.array(y)-np.array(z))**.5).is_integer()
print(count)
解答例
import numpy as np
from itertools import combinations
N,D=map(int,input().split())
X=[list(map(int,input().split())) for i in range(N)]
print(sum((sum(diff**2 for diff in np.array(y)-np.array(z))**.5).is_integer() for y,z in combinations(X,2)))
norm
解答例
import numpy as np
N,D=map(int,input().split())
X=np.array([list(map(int,input().split())) for i in range(N)])
print(sum(1 for i in range(N) for j in range(i+1,N) if np.linalg.norm(X[i]-X[j]).is_integer()))
解答例
N=int(input())
xy=[list(map(int,input().split())) for i in range(N)]
print(sum(((xi-xj)**2+(yi-yj)**2)**.5 for xi,yi in xy for xj,yj in xy)/N)
解答例
import numpy as np
N=int(input())
xy=np.array([list(map(int,input().split())) for i in range(N)])
print(sum(np.linalg.norm(i-j) for i in xy for j in xy)/N)
6.10 re
- re：正規表現操作（Regular Expression operations）
findall
- re.findall()：string 中の pattern による重複しないマッチを文字列のリストとして返す
解答例
import re
print(max(map(len,re.findall("[ACGT]*",input()))))
解答例
import re
print(*re.findall("[0-9]+",input()))
match
- re.match()：string の先頭で0文字以上がこの正規表現とマッチするなら、対応する マッチオブジェクト を返す
解答例
import re
print("YES" if re.match("^(dream|dreamer|erase|eraser)+$",input()) else "NO")
解答例
import re
s=input().replace("?",".")
t=input()
for i in range(len(s)-len(t),-1,-1):
if re.match(s[i:i+len(t)],t):
s=s.replace(".","a")
print(s[:i]+t+s[i+len(t):])
exit()
print("UNRESTORABLE")
解答例
import re
s=input().replace("?",".")
t=input()
for i in range(len(s)-len(t)+1):
if re.match(s[i:i+len(t)],t):
s=s.replace(".","a")
print(s[:i]+t+s[i+len(t):])
exit()
print("UNRESTORABLE")
解答例
import re
n=int(input())
f=re.match(r"^((bcabca)*b|(cabcab)*cabca|(abcabc)*abc)$",input())
print(n//2 if f else -1)
解答例
import re
print("AC" if re.match("^A[a-z]+C[a-z]+$",input()) else "WA")
search
- re.search()：string を走査し、この正規表現がマッチを生じさせる最初の場所を探して、対応する マッチオブジェクト を返す
解答例
import re
print("YES" if re.search("i.*c.*t",input().lower()) else "NO")
解答例
import re
print("YES" if re.search("[i|I].*[c|C].*[t|T]",input()) else "NO")
split
- re.split()：string を pattern で分割し、リストを返す
解答例
import re
print(max(map(len,re.split("[^ACGT]",input()))))
sub
- re.sub()：string を重複しない pattern を置換した文字列を返す
解答例
import re
print(re.sub("\D","",input()))
解答例
import re
print(re.sub("[aiueo]","",input()))
解答例
import re
print("NO" if re.sub(r"ch|o|k|u","",input()) else "YES")
第7章 データ構造観点
7.1 スタック
- スタック：Stack
サンプルコード
stack=[3,4,5]
stack.append(6)
print(stack)
stack.pop()
print(stack)
実行結果
[3, 4, 5, 6]
[3, 4, 5]
サンプルコード
stack=["a","b","c"]
stack.pop()
print(stack)
stack.append("d")
print(stack)
実行結果
['a', 'b']
['a', 'b', 'd']
サンプルコード
import queue
q=queue.LifoQueue()
List=[3,4,5]
for l in List:q.put(l)
while not q.empty():print(q.get())
実行結果
5
4
3
解答例
S=input()
stack=[]
count=0
for s in S:
if not stack:
stack.append(s)
elif stack[-1]!=s:
stack.pop()
count+=2
else:
stack.append(s)
print(count)
ABC043 B - バイナリハックイージー / Unhappy Hacking (ABC Edit)
解答例
t=""
for x in input():
if x=="B":t=t[:-1]
else:t+=x
print(t)
解答例
t=[]
for x in input():
if x=="B":
try:t.pop()
except:pass
else:t.append(x)
print("".join(t))
7.2 キュー
- キュー：Queue
サンプルコード
queue=[3,4,5]
queue.append(6)
print(queue)
queue.pop(0)
print(queue)
実行結果
[3, 4, 5, 6]
[4, 5, 6]
サンプルコード
queue=["a","b","c"]
queue.pop(0)
print(queue)
queue.append("d")
print(queue)
実行結果
['b', 'c']
['b', 'c', 'd']
サンプルコード
from collections import deque
queue=deque(["a","b","c"])
queue.append("d")
print(queue)
queue.popleft()
print(queue)
実行結果
deque(['a', 'b', 'c', 'd'])
deque(['b', 'c', 'd'])
サンプルコード
import queue
q=queue.Queue()
List=[3,4,5]
for l in List:q.put(l)
while not q.empty():print(q.get())
実行結果
3
4
5
解答例
N=int(input())
A=list(map(int,input().split()))
print(*A[::-2],*A[N%2::2])
解答例(TLE)
N=int(input())
A=list(map(int,input().split()))
B=[]
for i in range(N):
B.append(A[i])
B=B[::-1]
print(*B)
ABC045 B - 3人でカードゲームイージー / Card Game for Three (ABC Edit)
解答例
S={c:list(input()) for c in "abc"}
s="a"
while S[s]:s=S[s].pop(0)
print(s.upper())
第8章 アルゴリズム観点
ユークリッドの互除法
- 2つの自然数の最大公約数を求める手法
サンプルコード
a,b=12,8
x,y=max(a,b),min(a,b)
while y!=0:x,y=y,x%y
print(x)
実行結果
4
解答例
a,b,n=[int(input()) for i in range(3)]
while n%a!=0 or n%b!=0:n+=1
print(n)
最大公約数（GCD）
- GCD：Greatest Common Divisor
サンプルコード
from functools import reduce
from fractions import gcd
List=[15,25,30]
print(reduce(gcd,List))
実行結果
5
解答例
from functools import reduce
from fractions import gcd
N,X=map(int,input().split())
x=[abs(X-int(i)) for i in input().split()]
print(reduce(gcd,x))
ABC118 C - Monsters Battle Royale
解答例
import functools
import fractions
N=int(input())
A=list(map(int,input().split()))
print(functools.reduce(fractions.gcd,A))
解答例
from fractions import gcd
A,B,C,D=map(int,input().split())
A-=1
lcm=C*D//gcd(C,D)
print(B-B//C-B//D+B//lcm-(A-A//C-A//D+A//lcm))
最小公倍数（LCM）
- LCM：Least Common Multiple
サンプルコード
from functools import reduce
from fractions import gcd
def lcm(x,y):return x*y//gcd(x,y)
def lcmm(l):return reduce(lcm,l)
List=[15,25,30]
print(lcmm(List))
実行結果
150
解答例
from fractions import gcd
def lcm(a,b): return a*b//gcd(a,b