백준 알고리즘 (Python 3)

컴맹 초보가 공부하면서 적는 기록 노트입니다 ㅎㅎ

혹시 같이 공부하실 분 계시면 같이 공부해요!! :)

+) 적다보니 느낀건데요 ㅋㅋㅋㅋ

제가 너무 초보라서 뭐 1도 아는 게 없는지라.... ㅠㅠ

그냥 죄다 리스트 안에 싸그리 때려박고 "정렬!!" 하고 주문을 외친 다음, 하나씩 꺼내오는 코드가 거의 대부분이네요 ㅋㅋㅋㅋ

조무래기라서 그래요.... ㅎㅎ 언젠가 실력을 쌓아서 이 코드들을 죄다 리팩토링할 수 있는 날이 오기를.. ㅜㅜㅜㅜ

리스트를 사랑하는 코딩 초보입니다 헷- :P

리스트의 "파가니니 주제에 의한 대연습곡 6번"을 제일 좋아해요

<1110번: 더하기 사이클>

// 정답이긴 한데.... 뭔가 코드가 너무 더러워서 이렇게 푸는 게 아닌 것 같긴 한데.... 방법을 잘 모르겠 ㅜㅜ

n = input()

if len(n)==1:

    n = "0" + n

n_org = n

new = str(int(n[0]) + int(n[1]))

if len(new) == 1:

  new = "0" + new

n = str(n[1]) + new[1]

if len(n) == 1:

  n = "0" + n

i = 1

while n != n_org:

    new = str(int(n[0]) + int(n[1]))

    if len(new) == 1:

      new = "0" + new

    n = n[1] + new[1]

    if len(n) == 1:

      n = "0" + n

    i += 1

print(i)

<1152번: 단어의 개수>

-------------------- 다시 제출한 코드

s = input()

print(s.strip().count(' ')+1 if s!=" " else 0)

------------------- 처음 제출한 코드

import sys

line = sys.stdin.readline()

a = line.count(" ")

if line[0] == " ":

  a -= 1

if line[-2] == " ":

  a -= 1

print(a + 1)

<1157번: 단어 공부>

str = input().upper()

lst = []

lst2 = []

for a in str:

  if a in lst:

    continue

  else:

    lst.append(a)

for e in lst:

  lst2.append(str.count(e))

lst2.sort()

if len(lst2) == 1:

  print(str)

elif lst2[-1] == lst2[-2]:

  print("?")

else:

  for a in str:

    if str.count(a) == lst2[-1]:

      print(a)

      break

<1159번: 농구 경기>

n = int(input())

lst = []

for _ in range(n):

  lst.append(input()[0])

lst2 = []

for e in lst:

  if lst.count(e) >= 5 and (e not in lst2):

    lst2.append(e)

if len(lst2) != 0:

  lst2.sort()

  for e in lst2:

    print(e, end="")

else:

  print("PREDAJA")

<1193번: 분수찾기>

n = int(input())

if n == 1:

  print("1/1")

elif n == 2:

  print("1/2")

else:

  for i in range(n):

    if (i * (i+1) / 2) >= n:

      break

  numerator = n - int(i * (i-1) / 2)

  denominator = i - numerator + 1

  if (numerator + denominator)%2 == 0:

    print("{0}/{1}".format(denominator, numerator))

  else:

    print("{0}/{1}".format(numerator, denominator))

<1316번: 그룹 단어 체커>

n = int(input())

lst = [input() for _ in range(n)]

count = 0

for e in lst:

  char = {}

  for index in range(len(e)):

    if not e[index] in char:

      char[e[index]] = index

    elif e[index] in char and char[e[index]] + 1 == index:

      char[e[index]] = index

    else:

      count -= 1

      break

  count += 1  

print(count)

<1330번: 두 수 비교하기>

(a, b)=map(int, input().split())

if a>b:

    print(">")

elif a<b:

    print("<")

else:

    print("==")

<1357번: 뒤집힌 덧셈>

a, b = map(str, input().split())

def rev(n):

  n = str(n)

  n = n[::-1]

  return int(n)

print(rev(rev(a) + rev(b)))

<1427번: 소트인사이드>

n = input()

lst = []

for i in range(len(n)):

  lst.append(n[i])

lst.sort()

lst.reverse()

for e in lst:

  print(str(e), end="")

print()

<1546번: 평균>

t = int(input())

