「Pythonで競プロ」の記事 - Crieit

# AtCoder Beginner Contest 123CをPythonで解く

2019-06-14T07:56:24+09:00

Five Transportations

Five Transportations

問題

AtCoder社は成長し、2028 年になってついに6つの都市 (都市1,2,3,4,5,6) からなる AtCoder 帝国を作りました！
AtCoder帝国には5つの交通機関があります。
電車：都市1から2まで1分で移動する。1つの電車にはA人まで乗ることができる。
バス：都市2から3まで1分で移動する。1つのバスには B人まで乗ることができる。
タクシー：都市3から4まで1分で移動する。1つのタクシーにはC人まで乗ることができる。
飛行機：都市4から5まで1分で移動する。1つの飛行機にはD人まで乗ることができる。
船：都市5から6までを1分で移動する。1つの船にはE人まで乗ることができる。
それぞれの交通機関は、各整数時刻 (0,1,2,3,...) に、都市から出発します。
いま、N人のグループが都市1におり、全員都市6まで移動したいです。全員が都市
6に到着するまでに最短で何分かかるでしょうか？
なお、乗り継ぎにかかる時間を考える必要はありません。

こう考えた

詰まるポイントのA1>A2はA1/A2の切り上げ分時間がかかる。
Forループでそれぞれの時間を計算して追加する。

🐍これでいけんちゃうかな？入力のところは改善できるはず。

解説読む

最も通りにくい、つまり容量が最も小さい交通機関を考えます。直感的には、最短時間で移動する場合でも、1 分でたどり着けるのは 𝑋 人 (交通機関の容量の最小を 𝑋 とする) になります。これは、「1秒に 15 L のペースで水を送れるポンプと、1 秒に 12 L のペースで水を送れるポンプをつなげたときに、1 秒に送れる水の量は小さい方を取って 12 L」になるのと同じ原理です。つまり、先ほどの貪欲的な方法ではなくて、1 分に 𝑋 人ずつ 1 つ先に進めてやっても、直感的には
かかる時間は変わらなさそうです。また、移動時間は「(𝑁 ÷ 𝑋 を切り上げた値) + 4 分」と、簡単に計算できるようになります。

🐍WA！！

解説を読んで実装したコード

import math
N=int(input())
L=[int(input()) for i in range(5)]
m=math.ceil(N/min(L))
print(m+4)

AtCoder Beginner Contest 063bをPythonで解く

2019-06-13T08:22:34+09:00

AtCoder Beginner Contest 063bをPythonで解く

今回は、AtCoder Beginner Contest 063bを解いていきたいと思います。

問題

英小文字からなる文字列Sが与えられます。Sに含まれる文字がすべて異なるか判定してください。

こう考えた

set()に変換し長さを取得
len(S)とset(len(S))を比較し同じであればYes、そうでなければNo

AtCoder Beginner Contest 122cをPythonで解く

2019-06-13T08:03:00+09:00

GeT AC

GeT AC

問題

