ABC183 C - Travel에서 배운





일순간 그래프인가? 라고 생각했다.
하지만 출력 예 1에 힌트가 있었다.

1 => ********** => 1로 돌아온다.
사이 **********는 2 ~ 8의 정렬.

예를 들면 1=> 2 => 3 => 4 => 1 을 생각한다.

행렬 T 로부터 다음과 같이 값을 꺼내 덧붙여 K 와 일치하는 조합을 카운트하면 된다.
1행 2열 1=>2 => 3 => 4 => 1
2행 3열 1=>2 => 3 => 4 => 1
3행 4열 1=>2 => 3 => 4 => 1
4행 1열 1=>2 => 3 => 4 => 1

Travel.py
n,k = map(int,input().split())
T = [list(map(int,input().split())) for _ in range(n)]

from itertools import permutations
cnt = 0
for nums in permutations(range(1,n)):#O(ザックリ40000)
    #print(nums)
    nums = [0]+list(nums)+[0]
    score = 0
    for i in range(len(nums)-1):#O(9)
        score += T[nums[i]][nums[i+1]]
    #print(score)
    if score == k:
        cnt += 1
print(cnt)
#O(360000) .. ザックリだけど間に合うっしょ。

좋은 웹페이지 즐겨찾기