校招真題練習019 畢業旅行問題(頭條)

畢業旅行問題

題目描述
小明目前在做一份畢業旅行的規划。打算從北京出發，分別去若干個城市，然后再回到北京，每個城市之間均乘坐高鐵，且每個城市只去一次。由於經費有限，希望能夠通過合理的路線安排盡可能的省一些路上的花銷。給定一組城市和每對城市之間的火車票的價錢，找到每個城市只訪問一次並返回起點的最小車費花銷。

輸入描述:
城市個數n（1<n≤20，包括北京）
城市間的車票價錢 n行n列的矩陣 m[n][n]

輸出描述:
最小車費花銷 s

牛客網上對此題的python語言判斷似乎有問題，一直超時，之前做劍指Offer的題目，也有一道出現這種情況。

這種平台級語言殺還是挺坑人的。

找了一份C++的代碼：

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;
 
int getAns(vector<vector<int>> &nums){
     
    int M = 0x7ffffff;
    int n = nums.size();
    vector<vector<int>> dp(1<<n, vector<int>(n,M));
    dp[1][0] = 0;
    for(int i=1; i<n; i++) dp[1<<i][i] = M;
    for(int i=1; i<(1<<n); i++){
        for(int j=0; j<n; j++){
            if(dp[i][j]!=M){
                for(int k=0; k<n; k++){
                    if((i&(1<<k))==0){
                        dp[i|(1<<k)][k] = min(dp[i|(1<<k)][k], dp[i][j]+nums[j][k]);
                    }
                }
            }
        }
    }
    int ans = M;
    for(int i=1; i<n; i++){
        ans = min(ans, dp[(1<<n)-1][i]+nums[i][0]);
    }
    return ans;
}
int main(){
    int n;
    while(cin>>n){
        vector<vector<int>> edges(n,vector<int>(n,0));
        int x;
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                cin>>edges[i][j];
            }
        }
        cout<<getAns(edges)<<endl;
    }
    return 0;
}