这道题就是找出1到k有多少条路径,看起来没什么坑,可是写起来总是TLE。其实这道题的数据会专门设置一种数据,那就是1和除了k以外的点都是连通的,可是就是和k不连通。所以我们会浪费很多时间做无用功,其实只要刚开始就判断1和k是否连通就可以避免这种情况。
此外在找出1的k的路径总数的时候,刚开始我是这么写的
for (int v = 1; v <= maxu; v++)//maxu是最大的点
{
if (G[u][v] && !vis[v])
{
ans.push_back(v);
vis[v] = 1;
dfs2(v);
ans.pop_back();
}
}
乍一看起来好像是对的,可是仔细一想在dfs2(v)之后没有及时的将v标记成0,这样一来1到k的总路径数会减少。所以应该dfs2(v)后面加上vis[v] = 0。详细代码如下:
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <vector>
#define maxn 25
using namespace std;
typedef vector<int> vec;
int G[maxn][maxn];
int k, maxu, cnt;
int vis[maxn];
vec ans;
void dfs1(int u)//判断1到k是否连通
{
vis[u] = 1;
for (int v = 1; v <= maxu; v++)
{
if (G[u][v] && !vis[v])
dfs1(v);
}
}
void dfs2(int u)//dfs找出1到k的各种路径
{
vis[u] = 1;
if (u == k)
{
int len = ans.size();
for(int i = 0; i < len; i++)
printf("%d%c", ans[i], i< len - 1?' ' : '\n');
cnt++;
return ;
}
for (int v = 1; v <= maxu; v++)
{
if (G[u][v] && !vis[v])
{
ans.push_back(v);
dfs2(v);
vis[v] =0;//一定不要忘了这句,否则路径总数会减少
ans.pop_back();
}
}
}
int main()
{
int Case = 1;
while (~scanf("%d", &k))
{
memset(G, 0, sizeof(G));
maxu = 0;
int u, v;
while (~scanf("%d%d", &u, &v) && u && v)//maxu为输入的最大点
{
G[u][v] = G[v][u] = 1;
maxu = max(maxu, u > v ? u : v);
}
printf("CASE %d:\n", Case++);
memset(vis, 0, sizeof(vis));
dfs1(1);
if(vis[k])//如果连通,则继续dfs
{
memset(vis, 0, sizeof(vis));
cnt = 0;
ans.clear();
ans.push_back(1);
dfs2(1);
printf("There are %d routes from the firestation to streetcorner %d.\n", cnt, k);
}else//如果不连通,直接输出0
{
printf("There are 0 routes from the firestation to streetcorner %d.\n", k);
}
}
return 0;
}