[Python] BOJ/백준 1225번 이상한 곱셈

[문제]

https://www.acmicpc.net/problem/1225

1225번: 이상한 곱셈

 

[풀이]

되게 쉽다고 생각하고 풀었다가 시간 초과 떠서 슬펐음..

 

다른 방법은 진짜 모르겠어서 찾아봤더니 문자 하나씩 곱해 더하는 거랑 각자 더한 것을 곱하는 것과 결괏값이 같다고 한다.. 이거를 어떻게 생각해내지?!?!..

 

예제를 풀어보면 sum(A) = 6, sum(B) = 9 이기 때문에 sum(A) * sum(B) 는 54가 된다.

[코드]

- 시간 초과

import sys

if __name__ == '__main__':
    A, B = sys.stdin.readline().split()
    result = 0
    for i in A:
        for j in B:
            result += int(i) * int(j)
    print(result)

 

- 제출 코드

import sys

if __name__ == '__main__':
    A, B = map(list, sys.stdin.readline().split())
    A = list(map(int, A))
    B = list(map(int, B))
    print(sum(A) * sum(B))