lst = list(map(int, input().split()))

lst.sort()

sum = 0

for e in lst:

  sum += e/lst[-1]*100

print(sum / t)

<1568번: 새>

n = int(input())

second = 0

while n != 0:

  i = 1

  while n >= i:

    n -= i

    i += 1

    second += 1

print(second)

<1712번: 손익분기점>

a, b, c = map(int, input().split())

def breakeven(b, c):

  if (b == c):

    n = -1

    return n

  elif (b != c):

    n = a / (c-b)

    if (n < 0):

      return -1

    else:

      return int(n)+1

print(breakeven(b, c))

<1924번: 2007년>

month = [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

x, y = map(int, input().split())

day = y

for i in range(len(month)):

  if i < x:

    day += month[i]

  else:

    break

if day % 7 == 1:

  print("MON")

elif day % 7 == 2:

  print("TUE")

elif day % 7 == 3:

  print("WED")

elif day % 7 == 4:

  print("THU")

elif day % 7 == 5:

  print("FRI")

elif day % 7 == 6:

  print("SAT")

else:

  print("SUN")

<2292번: 벌집>

*저는 수열로 풀었어요.

벌집의 겹겹이에서 가장 큰 마지막 숫자들만 빼서 수열을 만들면 {1, 7, 19, 37, 61 ... } 이렇게 나아가고,

이 원수열의 계차수열이 등차수열이에요. (앞뒤항의 차이가 {6, 12, 18, 24 ...} 이렇게 나아감)

그래서 결국 원수열의 숫자들 = 3n(n+1) + 1 로 올라가고, 이건 2차방정식이니까요.

머리로는 count 변수를 만들어서 이걸 1씩 올려가며 풀어야할 거 같은데.. 싶었지만

초보에게 이걸 코딩하기 위한 로직은 아직 너무 어려워서..

제 주제를 빠르게 파악하고 근의 공식을 하드코딩해서 풀었.... ㅎㅎ

그래도 iteration을 하면 코드를 읽는 데 많은 시간과 데이터가 들거야.. 내 근의 공식은 값을 바로 출력해주니까..

라고 위로하며 그냥 넘어가기로 했습니다 ㅠㅠ

import math

a = 3

b = 3

c = (int(input())-1) * -1

n = ((b*-1) + (b**2 - 4*a*c)**0.5) /(2*a)

n = math.ceil(n)

print(n+1)

<2438번: 별 찍기 - 1>

n = int(input())

for i in range(n):

    print("*"*(i+1))

<2439번: 별 찍기 - 2>

n = int(input())

for i in range(n):

    print(" "*(n-i-1) + "*"*(i+1))

<2442번: 별 찍기 - 5>

n = int(input())

for i in range(n):

  print(" "*(n-i-1), end="")

  print("*"*(2*i+1))

<2446번: 별 찍기 - 9>

n = int(input())

j=0

for i in range(n, 1, -1):

  print(" " * j, end="")

  print("*" * (2 * i - 1))

  j+=1

j=n-1

for i in range(1, n+1):

  print(" " * j, end="")

  print("*" * (2 * i - 1))

  j -= 1

<2475번: 검증수>

lst = list(map(int, input().split()))

sum = 0

for e in lst:

  sum += e**2

print(sum%10)

<2480번: 주사위 세개>

lst = list(map(int, input().split()))

lst.sort()

if lst[0] == lst[2]:

  print(10000 + lst[0] * 1000)

elif lst[1] == lst[2] or lst[0] == lst[1]:

  print(1000 + lst[1] * 100)

else:

  print(lst[2]*100)

<2523번: 별 찍기 - 13>

n = int(input())

for i in range(1, n):

  print("*" * (i))

for i in range(n, 0, -1):

  print("*" * i)

<2562번: 최댓값>

lst = []

for _ in range(9):

  lst.append(input())

a = 0

b = 0

for i in range(len(lst)):

  if int(lst[i]) > a:

    a = int(lst[i])

    b = i+1

  else:

    continue

print(a)

print(b)

<2577번: 숫자의 개수>

a = int(input())

b = int(input())

c = int(input())

d = str(a*b*c)

print(d.count("0"))

print(d.count("1"))

print(d.count("2"))

print(d.count("3"))

print(d.count("4"))

print(d.count("5"))

print(d.count("6"))

print(d.count("7"))

print(d.count("8"))

print(d.count("9"))

<2587번: 대표값2>

lst = [int(input()) for _ in range(5)]

sum = 0

for e in lst:

  sum += e

print(sum//5)

lst.sort()

print(lst[2])

<2588번: 곱셈>

a = int(input())

b = input()

c = a*eval(b[2])

d = a*eval(b[1])

e = a*eval(b[0])

print(c)

print(d)

print(e)

print(c + 10*d + 100*e)

<2675번: 문자열 반복>

t = int(input())

for _ in range(t):

  a, b = map(str, input().split())

  for e in b:

    print(e * int(a), end="")

  print()

<2739번: 구구단>

a = int(input())

for i in range(1, 10):

  print(a, "*", i, "=", a * i)

<2741번: N 찍기>

n = int(input())

for i in range(n):

    print(i+1)

<2742번: 기찍 N>

n = int(input())

for i in range(n):

  print(n-i)

<2750번: 수 정렬하기>

* 쓰고나서 2751번 문제를 풀려고보니 어라?? 왜 2750번과 똑같은거죠?? 했는데,

알고보니 각 문제마다 사용해야하는 정렬 알고리즘이 정해져있었음.... ㅋㅋㅋㅋ

2750번을 풀고난 후에 봤어요 ㅠㅠ 그래서 아래 적은 코드는 설명부분을 읽기 전에 푼거라.... 양해를 부탁드립ㄴ..

n = int(input())

lst = []

for _ in range(n):

  lst.append(int(input()))

lst.sort()

for e in lst:

  print(e)

<2753번: 윤년>

a = int(input())

if a%400 == 0:

    print(1)

elif (a%4 == 0) and (a%100 != 0):

    print(1)

else:

    print(0)

<2783번: 삼각 김밥>

x, y = map(int, input().split())

minprice = x * 1000 / y

n = int(input())

for _ in range(n):

  a, b = map(int, input().split())

  if (a*1000/b < minprice):

    minprice = a*1000/b

print("%.2f" % minprice)

<2884번: 알람 시계>

(h, m) = map(int, input().split())

if m>=45:

  print(h, m-45)

else:

  if h == 0:

    print(23, m+15)

  else:

    print(h-1, m+15)

<2908번: 상수>

a, b = map(str, input().split())

c = int(a[0]) + int(a[1])*10 + int(a[2])*100

d = int(b[0]) + int(b[1])*10 + int(b[2])*100

print(max(c, d))

<2920번: 음계>

lst1 = [1, 2, 3, 4, 5, 6, 7, 8]

lst2 = [8, 7, 6, 5, 4, 3, 2, 1]

lst = list(map(int, input().split()))

if lst == lst1:

  print("ascending")

elif lst == lst2:

  print("descending")

else:

  print("mixed")

<2941번: 크로아티아 알파벳>

import re

cro = ["c=", "c-", "dz=", "d-", "lj", "nj", "s=", "z="]

string = input()

lst = []

for e in cro:

  for a in re.finditer(e, string):

    i = a.start()

    add = (string[i] + string[i+1])

    lst.append(add)

print(len(string)-len(lst))

<3035번: 스캐너>

r, c, a, b = map(int, input().split())

lst = [input() for _ in range(r)]

for string in lst:

  string = ''.join(char*b for char in string)

  for _ in range(a):

    print(string)

<3047번: ABC>

lst1 = list(map(int, input().split()))

string = input()

lst1.sort()

dict = {

  "A": lst1[0],

  "B": lst1[1],

  "C": lst1[2]

}

result = ""

for char in string:

  result += str(dict[char]) + " "

print(result.strip())

<3052번: 나머지>

lst1 = []

lst2 = []

for _ in range(10):

  lst1.append(int(input()))

for e in lst1:

  if e%42 in lst2:

    continue

  else:

    lst2.append(e%42)

print(len(lst2))

<4344번: 평균은 넘겠지>

t = int(input())

for _ in range(t):

  lst = list(map(int, input().split()))

  n = lst[0]

  del lst[0]

  sum1 = 0

  for e in lst:

    sum1 += e

  avg = (sum1/n)

  count = 0

  for e in lst:

    if e > avg:

      count += 1

  print("%.3f"%((count/n)*100) + "%")

<5543번: 상근날드>

hamburgers = [int(input()) for _ in range(3)]

soda = [int(input()) for _ in range(2)]

hamburgers.sort()

soda.sort()

print(hamburgers[0] + soda[0] - 50)

<5585번: 거스름돈>

n = int(input())

n = 1000 - n

result = 0

lst = [500, 100, 50, 10, 5, 1]

for e in lst:

  result += n//e

  n %= e

print(result)

<5586번: JOI와 IOI>

n = input()

a = n.count("JOI")

b = 0

for i in range(len(n)):

  if i == len(n)-2:

    if n[i] == "I" and n[i+1] == "O" and n[i+2] == "I":

      b += 1

      break

  elif i == len(n) -1:

    break       

  elif n[i] == "I" and n[i+1] == "O" and n[i+2] == "I":

    b += 1

print(a)

print(b)

<5597번: 과제 안 내신  분..?>

lst = [int(input()) for _ in range(28)]

lst2 = [ x for x in range(1, 31)]

for number in lst2:

  if number in lst:

    continue

  else:

    print(number)

<5622번: 다이얼>

*이건 제일 처음 떠올렸던 방식은 if statement인데,

저는 평소에 딕셔너리에 대한 개념이 좀 없고 딕셔너리를 잘 사용하지도 못 해서, 이번에 딕셔너리로 써봤어요.

if statement로 적은 코드도 아래에 같이 적어둘게요-

- dictionary를 사용한 코드 -

dial = {'A':3, 'B':3, 'C':3, 'D':4, 'E':4, 'F':4, 'G':5, 'H':5, 'I':5, 'J':6, 'K':6, 'L':6, 'M':7, 'N':7, 'O':7, 'P':8, 'Q':8, 'R':8, 'S':8, 'T':9, 'U':9, 'V':9, 'W':10, 'X':10, 'Y':10, 'Z':10}

str = input()

time = 0

for a in str:

  time += dial[a]

print(time)

---

- if statement를 사용한 코드 -

str = input()

time = 0

for alphabet in str:

  if alphabet <= 'C':

    time += 3

  elif alphabet <= 'F':

    time += 4

  elif alphabet <= 'I':

    time += 5

  elif alphabet <= 'L':

    time += 6

  elif alphabet <= 'O':

    time += 7

  elif alphabet <= 'S':

    time += 8

  elif alphabet <= 'V':

    time += 9

  else:

    time += 10

print(time)

<8393번: 합>

a = int(input())

sum = 0

for i in range(1, a+1):

    sum += i

print(sum)

<8958번: OX퀴즈>

n = int(input())

lst = []

for _ in range(n):

    lst.append(input())

for line in lst:

    sum = 0

    i = 0

    for e in line:

        if e == 'X':

            i = 0

        else:

            i += 1

            sum += i

    print(sum)

<9012번: 괄호>

n = int(input())

lstAnswer = []

for _ in range(n):

  lst = []

  string = input()

  try:

    for char in string:

      if char == "(":

        lst.append(char)

      else:

        lst.pop()

    if len(lst) == 0:

      lstAnswer.append("YES")

    else:

      lstAnswer.append("NO")

  except:

    lstAnswer.append("NO")

for e in lstAnswer:

  print(e)

<9498번: 시험 성적>

a = int(input())

if a >= 90:

    print("A")

elif a >= 80:

    print("B")

elif a >= 70:

    print("C")

elif a >= 60:

    print("D")

else:

    print("F")

<10039번: 평균 점수>

lst = [int(input()) for _ in range(5)]

sum = 0

for i in range(5):

  if lst[i] < 40:

    lst[i] = 40

  sum += lst[i]

print(int(sum/5))

<10093번: 숫자>

a, b = map(int, input().split())

if a > b:

  a, b = b, a

lst = []

if a == b or a+1 == b:

  print(0)

  print()

else:

  for _ in range(a, b):

    lst.append(_)

  lst.pop(0)

  print(len(lst))

  for e in lst[:-1]:

    print(e, end=" ")

  print(lst[-1])

<10172번: 개>

print("""|\_/|

|q p|   /}

( 0 )\"\"\"\\

|\"^\"`    |

||_/=\\\__|""")

<10610번: 30>

n = input()

a = ''.join(sorted(list(n),reverse=True))

print(a if int(a)%30==0else-1)

<10797번: 10부제>

n = int(input())

lst = list(map(int, input().split()))

count = 0

for e in lst:

  if e%10 == n:

    count += 1

print(count)

<10809번: 알파벳 찾기>

alphabet = {'a':-1, 'b':-1, 'c':-1, 'd':-1, 'e':-1, 'f':-1, 'g':-1, 'h':-1, 'i':-1, 'j':-1, 'k':-1, 'l':-1, 'm':-1, 'n':-1, 'o':-1, 'p':-1, 'q':-1, 'r':-1, 's':-1, 't':-1, 'u':-1, 'v':-1, 'w':-1, 'x':-1, 'y':-1, 'z':-1}

string = input()

for i in range(len(string)):

  if alphabet[string[i]] == -1:

    alphabet[string[i]] = i

  else:

    continue

lst = list(alphabet.values())

for e in lst:

  print(e, end=" ")

<10814번: 나이순 정렬>

n = int(input())

lst = []

for _ in range(n):

  a, b = input().split()

  lst.append([int(a), b])

lst.sort(key=lambda x:x[0])

for e in lst:

  print(e[0], e[1])

<10817번: 세 수>

l = list(map(int, input().split()))

l.sort()

print(l[1])

<10818번: 최소, 최대>

n = int(input())

lst = list(map(int, input().split()))

lst.sort()

print(lst[0], lst[-1])

<10869번: 사칙연산>

(a, b)=map(int, input().split())

print(a+b)

print(a-b)

print(a*b)

print(int(a/b))

print(a%b)

<10870번: 피보나치 수 5>

n = int(input())

a = 1

b = 0

sum = 0

for i in range(n):

  b = a

  a = sum

  sum = a + b

print(sum)

<10871번: X보다 작은 수>

(n, x) = map(int, input().split())

l = list(map(int, input().split()))

for e in l:

  if e < x:

    print(e, end=" ")

<10872번: 팩토리얼>

n = int(input())

a = 1

for i in range(1, n+1):

  a *= i

print(a)

<10950번: A+B - 3>

t = int(input())

for i in range(t):

  (a, b)=map(int, input().split())

  print(a+b)

<10951번: A+B - 4>

import sys

for line in sys.stdin:

  a, b = map(int, line.split())

  print(a+b)

<10952번: A+B - 5>

(a, b) = map(int, input().split())

while a+b != 0:

  print(a+b)

  (a, b) = map(int, input().split())

<11021번: A+B - 7>

t = int(input())

for i in range(1, t+1):

    (a, b)=map(int, input().split())

    print("Case #" + str(i) + ": " + str(a+b))

<11022번: A+B - 8>

t = int(input())

for i in range(1, t+1):

    (a, b)=map(int, input().split())

    print("Case #" + str(i) + ": " + str(a) + " + " + str(b) + " = " + str(a+b))

<11365번: !밀비 급일>

str = input()

lst = []

lst.append(str)

while str != "END":

  str = input()

  lst.append(str)

for e in lst[:-1]:

  print(e[::-1])

<11399번: ATM>

n = int(input())

lst = list(map(int, input().split()))

lst.sort()

total = 0

for i in range(len(lst)):

  total += (len(lst)-i) * lst[i]

print(total)

<11653번: 소인수분해>

n = int(input())

i = 2

lst = []

while n != 1:

  if n%i == 0:

    lst.append(i)

    n = n//i

  else:

    i += 1

for e in lst:

  print(e)

<11654번: 아스키 코드>

print(ord(input()))

<11656번: 접미사 배열>

string = input()

lst = []

for i in range(len(string)):

  lst.append(string[i:])

lst.sort()

for e in lst:

  print(e)

<11718번: 그대로 출력하기>

while True:

  try:

    print(input())

  except:

    break

<11720번: 숫자의 합>

n = int(input())

a = input()

sum1 = 0

for i in range(n):

  sum1 += eval(a[i])

print(sum1)

<14656번: 조교는 새디스트야!!>

n = int(input())

lst = list(map(int, input().split()))

result = 0

for i in range(len(lst)):

  if lst[i] != i+1:

    result += 1

print(result)

<15552번: 빠른 A+B>

import sys

t = int(input())

for i in range(t):

    (a, b)=map(int, sys.stdin.readline().split())

    print(a+b)

<15596번: 정수 N개의 합>

def solve(a):

    sum1 = 0

    for e in a:

        sum1 +=e

    return sum1

<18108번: 1998년생인 내가 태국에서는 2541년생?!>

n = int(input())

print(n - 543)