A, C, G, T からなる長さNの文字列 Sが与えられます。以下の Q個の問いに答えてください。問 i (1≤i≤Q): 整数 li,ri (1≤li
Sの先頭から li 文字目から ri 文字目までの (両端含む) 部分文字列を考える。この文字列に AC は部分文字列として何回現れるか。

AtCoder Beginner Contest 129DをPythonで解く

2019-06-12T11:27:17+09:00

AtCoder Beginner Contest 129DをPythonで解く

今回は、AtCoder Beginner Contest 129D
を解いていきたいと思います。

問題文

縦H行横W列のグリッドが与えられます。このグリッドのうち、いくつかのマスには障害物が存在します。すぬけ君は、障害物のないマスのうち一つを選び、そのマスに明かりを設置しようとしています。設置されたマスから、上下左右の四方向にまっすぐに光線が伸びます。それぞれの方向について、最初に障害物が存在するマスにぶつかる、もしくはグリッドの端にぶつかる手前のマスまで照らされます。明かりを設置したマスも照らされますが、障害物が存在するマスは照らされません。
すぬけ君は明かりによって照らされるマスの個数を最大化したいです。
H個の長さWの文字列 Si(1≤i≤H)が与えられます。Siの j文字目 (1≤j≤W) が # のとき、グリッドの上からi行目で左からj列目のマスには障害物があり、 . のときは障害物がありません。
照らされるマスの個数の最大値を求めてください。

こう考えた

numpyを使う
二次元配列に[h,w]を入力する空の配列を作る。
マス目毎に上、下、右、左を確認するコードを作成し、h,wに格納する。
max()で出力

AtCoder Beginner Contest 090BをPythonで解く

2019-06-12T08:28:48+09:00

AtCoder Beginner Contest 090BをPythonで解く

今回は、AtCoder Beginner Contest 090Bを解いていきたいと思います。

問題

A以上B以下の整数のうち、回文数となるものの個数を求めてください。ただし、回文数とは、先頭に0をつけない10進表記を文字列として見たとき、前から読んでも後ろから読んでも同じ文字列となるような正の整数のことを指します。

考え方

文字列で入力をうけて
[-1::-1]で逆の文字列を作成
if文で逆の文字列と入力の文字列が同じならカウントを1追加

AtCoder Beginner Contest 044AをPythonで解く

2019-06-12T07:53:01+09:00

AtCoder Beginner Contest 044AをPythonで解く

今回は、AtCoder Beginner Contest 044Aを解いていきたいと思います。

問題

1軒のホテルがあります。このホテルの宿泊費は、次のようになっています。
最初のK泊までは、1泊あたり X円K+1泊目以降は、1泊あたり Y円高橋君は、このホテルにN泊連続で宿泊することにしました。高橋君の宿泊費は合計で何円になるか求めてください。

こう考えた

N<=Kでif分岐させて出力を分ける。
if N<=K:はprint(X*N)
else:はprint(X*K+Y*(N-K))

AtCoder Beginner Contest 087CをPythonで解く

2019-06-11T14:00:20+09:00

AtCoder Beginner Contest 087CをPythonで解く

今回は、AtCoder Beginner Contest 087Cを解いていきたいと思います。

　問題

2×Nのマス目があります。上からi行目、左からj列目 (1≤i≤2, 1≤j≤N) のマスをマス(i,j)と表すことにします。あなたははじめ、左上のマス(1,1)にいます。
あなたは、右方向または下方向への移動を繰り返し、右下のマス(2,N)に移動しようとしています。マス(i,j)にはAi,j個のアメが置かれています。
あなたは移動中に通ったマスに置いてあるアメをすべて回収します。左上および右下のマスにもアメが置かれており、あなたはこれらのマスに置かれているアメも回収します。
移動方法をうまく選んだとき、最大で何個のアメを回収できるでしょうか。

こう考えた。

入力をN,L1,L2"でそれぞれ受ける。
forで全組み合わせを列挙し回答の配列に格納する。
printで表示する。

AtCoder Beginner Contest 088BをPythonで解く

2019-06-11T08:13:15+09:00

AtCoder Beginner Contest 088BをPythonで解く

今回は、AtCoder Beginner Contest 088Bを解いていきたいと思います。

問題文

N枚のカードがあります. i枚目のカードには, aiという数が書かれています.
Alice と Bob は, これらのカードを使ってゲームを行います. ゲームでは, Alice と Bob が交互に 1 枚ずつカードを取っていきます. Alice が先にカードを取ります.
2 人がすべてのカードを取ったときゲームは終了し, 取ったカードの数の合計がその人の得点になります. 2 人とも自分の得点を最大化するように最適な戦略を取った時, Alice は Bob より何点多く取るか求めてください.

こう考えた

リストで標準入力を取得
ターンを%2表現してAliceとBobの番を表現
(max)を使い最大値を取得しAliceまたはBobに加算
removeを用いて配列から最大値を除去

勉強になったコード

N = int(input())
a = sorted(list(map(int, input().split())), reverse=True)
alice = sum(a[0::2])
bob = sum(a[1::2])
print(abs(alice - bob))

🐍 reverse=Trueで配列を逆にして、リストのstepを2にしてsum()で合計値を出しているんやな。

AtCoder Beginner Contest 095CをPythonで解く

2019-06-11T08:03:29+09:00

AtCoder Beginner Contest 095CをPythonで解く

今回は、AtCoder Beginner Contest 095Cを解いていきたいと思います。

問題

ファーストフードチェーン「ピザアット」のメニューは「Aピザ」「Bピザ」「ABピザ」の3種類です。AピザとBピザはまったく異なるピザで、これらをそれぞれ半分に切って組み合わせたものがABピザです。AピザBピザ、ABピザ1枚あたりの値段はそれぞれA円、B円、C円です。

中橋くんは、今夜のパーティーのためにAピザX枚とBピザY枚を用意する必要があります。これらのピザを入手する方法は、AピザやBピザを直接買うかABピザ2枚を買ってAピザ1枚とBピザ1枚に組み替える以外にはありません。このためには最小で何円が必要でしょうか？なお、ピザの組み替えにより、必要な量を超えたピザが発生しても構いません。

考えたこと

Cピザが、AピザとBピザの値段両方より高ければA*X＋B*Yが最安値。
Cピザが、AピザとBピザの両ピザより安ければC*max(X,Y)*2が最安値。
Cピザが、AピザかBピザどちらかの1つのピザより安く、かつ安いピザが枚数の多い方ならばC*min(X,Y)*2+min(A,B)*abs(X-Y))
Cピザが、AピザかBピザどちらかの1つのピザより安く、かつ高いピザが枚数の多い方ならばC*max(X,Y)*2

