洛谷【数据结构1-2/4】二叉树/图

二叉树

P4913 【深基16.例3】二叉树深度

#include<bits/stdc++.h>
using namespace std;
const int maxn = 100010;

struct node{
   
	int left,right;
}tree[maxn]; 
int n,ans;
void dfs(int id,int deep){
   
	if(id == 0) return;//到达叶子结点返回 
	ans = max(ans,deep);
	dfs(tree[id].left,deep+1);
	dfs(tree[id].right,deep+1);
}
int main(){
   

	scanf("%d",&n);//节点数
	for(int i=1;i<=n;i++){
   
		scanf("%d %d",&tree[i].left,&tree[i].right);
	}
	dfs(1,1);
	printf("%d\n",ans);
	return 0;
}

P1229 遍历问题

已知前序和后序遍历序列，求中序遍历序列的个数。

只知道前序和后序序列，中序序列的不确定性在于不知道子节点是左子树还是右子树。

a b c
c b a

根节点为a，根节点的左子树一定是以b为根节点，此时若b恰好为后序序列的a之前一位，那么说明根节点a没有第二棵子树，此时子树bc可以为左子树也可以为右子树，因此有两种情况。

a b c
b c a

此时，则bc分别为a的左右子树，中序序列固定，只有一种情况。

因此使用DFS对一棵树进行深搜，左子树的情况*右子树的情况 * 总根节点a的情况即为总的中序序列个数。

递归边界为：此时树只有一个结点或无节点，则只有一种情况。

int ltree = index[s1[st1+1]]-st2+1;

计算左子树的长度，需要一个index数组记录下来后序序列中每个字符的位置。

for(int i =0;i<len;i++){
   
		index[s2[i]] = i;//记录s2中每个字符的位置 
	}

此时左子树的根节点在后序序列中的位置-后序序列开头index+1，即为左子树的结点个数。

if(st1+ltree == en1) k=2;

如果先序序列根节点index+左子树结点个数== 先序序列的最后一个节点index，则根节点只有一个子树，此时子树情况有两种，可以为左或右。

之后再对先序序列根节点的左子树和右子树分别进行dfs，得到每一层的情况数，再相乘即为最后的结果。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1010;

string s1,s2;
int index[maxn];
int dfs(int st1,int en1,int st2,int en2){
   
	if(st1>=en1) return 1;//只有一个结点或者无节点，是一种情况
	int ltree = index[s1[st1+1]]-st2+1;//得到st1根节点左子树的长度
	int k =1;//假设子节点只有一种情况 
	if(st1+ltree == en1) k=2;//如果st1只有一个子节点，则可以为左节点也可以为右节点，k=2种情况 
	return (dfs(st1+1,st1+ltree,st2,st2+ltree-1) * dfs(st1+ltree+1,en1,st2+ltree,en2) ) * k;
}
int main(){
   
	cin>>s1;//前序序列 
	cin>>s2;//后序序列 
	int len = s2.length();
	for(int i =0;i<len;i++){
   
		index[s2[i]] = i;//记录s2中每个字符的位置 
	}
	int cnt =dfs(0,s1.length()-1,0,len-1);
	printf("%d\n",cnt);
	return 0;
}

图

P5318 查找文献

int n,m;//n篇文章,m个引用关系 
vector<int> adj[maxn]; //使用邻接表存图 
bool vis[maxn];
bool inq[maxn];

void dfs(int u){
   
	vis[u] = true