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ループでそれぞれの時間を計算して追加する。

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

書いたコード

import math
L=[int(input()) for i in range(6)]
Step=0
min=L[0]
for i in range(5):
  if min > L[i+1]:
    Step=Step+math.ceil(min/L[i+1])
    min=L[i+1]
  else:
    Step=Step+1
print(Step)

🐍これならループも5回やしTLEにはならへんやろ。。

結果

WA！！

🐍WA！！

解説読む

最も通りにくい、つまり容量が最も小さい交通機関を考えます。直感的には、最短時間で移動する場合でも、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)

結果

AC！！

🐍でけたで！

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

書いたコード

S=input()
if len(set(S))==len(S):
  print("yes")
else:
  print("no")

🐍これでいけんちゃうかな？

結果

AC！！

🐍おし！

勉強になったコード


a=input();print(['no','yes'][len(a)==len(set(a))])

🐍print([][])みたいなので条件かけるんや

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 は部分文字列として何回現れるか。

こう考えた

countでACの数を計算して出力する。

書いたコード

N,Q=map(int,input().split())
S=input()
for q in range(Q):
  l,r=map(int,input().split())
  s=S[l-1:r]
  print(s.count("AC"))

🐍これでいけんちゃうかな？

結果

TLE

🐍計算量の問題か

Atcoderの標準回答

N,Q=map(int,input().split())
S=input()
t = [0] * (N + 1)
for i in range(N):
  t[i + 1] = t[i] + (1 if S[i : i + 2] == 'AC' else 0)
for q in range(Q):
  l,r=map(int,input().split())
  print(t[r-1] - t[l-1])

🐍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()で出力

書いたコード

L=[list(input() )for i in range(H)]
L = np.array(L)
print(L)
M=[]
for h in range(H):
  for w in range(W):
    if L[h][w]=="#":
      M.append([0,0])
    else:
      R=1
      for r in range(W-w-1):
        if L[h][w+r+1]==".":
          R=R+1
        else:
          break
      for r in range(w-1):
        if L[h][w-r-1]=="." and w-r-1>=0:
          R=R+1
        else:
          break
      M.append(R)
print(M)

🐍ここまで書いたけど知識不足で書けない。。

コードを読んでみる

from numpy import*
h,*s=open(0)
h,w=int64(h.split())
r=range
s=array([list(t[:-1])for t in s])
a,b,c,d=eval('zeros((h,w),"i"),'*4)
for i in r(w):
    a[:,i]=-~a[:,i-1]*(s[:,i]>'#')
    b[:,~i]=-~b[:,-i]*(s[:,~i]>'#')
for i in r(h):
    c[i,:]=-~c[i-1,:]*(s[i,:]>'#')
    d[~i,:]=-~d[-i,:]*(s[~i,:]>'#')
print(int((a+b+c+d).max())-3)

3行目まで

h,*s=open(0)
print(h)
print(s)
h,w=int64(h.split())
print(h)
print(w)

🐍とりあえず出力確認や

4 6
['#..#..\n', '.....#\n', '....#.\n', '#.#...']
4
6

🐍なるほど

r=range

🐍なるほど後でfor i in range()が何度か出てくるからここで定義してるねんな。

s=array([list(t[:-1])for t in s])

🐍numpyのarrayに変換してるねんな。なんで[list(t[:-1])なんやろ？

[['#', '.', '.', '#', '.', '.'] ['.', '.', '.', '.', '.', '#']
 ['.', '.', '.', '.', '#', '.'] ['#', '.', '#', '.', '.']]

['#..#..\n' '.....#\n' '....#.\n' '#.#...']

🐍1次元配列なるし、改行も含まれてまうねんな。

a,b,c,d=eval('zeros((h,w),"i"),'*4)

🐍 eval は第1引数を式として評価してるから'zeros((h,w),"i"),'までが今回は引数。zerosは、0で初期化されたndarrayを生成する関数やな。abcdで4つ生成してるねんな。

for i in r(w):
    a[:,i]=-~a[:,i-1]*(s[:,i]>'#')
    b[:,~i]=-~b[:,-i]*(s[:,~i]>'#')

Traceback (most recent call last):
  File "./Main.py", line 8, in 
    a[:,i]=-~a[:,i-1]*(s[:,i]>'#')
IndexError: too many indices

🐍 ACしてるのにエラーでるな。このコードの読コードはここまでや。

参考にしたコード

import numpy as np
H,W=map(int,input().split())
S=[list(input()) for h in range(H)]
S=[[1 if s=='.' else 0 for s in S[h]] for h in range(H)]
S=np.array(S)
L=S.copy()
R=S.copy()
U=S.copy()
D=S.copy()
for w in range(1,W):
    L[:,w]=(L[:,w-1]+1)*S[:,w]
    R[:,-(1+w)]=(R[:,-w]+1)*S[:,-(1+w)]
for h in range(1,H):
    U[h,:]=(U[h-1,:]+1)*S[h,:]
    D[-(1+h),:]=(D[-h,:]+1)*S[-(1+h),:]
print(int((L+R+U+D).max())-3)

S=[[1 if s=='.' else 0 for s in S[h]] for h in range(H)]

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

🐍 ここは文字列を0と1に置き換えてるんやな。

L[:,w]=(L[:,w-1]+1)*S[:,w]

🐍 ,ってなんや外したろ

Traceback (most recent call last):
  File "./Main.py", line 12, in 
    L[:w]=(L[:w-1]+1)*S[:w]
ValueError: could not broadcast input array from shape (0,6) into shape (1,6)

🐍多次元の場合は、カンマ（,）で区切ってそれぞれを指定せなあかんねんな。

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追加

実際書いたコード

A,B=map(int,input().split())
n=0
for i in range(A,B):
  C=(str(i)[-1::-1])
  if int(C)==i:
    n=n+1
print(n)

🐍これでいけそう

結果

🐍あっ、rangeはB+1にしなあかん。

修正したコード

A,B=map(int,input().split())
n=0
for i in range(A,B+1):
  C=(str(i)[-1::-1])
  if int(C)==i:
    n=n+1
print(n)

🐍これでいけそう

結果

🐍ケアレスミスや

勉強になったコード

a,b=map(int,input().split())
print(len([i for i in map(str,range(a,b+1)) if i==i[::-1]]))

🐍 map（str,)をrangeにつかえるんや。