🐍これでいけそう。

正解のコード

A,B,C,X,Y = map(int,input().split())
print(min(X*A+Y*B,X*2*C+B*max(0,Y-X),Y*2*C+A*max(0,X-Y)))

🐍　X*2*C+B*max(0,Y-X)とY*2*C+A*max(0,X-Y)で分岐しているねんな。

AtCoder Beginner Contest 073CをPythonで解く

2019-06-10T13:04:01+09:00

AtCoder Beginner Contest 073CをPythonで解く

今回は、AtCoder Beginner Contest 073Cを解いていきたいと思います。

問題文

あなたは、joisinoお姉ちゃんと以下のようなゲームをしています。
最初、何も書いていない紙がある。
joisinoお姉ちゃんが一つの数字を言うので、その数字が紙に書いてあれば紙からその数字を消し、書いていなければその数字を紙に書く。これをN回繰り返す。
その後、紙に書かれている数字がいくつあるかを答える。
joisinoお姉ちゃんが言った数字が、言った順番に A1,...,AN として与えられるので、ゲーム終了後に紙に書かれている数字がいくつあるか答えてください。

こう考えた

空の配列Lを作成
forで配列Lに要素が無ければappend
forで配列Lに要素があればremove
len(L)で答えを出力

🐍for1つしか使ってないイケてるコードが書けそうや

実際書いたコード

N=int(input())
L=[]
for i in range(N):
  A=int(input())
  if A in L:
    L.remove(A)
  else:
    L.append(A)
print(len(L))

解説を元に考え直す。

Aをソートすると、同じ数が隣り合うように並べ替えられる。

これを左か順にみていくと、奇数回登場する数が何種類あるかを数えることができる。

🐍実装してみよ。

書いたコード

N=int(input())
L=[]
for i in range(N):
  L.append(int(input()))
ans=0
for i in set(L):
  if L.count(i)%2==1:
    ans=ans+1
print(ans)

書いたコード

