codevs 1506 传话

网上看到的floyed方法,也学到了这样求解传递闭包。(就是时间复杂度较高)

#include <bits/stdc++.h>

typedef long long int ll;

#define PI 3.14159265358979323846
#define INF 0x3f3f3f3f
#define INFLL 0x3f3f3f3f3f3f3f3fll
#define maxn 2000+10
#define mod 2000120420010122

using namespace std;

int n,m;
bool w[maxn][maxn];

void init(){
    cin>>n>>m;
    int a,b;
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        w[a][b]=1;
    }
}

void floyed(){
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                w[i][j]=w[i][j] || (w[i][k] && w[k][j]);
}

void solve(){
    for(int i=1;i<=n;i++){
        if(w[i][i]){
            cout<<'T'<<endl;
        }
        else{
            cout<<'F'<<endl;
        }
    }
}

int main(int argc, char const *argv[]){
    init();
    floyed();
    solve();
    return 0;
}

dfs解法如下,时间相比于上面的解法,优化了很多~

#include <bits/stdc++.h>

typedef long long int ll;

#define PI 3.14159265358979323846
#define INF 0x3f3f3f3f
#define INFLL 0x3f3f3f3f3f3f3f3fll
#define maxn 1000+10
#define mod 2000120420010122

using namespace std;

int n,m;
int flag;
int temp,sucess;
vector<int> vec[maxn];
int visit[maxn];

void dfs(int cur){
    if(cur==temp && flag==0){
        sucess=1;
        return ;
    }
    int next;
    flag=0;
    for(int i=0;i<vec[cur].size();i++){
        next=vec[cur][i];
        if(!visit[next]){
            visit[next]=1;
            dfs(next);
        }
    }
}

int main(int argc, char const *argv[]){
    cin>>n>>m;
    int a,b;
    for(int i=0;i<m;i++){
        scanf("%d%d",&a,&b);
        vec[a].push_back(b);
    }
    for(int i=1;i<=n;i++){
        memset(visit, 0, sizeof(visit));
        temp=i;
        flag=1;
        sucess=0;
        dfs(i);
        if(sucess){
            printf("T\n");
        }else{
            printf("F\n");
        }
    }
    return 0;
}

阅读更多
个人分类: 算法学习
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