[PGS] 롤케이크자르기 / lv2 / 50min

https://school.programmers.co.kr/learn/courses/30/lessons/132265

Author: zaqquum

Hongjoo/lv2/롤케이크자르기.py

@@ -0,0 +1,27 @@
+"""
+# 계수 정렬 idea 적용
+"""
+
+def solution(topping):
+    answer = 0 # 공평하게 자를 수 있는 경우의 수
+    # 0, graph
+    forward = set()
+    backward = dict()
+    # 1. key 종류, value: toppping 중복 개수
+    for t in topping:
+        backward[str(t)]= backward.get(str(t),0)
+        backward[str(t)] += 1
+    print(backward)
+    #2. forward vs backward
+    for t in topping :
+        # f 에서 t가 추가됨 == b에서 t 빠짐
+        forward.add(t)
+        backward[str(t)] -=1
+        if backward[str(t)] == 0 :
+            del backward[str(t)]
+        
+        # 토핑 종류 같은 경우 = 공평 한 경우
+        if len(forward) == len(backward.keys()):
+            answer+=1
+        
+    return answer

Hongjoo/백준/플로이드와샬.py

@@ -0,0 +1,35 @@
+INF = int(1e9) #무한을 나타내는 갑스올  10억 설정
+
+# 노드 개수 & 간선의 개수 입력
+n = int(input())
+m = int(input())
+#1. graph 설정 -인접 행렬(2차원 , nxn ) , 초기화
+graph = [[INF]*(n+1) for _ in range(n+1)]
+
+#1-1. 자기 자신에서 자기 자신 가는 비용 {Cost (A->A)}= 0 초기화
+for a in range(1,n+1) :
+    for b in range(1,n+1):
+        if a==b:
+            graph[a][a] = 0
+
+#1-2 .각 간선에 대한 정보를 입력 받아, 그 값으로 초기화 :   Cost(A->B) = 초기값
+for _ in range(m):
+    # Cost(A->B) = c
+    a,b,c = map(int, input().split())
+    graph[a][b]= c
+
+#2. 플로이드 워샬 점화식 수행
+# a->b : a->k->b vs a->b
+for k in range(1,n+1):   
+    for a in range(1,n+1):
+        for b in range(1,n+1):
+            graph[a][b] = min(graph[a][b], graph[a][k]+ graph[k][b])
+
+# 출력
+for a in range(1,n+1):
+    for b in range(1,n+1):
+        if graph[a][b] == INF :
+            print("INF", end="")
+        else :
+            print(graph[a][b], end ="")
+print()