codeforces 34D - Road Map DFS

              把相关联的两个点记下来,然后dfs.这个题n是5*10^4,无法开一个num[n][n]的二维数组,但是了可以用vector

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#define  MAXN 50010
using namespace std;
vector<int>st[MAXN];
int n,r1,r2,ans[MAXN],vis[MAXN];
void dfs(int cur)
{
    if(vis[cur])  return ;
    vis[cur]=true;
    for(int i=0; i<(int)st[cur].size(); i++)
    {
        int a=st[cur][i];
        if(vis[a])  continue;
        ans[a]=cur;
        dfs(a);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    cin>>n>>r1>>r2;
    for(int i=1; i<=n; i++)
    {
        if(i==r1)  continue;
        int a;
        cin>>a;
        st[a].push_back(i);
        st[i].push_back(a);
    }
    memset(ans,0,sizeof(ans));
    memset(vis,false,sizeof(vis));
    dfs(r2);
    int flag=0;
    for(int i=1; i<=n; i++)
    {
        if(i==r2)  continue;
        if(flag)  cout<<" ";
        cout<<ans[i];
        flag=1;
    }
    cout<<endl;
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值