-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path11049.py
More file actions
71 lines (54 loc) · 2.43 KB
/
11049.py
File metadata and controls
71 lines (54 loc) · 2.43 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
'''
11049번
제출
맞은 사람
숏코딩
재채점 결과
채점 현황
내 제출
강의
질문 검색
행렬 곱셈 순서 분류
시간 제한 메모리 제한 제출 정답 맞은 사람 정답 비율
1 초 256 MB 13211 6002 4284 43.826%
문제
크기가 N×M인 행렬 A와 M×K인 B를 곱할 때 필요한 곱셈 연산의 수는 총 N×M×K번이다. 행렬 N개를 곱하는데 필요한 곱셈 연산의 수는 행렬을 곱하는 순서에 따라 달라지게 된다.
예를 들어, A의 크기가 5×3이고, B의 크기가 3×2, C의 크기가 2×6인 경우에 행렬의 곱 ABC를 구하는 경우를 생각해보자.
AB를 먼저 곱하고 C를 곱하는 경우 (AB)C에 필요한 곱셈 연산의 수는 5×3×2 + 5×2×6 = 30 + 60 = 90번이다.
BC를 먼저 곱하고 A를 곱하는 경우 A(BC)에 필요한 곱셈 연산의 수는 3×2×6 + 5×3×6 = 36 + 90 = 126번이다.
같은 곱셈이지만, 곱셈을 하는 순서에 따라서 곱셈 연산의 수가 달라진다.
행렬 N개의 크기가 주어졌을 때, 모든 행렬을 곱하는데 필요한 곱셈 연산 횟수의 최솟값을 구하는 프로그램을 작성하시오. 입력으로 주어진 행렬의 순서를 바꾸면 안 된다.
입력
첫째 줄에 행렬의 개수 N(1 ≤ N ≤ 500)이 주어진다.
둘째 줄부터 N개 줄에는 행렬의 크기 r과 c가 주어진다. (1 ≤ r, c ≤ 500)
항상 순서대로 곱셈을 할 수 있는 크기만 입력으로 주어진다.
출력
첫째 줄에 입력으로 주어진 행렬을 곱하는데 필요한 곱셈 연산의 최솟값을 출력한다. 정답은 231-1 보다 작거나 같은 자연수이다. 또한, 최악의 순서로 연산해도 연산 횟수가 231-1보다 작거나 같다.
예제 입력 1
3
5 3
3 2
2 6
예제 출력 1
90
'''
#dfs 방식으로 최소값을 탐색한다.
INF = 987654321
n = int(input())
matrix = [[]]
answer = 0
dp = [[INF] * 501 for _ in range(501)]
import sys
for _ in range(n):
matrix.append(list(map(int,sys.stdin.readline().split())))
for i in range(1,501):
dp[i][i] = 0
for step in range(2,n + 1): # 2 ~ 500
for offset in range(1, n-step + 2): #1~
tx = offset
ty = offset + step - 1
dp[tx][ty] = INF
for mid in range(tx,ty):
cost = matrix[tx][0] * matrix[mid][1] * matrix[ty][1]
dp[tx][ty] = min(dp[tx][ty], dp[tx][mid] + dp[mid+1][ty] + cost)
print(dp[1][n])