1131. Subway Map (30)

1131. Subway Map (30)
更新（2017.9.14）
思路：dfs回溯

#include <iostream>
#include <vector>
#include <cstdio>
using namespace std;
struct node
{
    int line,s,e;
    node(int xl,int xs,int xe):line(xl),s(xs),e(xe){}
};
int searchLine[10000][10000];//判断两个站点在哪条线上
int n,q,minstation,mintransfer;//线个数，请求个数，最小的站点，最小的换乘
vector<vector<int>> v(10000);//图的邻接表
vector<node> ans,recp;//最终结果路径和中间记录路径
bool visit[10000];//判断是否已经访问数组
void subwayMap(int head,int cur,int end,int staioncnt,vector<node> &recp,int pre)
{
    if(cur==end&&(staioncnt<minstation||(staioncnt==minstation&&mintransfer>(int)recp.size()+1)))
    {
        minstation=staioncnt;
        ans=recp;
        ans.push_back(node(searchLine[pre][end],head,end));
        mintransfer=ans.size();
        return;
    }
    visit[cur]=1;
    for(auto next:v[cur])
    {
        if(visit[next]) continue;
        if(pre!=cur&&searchLine[pre][cur]!=searchLine[cur][next])//线路不同，cur为换乘节点
        {
            recp.push_back(node(searchLine[pre][cur],head,cur));//收集上一个线路信息：几号线，首站，尾站
            subwayMap(cur,next,end,staioncnt+1,recp,cur);
            recp.pop_back();
        }
        else subwayMap(head,next,end,staioncnt+1,recp,cur);//线路相同，首站不变;
    }
    visit[cur]=0;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        int k,idx,p;
        scanf("%d %d",&k,&p);
        for(int j=0;j<k-1;++j)
        {
            scanf("%d",&idx);
            searchLine[p][idx]=searchLine[idx][p]=i;//两个相邻节点的线路
            v[p].push_back(idx);
            v[idx].push_back(p);
            p=idx;
        }
    }
    scanf("%d",&q);
    while(q--)
    {
        int s,e;
        scanf("%d %d",&s,&e);
        minstation=mintransfer=0x7fffffff;
        subwayMap(s,s,e,0,recp,s);
        printf("%d\n",minstation);
        for(auto x:ans) printf("Take Line#%d from %04d to %04d.\n",x.line,x.s,x.e);
    }
    return 0;
}

分割线———————————以前ac代码
啊啊啊啊~，一开始提交一看，竟然1,4,5测试例都错了，还以为我算法错了，后面才发现，输出是四位啊，所以呢，一定要注意格式输出啊（printf大法好，逃~）
分析：
如果我们首先忽略地铁线的这条信息，那么，这里本身就是一张图，有n多个节点，要求就是给你两个节点，你要找到他们之间经过节点最少的路径，显然，利用dfs进行遍历是首先要想到的
那么，如何确定是哪条地铁线呢？想想还是用二维数组叭，简单又直接
解题思路：
dfs深搜+回溯剪枝
恩，前面有些题很相似（逃~）

#include <cstdio>
#include <vector>
using namespace std;
const int maxv=10000;
int arc[maxv][maxv];//储存在哪条地铁线上
int minscnt=0x7fffffff,mintrans=0x7fffffff;//最小的经站和最小换乘
int lastline=0,transfercnt=0;//上一条地铁线，
vector<int> v[maxv];//邻接表
vector<int> ans,line;//ans保存结果路径,line保存当前路径
void dfs(const int s,const int e,int vis[],int stationcnt,int transfercnt,int lastline)
{
    vis[s]=1;
    line.push_back(s);//保存当前路径
    if(s==e)//找到终点
    {
        if(stationcnt<minscnt||(stationcnt==minscnt&&transfercnt<mintrans))
        {
            minscnt=stationcnt;
            mintrans=transfercnt;
            ans.clear();
            ans=line;
        }
    }
    for(auto it=v[s].begin();it!=v[s].end();++it)
    {
        if(!vis[*it])//判断transfercnt
        {
            lastline==0?dfs(*it,e,vis,stationcnt+1,1,arc[s][*it]):
            lastline==arc[s][*it]?
            dfs(*it,e,vis,stationcnt+1,transfercnt,arc[s][*it]):
            dfs(*it,e,vis,stationcnt+1,transfercnt+1,arc[s][*it]);
        }
    }
    vis[line.back()]=0;//回置0
    line.pop_back();//扔掉它
}
int main()
{
    int n,k;
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        int m,u,w;
        scanf("%d %d",&m,&u);
        for(int j=2;j<=m;++j)
        {
            scanf("%d",&w);
            arc[u][w]=arc[w][u]=i;//储存line
            v[u].push_back(w);//邻接表
            v[w].push_back(u);
            u=w;
        }
    }
    scanf("%d",&k);
    while(k--)
    {
        int s,e;
        scanf("%d %d",&s,&e);//接下来5行进行每次初始化
        int vis[maxv]={0};
        minscnt=0x7fffffff;
        mintrans=0x7fffffff;
        line.clear();
        ans.clear();
        dfs(s,e,vis,1,1,0);
        printf("%d\n",ans.size()-1);
        auto it=ans.begin(),p=it+1;
        int last=arc[*it][*p];
        while(true)//按要求打印
        {
            while(p!=ans.end()&&arc[*(p-1)][*p]==last)++p;
            printf("Take Line#%d from %04d to %04d.\n",last,*it,*(p-1));
            if(p==ans.end())break;
            it=p-1;
            last=arc[*it][*p];
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/xLester/p/7570327.html