dfs(小天狼星的访问,CF - 1249-B2)

小天狼星的访问

Description

经过数月的准备，小天狼星，一个被诬陷的杀人犯，准备闯入霍格沃茨见见他的侄子。霍格沃茨的地图呈一颗树状分布。每个房间由若干跳过道通向其他房间。由于小天狼星想尽快找到哈利：

0.他会从房间0开始找
1.他总是会选择离自己最近的房间找
2.如果没找到，则继续选最近的房间深入
3.如果已没有房间可走，则返回到上一个房间，继续选择（往回走也算时间哦）。
4.当然，除了往回走,小天狼星是不会去一个房间两次的。

Input

第1行，n房间个数，p哈利所在的房间。(p≤n<100)
第2∼n行，每行3个整数s,t,l。从房间s到房间t的时间l。(s≠t, 0≤s<t<n, 0<l≤10000)

Output

输出找到哈利的时间(开始时间为0)。

Input

5 2
0 1 1
0 2 2
1 3 3
1 4 4

Output

18

思路:

一步一步dfs就行,调了好久QAQ

代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <cstring>
#include <set>
#include <map>
#include <sstream>
#define LL long long
#define mem(f, x) memset(f,x,sizeof(f)) 
#define fo(i,a,n) for(int i=a;i<n;++i)
#define fo2(i,a,n) for(int i=a;i<=n;++i)
const int INF = 0x3f3f3f3f;
using namespace std;
template<class T>inline void read(T &x){
    x=0; char c=getchar(); bool f=0;
    while(!isdigit(c))f^=c=='-',c=getchar();
    while(isdigit(c))x=(x<<3)+(x<<1)+(c^48),c=getchar();
    if(f)x=-x;
}
const int MAXN=105;
int n,p,ans=0,d[MAXN][MAXN];
bool vis[MAXN];
void dfs(int i){
	while(true){
		int k=0,mn=1e9;
		for(int j=1;j<=n;j++)
			if(!vis[j] && d[i][j]>0 && mn>d[i][j])
				mn=d[i][j],k=j;
		if(mn==1e9) return;
//		if(d[i][k]==1e9) return;会WA,用k做标记可能会出问题 
		vis[k]=true;
		ans+=d[i][k]; 
		if(k==p){printf("%d",ans);exit(0);}
		dfs(k);
		ans+=d[i][k]; 
	}
}

int main(){
	read(n);read(p);
	int s,t,l;
	if(n==1){
		cout<<"0";return 0;
	}
	fo(i,0,n-1){
		read(s);read(t);read(l);
		d[s][t]=d[t][s]=l;
	}
	vis[0]=true;
	dfs(0);
	return 0;
}

注释版本:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <cstring>
#include <set>
#include <map>
#include <sstream>
#define LL long long
#define mem(f, x) memset(f,x,sizeof(f)) 
#define fo(i,a,n) for(int i=a;i<n;++i)
#define fo2(i,a,n) for(int i=a;i<=n;++i)
const int INF = 0x3f3f3f3f;
using namespace std;
template<class T>inline void read(T &x){
    x=0; char c=getchar(); bool f=0;
    while(!isdigit(c))f^=c=='-',c=getchar();
    while(isdigit(c))x=(x<<3)+(x<<1)+(c^48),c=getchar();
    if(f)x=-x;
}
const int MAXN=105;
int n,p,ans=0,d[MAXN][MAXN];
bool vis[MAXN];
void dfs(int i){
	while(true){
		int k=0,mn=1e9;
		for(int j=1;j<=n;j++)//选择离自己最近的房间找
			if(!vis[j] && d[i][j]>0 && mn>d[i][j])
				mn=d[i][j],k=j;
		if(mn==1e9) return;//已没有房间可走
//		if(d[i][k]==1e9) return;会WA,用k做标记可能会出问题 
		vis[k]=true;
		ans+=d[i][k];//从 i 到 k 
		if(k==p){printf("%d",ans);exit(0);}
		//没有找到哈利,深入去找 
		dfs(k);
		//此路不通,往回走
		ans+=d[i][k];//无路可走时从 k 回到 i
		//走回去不意味着函数结束,要接着找房间 
	}
}

int main(){
	read(n);read(p);
	int s,t,l;
	if(n==1){//细节 
		cout<<"0";return 0;
	}
	fo(i,0,n-1){
		read(s);read(t);read(l);
		d[s][t]=d[t][s]=l;
	}
	vis[0]=true;//从房间0开始找
	dfs(0);
	return 0;
}

CF - 1249-B2

B2. Books Exchange (hard version)

大意:

有n个孩子，每个孩子都有一本书，每天结束之后，第i个孩子就会把自己的书给第p[i]个孩子,问：第i个（1<=i<=n）孩子的书最少要经过多少天才会重新回到他手里(注意p数组是n的一个全排列)

分析:(思路借鉴传送门)

我们可以发现每个点的路径都组成了一个环，每个环上的所有点的答案都是相同的

因此我们可以进行如下的优化：
先选择一个点进行路径模拟，同时记录下该路径上所有的点。因为这条路径上所有的点的答案都是相同的，所以把这些点的答案全部赋成与模拟点相同的答案即可

代码:

#include<iostream>
#include<cstring>
using namespace std;
const int N=200005;
int n,t;
int d[N],a[N],sum[N];
int dfs(int x,int step){
	d[x] = 1;
	if(d[a[x]] == 1) return sum[x] = step;
	else return sum[x] = dfs(a[x],step+1);
}
int main(){
	cin>>t;
	while(t--){
		cin>>n;
		memset(d,0,sizeof d);
		memset(sum,0,sizeof sum);
		for(int i=1;i<=n;i++) cin>>a[i];
		for(int i=1;i<=n;i++) if(d[i] == 0) dfs(i,1);
		for(int i=1;i<=n;i++) cout<<sum[i]<<" ";
		puts("");
	}
	return 0;
} 

注释版本:

#include<iostream>
#include<cstring>
using namespace std;
const int N=200005;
int n,t;
int d[N],a[N],sum[N];
//d[x]记录x位置是否被访问过,sum[x]记录x位置的答案 
//dfs求x位置上的数字最少循环几次回来,返回这个数字到sum[x] 
int dfs(int x,int step){
	d[x] = 1;//这个位置将被访问,不是这个位置上的数将被访问
	if(d[a[x]] == 1) return sum[x] = step;//如果这个位置上的数被访问过了
	else return sum[x] = dfs(a[x],step+1);
}
int main(){
	cin>>t;
	while(t--){
		cin>>n;
		memset(d,0,sizeof d);
		memset(sum,0,sizeof sum);
		for(int i=1;i<=n;i++) cin>>a[i];
		for(int i=1;i<=n;i++) if(d[i] == 0) dfs(i,1);
		for(int i=1;i<=n;i++) cout<<sum[i]<<" ";
		puts("");
	}
	return 0;
}