n = int(input())
s = set()
for i in range(n):
  a = input()
  if a in s:
    s.remove(a)
  else:
    s.add(a)
print(len(s))

Pythonで競プロ｜ABC129C｜

2019-06-10T11:33:42+09:00

AtCoder Beginner Contest 129CをPythonで解く

今回は、AtCoder Beginner Contest 129C
を解いていきたいと思います。

問題文

N段の階段があります。高橋君は現在、上り口(0段目)にいます。高橋君は一歩で 1段か 2段上ることができます。ただし、a1,a2,a3,...aM段目の床は壊れており、その段に足を踏み入れることは危険です。

壊れている床を踏まないようにしながら、最上段(N段目)にたどりつくまでの移動方法は何通りあるでしょうか？総数を1,000,000,007で割った余りを求めてください。

こう考えた

数字が連続した場合は移動方法がなくなり0になる。
a1段目までの登り方×a1からa2段目までの登り方×...となる。

実装したコード

N,M=map(int,input().split())
A=[]
for i in range(M):
  a=int(input())
  A.append(a)
print(A)
##連続していないかチェック
if M>1:
  for i in range(M-1):
    if A[i]+1==A[i+1]:
      print("No")
      break
step=0
for a in A:
  B=a-1-step
  for b in range(B):

勉強になったコード

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

b = [0]*(N+2)
b[0] = 1

for i in range(M):
  b[int(input())] = -1

for i in range(N):
  if b[i] != -1:
    if b[i+1] != -1:
      b[i+1] += b[i]
    if b[i+2] != -1:
      b[i+2] += b[i]

print(b[N]%1000000007)

# 入力は以下で行う
# 6 1
# 3

出力を見ていく

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

b = [0]*(N+2)
b[0] = 1

print(b)

# [1, 0, 0, 0, 0, 0, 0, 0]

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

b = [0]*(N+2)
b[0] = 1

for i in range(M):
  b[int(input())] = -1

print(b)

# [1, 0, 0, -1, 0, 0, 0, 0]

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

b = [0]*(N+2)
b[0] = 1

for i in range(M):
  b[int(input())] = -1

for i in range(N):
  if b[i] != -1:
  # b[i]が－1でないなら以下の2つの処理を行う
    if b[i+1] != -1:
      b[i+1] += b[i]
    if b[i+2] != -1:
      b[i+2] += b[i]
#[1, 1, 0, -1, 0, 0, 0, 0]
#[1, 1, 1, -1, 0, 0, 0, 0]
#[1, 1, 2, -1, 0, 0, 0, 0]
#[1, 1, 2, -1, 2, 0, 0, 0]
#[1, 1, 2, -1, 2, 2, 0, 0]
#[1, 1, 2, -1, 2, 2, 2, 0]
#[1, 1, 2, -1, 2, 2, 4, 0]
#[1, 1, 2, -1, 2, 2, 4, 2]

1回目のループはb[0]=1です。
+ b[1]=0ですのでb[1]にb[0]+b[1]で1が入ります。
+ b[2]=0ですのでb[2]にb[2]+b[0]で1が入ります。
2回目のループはb[1]=1から始まります。
+ b[2]=1ですのでb[2]にb[2]+b[1]で2が入ります。
+ b[3]=-1ですのでb[3]に対して何も行いません。
3回目のループはb[2]=2から始まります。
+ b[3]=-1ですので何も行いません。
+ b[4]=0ですのでb[2]+b[4]で2が入ります。
4回目のループはb[3]=-1から始まります。
+ b[3]=-1ですので何も行いません。
5回目のループはb[4]=2から始まります。
+ b[5]に2が入ります。
+ b[6]に2が入ります。
6回目のループはb[5]=2から始まります。
+ b[6]に4が入ります。
+ b[7]に2が入ります。

これで得た配列の後ろから2番目を表示すれば正解です。

勉強になったコード

