HDU-1572-下沙小面的(2)

B - 下沙小面的(2)
Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Submit

Status

Practice

HDU 1572
Description
前文再续，书接上一题。话说当上小面的司机的Lele在施行他的那一套拉客法则以后，由于走的路线太长，油费又贵，不久便亏本了。(真可怜～)于是他又想了一个拉客的办法。

对于每一次拉客活动，他一次性把乘客都拉上车(当然也不会超过7个，因为位置只有7个)。然后，Lele计算出一条路线(出于某些目的，Lele只把车上乘客的目的地作为这条路线上的站点），把所有乘客都送到目的地(在这路线上不拉上其他乘客)，并且使总路线长度最短。

不过Lele每次都要花很多时间来想路线，你能写个程序帮他嘛？

Input
本题目包含多组测试。最后一组测试后有一个0代表结束。
每组测试第一行有一个整数NCity(3<=NCity<=30)表示下沙一共有多少个站点(站点从0开始标号)。
然后给你一个 NCity * NCity 的矩阵，表示站点间的两两距离。即这个矩阵中第 i 行 第 j 列的元素表示站点 i 和站点 j 的距离。(0<=距离<=1000)
再然后有一个整数K(1<=K<=7)，表示Lele拉上车的人数。
接下来的一行里包括 K 个整数，代表上车的人分别要去的站点。(0<站点<NCity)

注意：
对于每组测试，Lele都是在站点0拉上乘客的。

Output
对于每一组测试，在一行内输出一个整数，表示最短路线的长度。

Sample Input
3
0 1 2
1 0 3
2 3 0
3
1 1 2
0

Sample Output
4

给个地图，求最短路径，数据比较少，就懒得动脑子了，直接全排列，然后逐个搜索一遍，每次只保留路径最小值。

刚开始学搜索，还不太适应，小练写完后，再写个搜索算法的总结

代码

#include<stdio.h>
#include<string.h>
#include<string>
#include<stack>
#include<queue>
#include<math.h>
#include<limits.h>
#include<malloc.h>
#include<iostream>
#include<algorithm>
using namespace std;
//以后不要改头文件
int main()
{
    int n;
    while(~scanf("%d",&n)&&n)
    {
        int dfs[33][33];
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++)
                scanf("%d",&dfs[i][j]);//存储地图数据
        int k;
        scanf("%d",&k);
        int s1[10];//接收站点
        int s2[10];//接收去重后的站点
        for(int i=0; i<k; i++)
            scanf("%d",&s1[i]);
        sort(s1,s1+k);//先升序，方便去重和调用全排列函数
        int i=0,j=0;
        for(; i<k; i++)
            if(s1[i]!=s1[i+1])
                s2[j++]=s1[i];//去重后存到s2
        k=j;//获取合并后的站点数
        int sum=INT_MAX;//记录距离,初始值为无穷大
        int flag;//临时存储距离，满足条件就传递给sum
        do
        {
            flag=dfs[0][s2[0]];
            for(int p=1;p<k;p++)
                flag=flag+dfs[s2[p-1]][s2[p]];
            sum=min(sum,flag);//每次只取最小值
        }while(next_permutation(s2,s2+k));//全排列，暴力搜索
        printf("%d\n",sum);
    }
}

比较懒，全排列直接用函数代替了。。。