가전제품 이것저것 이야기

www.acmicpc.net/problem/13305

그리디 알고리즘으로 주유소를 지나면서

가장 주유비가 작은 비용으로 계속 더하는 방식으로 풀었습니다.

import sys

min = sys.maxsize
n = (int)(input())

# list에 값 넣기
listDis = list(map(int,input().split()))
listCost = list(map(int,input().split()))

# 도시를 다니면서 비용을 더하기
result = 0
for i in range(n-1):
    if(i == 0):
        result += listDis[i] * listCost[i]
        min = listCost[i]
    else:
        if listCost[i] < min:
            min = listCost[i]
        result += listDis[i] * min

print(result)

새롭게 알게된 문법 )

리스트에 값을 입력받아서 바로 넣기

listDis = list(map(int,input().split()))
listCost = list(map(int,input().split()))

sort vs sorted

sort()는 one dimensional list에서 많이 쓰이고,

sorted()는 여러 차원이나 조건들을 설정할 때 많이 쓰인다.

sort()

오름차순

a = [1,4,2,5,3]
print(a.sort())

[1,2,3,4,5]

내림차순

a = [1,4,2,5,3]
print(a.sort(reverse=True))

[5,4,3,2,1]

sorted()

>>> students = [
        ('홍길동', 3.9, 2016303),
        ('김철수', 3.0, 2016302),
        ('최자영', 4.3, 2016301),
]

오름차순(세 번째 값을 기준으로)

>>> sorted(students, key=lambda student: student[2])
[('최자영', 4.3, 2016301), ('김철수', 3.0, 2016302), ('홍길동', 3.9, 2016303)]

내림차순

>>> sorted(students, key=lambda student: -student[2])
[('홍길동', 3.9, 2016303), ('김철수', 3.0, 2016302), ('최자영', 4.3, 2016301)]

만약 세 번째 값은 오름차순, 두 번째 값은 내림차순으로 정렬하고 싶다면?

>>> sorted(students, key=lambda student: (student[2],-student[1]))
[('최자영', 4.3, 2016301), ('김철수', 3.0, 2016302), ('홍길동', 3.9, 2016303)]

www.acmicpc.net/problem/1011

네 달 전부터 풀지 못했던 문제를 드디어 풀었습니다 !! 유후

이 문제는 숫자의 규칙을 이용해서 풀었습니다. 그리고 확실히 파이썬으로 하니 코드가 짧고 단순해 보이는 것 같습니다. 

코딩 테스트 볼 때에 시간이 많이 단축될 것 같아요 ㅎㅎ

파이썬 코드

import math

n = (int)(input())

for i in range(n):
    x,y = map(int, input().split())
    diff = y-x
    if diff <= 3:
        print(diff)
    else :
        x = (int)(math.sqrt(diff))
        if diff == x * x:
            print(2 * x - 1)
        elif diff <= (x * x + x):
            print(2 * x)
        else:
            print(2 * x + 1)

새로 사용한 기능은 math 모듈 안의 sqrt() 라는 기능입니다.

해당 숫자의 제곱근 값을 반환해 주는 것입니다. int로 형변환이 필요합니다.

- 그리고 한 가지 더!!

input()을 받을 때에는 꼭 int형으로 변환을 해줘야 합니다. 안 그러면 str으로 인식하는 것 같습니다ㅠㅠ

현재 상태

목표 : 큐, 덱까지 못 푼 문제 파이썬으로 다 풀어보기 !

21. 11. 19 중간점검

새로운 목표 : 위상 정렬까지 단계별 다 풀기

* 동적계획법 기초 문제를 풀어 보아야 겠다

+ 추가

Enumerate는 리스트의 인덱스와 함께 값을 추출할 때 사용한다.

Zip은 여러 개의 리스트에서 각각의 값을 병렬적으로 가져올 때 사용한다.

Enumerate

for i,j in enumerate(['tic','tac','toe']):
    print(i,j)

myList = ['a','b','c','d']
print(list(enumerate(myList)))

myTuple = {i:j for i,j in enumerate('Apple Banana Computer Dictionary Eraser Food'.split())}
print(myTuple)

0 tic
1 tac
2 toe
[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]
{0: 'Apple', 1: 'Banana', 2: 'Computer', 3: 'Dictionary', 4: 'Eraser', 5: 'Food'}

Zip

alist = ['a1','a2','a3']
blist = ['b1','b2','b3']

for a, b in zip(alist, blist):
    print(a,b)

a1 b1
a2 b2
a3 b3

Zip 과 Enumerate를 같이 사용하면?

# zip + enumerate
for i, (a, b) in enumerate(zip(alist, blist)):
    print(i,a,b)

0 a1 b1
1 a2 b2
2 a3 b3

이 때, (a,b)에서 괄호를 꼭 붙여 주어야 한다!!

리스트나 튜플에서 각각의 값을 꺼내어 새로운 리스트를 만들 수도 있다.

# 리스트 활용
cList = [1,2,3]
dList = [10,20,30]
eList = [100,200,300]

result = [sum(x) for x in zip(cList, dList, eList)]
print(result)

# 튜플 활용
aTuple = (1,2,3)
bTuple = (10,20,30)
cTuple = (100,200,300)

result = [sum(x) for x in zip(aTuple, bTuple, cTuple)]
print(result)

[111, 222, 333]
[111, 222, 333]

For Loop + append

result = [i for i in range(10)]
print(result)

result = [i for i in range(10) if i % 2 == 0]
print(result)

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
[0, 2, 4, 6, 8]

Nested For Loop

word1 = "ABC"
word2 = "CDF"
result = [i+j for i in word1 for j in word2]
print(result)

result = [i+j for i in word1 for j in word2 if not i==j]
print(result)

result = [[i+j for i in word1 ] for j in word2]
print(result)

['AC', 'AD', 'AF', 'BC', 'BD', 'BF', 'CC', 'CD', 'CF']
['AC', 'AD', 'AF', 'BC', 'BD', 'BF', 'CD', 'CF']
[['AC', 'BC', 'CC'], ['AD', 'BD', 'CD'], ['AF', 'BF', 'CF']]

이중 리스트

# 이중 리스트
words = 'apple banana computer doctor eraser fox grape'.split()
# print(words)

stuff = [[w.upper(), w.lower(), len(w)] for w in words]
for i in stuff:
    print(i)

['APPLE', 'apple', 5]
['BANANA', 'banana', 6]
['COMPUTER', 'computer', 8]
['DOCTOR', 'doctor', 6]
['ERASER', 'eraser', 6]
['FOX', 'fox', 3]
['GRAPE', 'grape', 5]