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;
}