UVA 208(dfs+剪枝)

刚开始做的时候没有剪枝,直接上来就是dfs,因为这种题太熟悉,估计像我这种菜鸡做这种题不带思考,直接上来就是dfs。
在TLE的一发后,意识到给的数据可能是个稠密图,dfs次数过多,肯定要超时,所以就得剪枝了,那么怎么减呢,很简单,先思考为什么会导致dfs次数过多,因为dfs了很多不必要的点,所以就是对不必要的点进行修剪,而判断一个点为不必要的点,那么这个点也就是肯定dfs到不了终点,现在看就有点并查集找爹的感觉了,所以从终点出发,进行dfs,对所能联通的点进行标记,此时得到的联通的点就是下一步剪枝过程中的判断依据了。
代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<cmath>
#include<cstdlib>
#include<list>
#include<queue>
#define mm(a,b) memset(a,b,sizeof(a))
#define ACCELERATE (ios::sync_with_stdio(false),cin.tie(0))
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
using namespace std;

bool vis[25];
list<int> v[25];
int n;
int num;
int ans[1000];
bool book[22];

void judge(int k){
    book[k]=1;
    for(list<int>::const_iterator it=v[k].begin();it!=v[k].end();it++){
        if(!book[*it]) judge(*it);
    }
}

void dfs(int i,int k){
    for(list<int>::const_iterator it=v[i].begin();it!=v[i].end();it++){
        if(ans[k-1]==n){
            for(int j=0;j<k-1;j++){
                printf("%d ",ans[j]);
            }
            printf("%d\n",ans[k-1]);
            num++;
            return;
        }
        if(!vis[*it]&&book[*it]){
            ans[k]=*it;
            vis[*it]=1;
            dfs(*it,k+1);
            vis[*it]=0;
        }
    }
}

int main()
{
    int a,b;
    int casei=1;
    while(scanf("%d",&n)!=EOF){
        mm(vis,0);
        mm(book,0);
        num=0;
        while(scanf("%d%d",&a,&b)&&(a||b)){
            v[a].push_back(b);
            v[b].push_back(a);
        }
        printf("CASE %d:\n",casei++);
        for(int i=1;i<=21;i++){
            if(!v[i].empty()) v[i].sort();
        }
        vis[1]=1;
        ans[0]=1;
        judge(n);
        dfs(1,1);
        printf("There are %d routes from the firestation to streetcorner %d.\n",num,n);
        for(int i=1;i<=21;i++){
            if(!v[i].empty()) v[i].clear();
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40679299/article/details/80687191