N,M = map(int, input().split())
As = [1] *( N+1)
for _ in range(M):
    i = int(input())
    As[i] = 0

for i in range(N-1):
    if As[i+2] > 0:
      As[i+2] = As[i+1] + As[i]

mod = 10**9+7
print(As[-1]%mod)

コードを読む

As = [1] *( N+1)
for _ in range(M):
    i = int(input())
    As[i] = 0

ここで階段を[1,1,1,1,0]という形で表現する。

    if As[i+2] > 0:
      As[i+2] = As[i+1] + As[i]

2段先が0で無ければ、2段先に1段先と現在の段の数値を足したものを代入する。

🐍ポイントは3段先を読むという形に問題文を小分けにすることやったんやな。

Pythonで競プロ｜ABC129B｜

2019-06-10T08:46:29+09:00

AtCoder Beginner Contest 129BをPythonで解く

今回は、AtCoder Beginner Contest 129B
を解いていきたいと思います。

問題文

1から Nの番号がついた N個の重りがあり、番号 iの重りの重さは Wiです。
ある整数 1≤T
このような分け方全てを考えた時、S1と S2の差の絶対値の最小値を求めてください。

こう考えた

S1とS2の合計は、おもりの全ての和sum()になる。
sum()-S1=S2が成り立つ。
全検索してリストに格納しmin()を用いて出力する。

実装したコード

N=int(input())
W=list(map(int,input().split()))
L=[]
for i in W:
  A=sum(W[:i])
  B=sum(W)-sum(W[:i])
  L.append(abs(A-B))
print(min(L))

結果

WAでした。

🐍なんでや！！

🐶落ち着け。Wでなんでforを回してるねん。順番ずれるし番号めちゃくちゃや。

🐍！！

冷静になりACしたコード

N=int(input())
W=list(map(int,input().split()))
L=[]
for i in range(N):
  A=sum(W[:i])
  B=sum(W)-sum(W[:i])
  L.append(abs(A-B))
print(min(L))

🐍ACしたで！

勉強になったコード

n,*w=map(int,open(0).read().split());print(min(abs(sum(w[:i])-sum(w[i:]))for i in range(n)))

🐍1行で書けるの？
🐶;で分割されてるで

n,*w=map(int,open(0).read().split())ですべての入力を受けてます
nとwにそれぞれ格納されます。
min(abs(sum(w[:i])-sum(w[i:]))for i in range(n)で
abs(sum(w[:i])-sum(w[i:])))で絶対値をとりにいきforでリストに格納します

🐍open(0).read().split()は便利そう。

Pythonで競プロ｜ABC129A｜

2019-06-10T07:45:51+09:00

今回は、AtCoder Beginner Contest 129A
を解いていきたいと思います。

問題文

空港 A, B, C があり、それぞれの空港の間では、双方向に飛行機が運航しています。
空港 A, B 間の飛行時間は片道P時間、空港 B, C 間の飛行時間は片道Q時間、空港 C, A 間の飛行時間は、片道 R時間です。
いずれかの空港からスタートして他の空港に飛行機で移動し、さらにそのどちらでもない空港に飛行機で移動するような経路を考えます。
飛行時間の和は最短で何時間になるでしょうか。

こう考えた

P、Q、Rをから2つを選択しそれぞれを加算する。
それらをリストに格納しmin()で出力する。

実装したコード

P,Q,R=map(int,input().split())
print(min(P+Q,Q+R,P+R))

結果

ACでした。

🐍やったで！

勉強になったコード

print(sum(sorted(map(int,input().split()))[:2]))

map(int,input().split()ですべての入力を受けて
sorted()で小さい順番に並べ替えます。
[:2]で小さい順番から2つを引っ張ってきて
sum()で合計をだしてprint()で出力します。

sort()とsorted()の違い

リスト型のメソッドsort(): 元のリストをソート
組み込み関数sorted(): ソートした新たなリストを生成

🐍いきなりsortedで並び替える発想無かった。