洛谷 P1690 貪婪的Copy

P1690 貪婪的Copy

題目描述

Copy從盧牛那裏據說在一片叫yz的神的領域埋藏着很多寶藏，因而Copy來到了這個被劃分爲個區域的神地。盧牛告訴了Copy這裏共有個寶藏，分別放在第Pi個(1<=Pi<=N)區域。Copy還得知了每一個區域之間的距離。如今Copy從1號區域出發，要得到全部的寶藏併到n號區域離開。Copy很懶，只好來找你爲他尋找一條合適的線路，使得他走過的距離最短。

輸入輸出格式

輸入格式：

 

第一行一個正整數N(1<=N<=100)

接下來一個N*N的矩陣，第i+1行第j列的數字表示區域i,j之間的距離。每一個距離用空格隔開，距離保證i,j<=1000。請注意的i to j距離並不必定等於j to i的距離。

第N+2行一個整數P(0<=P<=10)。

第N+3行共P個用空格隔開的整數，表示有寶藏的區域編號。

 

輸出格式：

 

一個整數，爲Copy得到所有寶藏須要的最短距離。數據保證答案小於等於maxlongint。

 

輸入輸出樣例

輸入樣例#1：

樣例輸入1
2
0 4
5 0
2
1 2

樣例輸入2
3
0 2 6
1 0 4
7 10 0
1
2

輸出樣例#1：

樣例輸出1
4

樣例輸出1
6

說明

對30%的數據，1<=n<=15，其他如題所述。

對100%的數據，所有數據範圍如題所述。

思路：floyd預處理+dfs

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 101
using namespace std;
int n,m;
int vis[MAXN],have[MAXN];
long long ans=0x7f7f7f7f;
long long map[MAXN][MAXN];
void dfs(long long num,int tot,int pre){
    if(num+map[pre][n]>ans)    return ;
    if(tot==m){
        ans=num+map[pre][n];
        return ;
    }
    for(int i=1;i<=m;i++)
        if(!vis[i]){
            vis[i]=1;
            dfs(num+map[pre][have[i]],tot+1,have[i]);
            vis[i]=0;
        }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            scanf("%lld",&map[i][j]);
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(i!=j&&i!=k&&map[i][j]>map[i][k]+map[k][j])
                    map[i][j]=map[i][k]+map[k][j];
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
        scanf("%d",&have[i]);
    dfs(0,0,1);
    cout<<ans;
}