Python/알고리즘

[백준] 1225 이상한 곱셈

vss121 2022. 7. 6. 19:57

교훈

0. 너무 직관적으로 처음 생각한 풀이에서 조금 더 시간을 줄일 수 있는 풀이를 고민하기

 

1. 형 변환 시 

(int)a가 아니라 int(a) 임!!

 

2. 이중for문으로 하니 오답이다

## 시간초과

a, b = input().split()
char_a = []
char_b = []

for i in a:
    char_a.append(int(i))

for j in b:
    char_b.append(int(j))

#print(char_a)
#print(char_b)

sum = 0

for i in char_a:
    for j in char_b:
        sum += j*i

print(sum)
## 시간초과
## 위의 것 짧게 쓴 version

a, b = input().split()
sum = 0

for i in a:
    for j in b:
        sum += int(i)*int(j)

print(sum)

 

최종

a, b = input().split()
listA = list(map(int, a))
listB = list(map(int, b))

print(sum(listA)*sum(listB))