ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

数据结构常用算法总结(一)AVL,Dijkstra,Floyd

2022-01-31 12:02:33  阅读:139  来源: 互联网

标签:Node right int Height AVL Dijkstra Floyd root left


一,建立使用AVL树

#include<iostream>
#include<queue>
using namespace std;
struct Node {//二叉树结点
	Node* left;
    Node* right;
	int key;
	Node(int a) {
		key = a;
		left = nullptr;
		right = nullptr;
	}
};
class AvlTree
{ 
public:
    Node* roots;   
    AvlTree() {
        roots = nullptr;
    }
    Node* L(Node* root) {
        Node* temp = root->right;
        root->right = temp->left;
        temp->left = root;
        return temp;
    }
    Node* R(Node* root) {
        Node* temp = root->left;
        root->left = temp->right;
        temp->right = root;
        return temp;
    }
    int get_Height(Node* root) {
        if (root == nullptr)  return 0;
        return max(get_Height(root->left), get_Height(root->right)) + 1;
    }
    Node* insert(Node* root, int key) {
        if (root == nullptr) {
            root = new Node(key);
        }
        else if (root->key > key) {//在左子树中插入
            root->left = insert(root->left, key);
            if (get_Height(root->left) - get_Height(root->right) == 2) {//因为是递归算法,所以这里检测到的就是最小不平衡子树
                if (get_Height(root->left->left) - get_Height(root->left->right) == 1) {
                    root = R(root);
                }
                else if (get_Height(root->left->left) - get_Height(root->left->right) == -1) {
                    root->left = L(root->left);
                    root = R(root);
                }
            }
        }
        else {
            root->right = insert(root->right, key);
            if (get_Height(root->left) - get_Height(root->right) == -2) {
                if (get_Height(root->right->left) - get_Height(root->right->right) == -1) {
                    root = L(root);
                }
                else if (get_Height(root->right->left) - get_Height(root->right->right) == 1) {
                    root->right = R(root->right);
                    root = L(root);
                }
            }
        }
        return root;
    }

    void StoreyLook()//层序遍历
	{
		//初始化队列
		queue<Node> q;
		q.push(*roots);
		while (!q.empty())
		{
			cout<<(q.front()).key<<' ';
			if ((q.front()).left != nullptr)
			{
				q.push(*(q.front().left));
			}
			if ((q.front()).right != nullptr)
			{
				q.push(*(q.front().right));
			}
			q.pop();
		}
	}
    void DFS(Node* root, int target,Node* res=nullptr, int depth=1)//深度优先遍历查找target——先序(根,左,右),顺便计算结点深度
    {
        if (root != nullptr)
        {
            if (root->key == target)
                res = root;
            else
            {
                DFS(root->left,target,res,depth + 1);
                DFS(root->right,target,res,depth + 1);
            }
        }
        if (depth == 1 && res == nullptr)
        {
            cout << "该值不存在" << endl;
        }
    }   
};

int main()
{
    AvlTree obj;
    int k;
    cin >> k;
    for (int i = 0; i < k; i++)
    {
        int value;
        cin >> value;
        obj.roots=obj.insert(obj.roots, value);
    }//插入结点建立AVL树
    obj.StoreyLook();//层序遍历
    cout << endl<<obj.get_Height(obj.roots);
    Node* res=nullptr;
    obj.DFS(obj.roots, 3, res);//在obj树中查找数字3,如果存在,记录3结点的地址到res
}

收获:深刻掌握递归算法。

二,图有关算法

1,迪杰斯特拉算法(重点)

#include<iostream>
#include<vector>
#include<Windows.h>
using namespace std;
const int Max = 9999;
struct edge {
	int cost;
	int eid;
};
void input(int& k,int &m,vector<edge>*map)
{
	cout << "请输入图中有多少结点" << endl;
	cin >> k;//图中有多少结点

	cout << "请输入图中有多少条边" << endl;
	cin >> m;
	cout << "请输入图中所有的边 起点id-终点id-代价,例如 0 1 4" << endl;
	for (int i = 0; i < m; i++)
	{
		int start, end,cost;
		cin >> start >> end>>cost;
		edge a1;
		edge a2;
		a1.eid = start;
		a1.cost = cost;
		a2.eid = end;
		a2.cost = cost;
		map[start].push_back(a2);
		map[end].push_back(a1);
	}
}
void searchLoad(vector<edge>* map, int start,int k)
{
	cout << "开始计算从" << start << "出发,到达所有结点的最短路径" << endl;
	int dist[100];//每个结点距离出发点的最短距离
	int path[100];//记录每个结点的上一条路径
	bool flag[100];//记录每个结点是否被处理过
	//初始化
	for (int i = 0; i < k; i++)
	{
		dist[i] = Max;
		path[i] = -1;
		flag[i] = false;
	}
	dist[start] = 0;
	//贪心求解子问题最优解
	for (int i = 0; i < k; i++)//找第i条路径
	{
		int min = Max;
		int min_id = -1;
		for (int j = 0; j < k; j++){
			if ((!flag[j])&& (dist[j] < min)) {
					min = dist[j];
					min_id = j;
			}
		}//贪心找到了第i个子问题
		flag[min_id] = true;
		if (min_id == -1)return;//说明该图不连通
		for (int t = 0; t < map[min_id].size(); t++){
			int eid = map[min_id][t].eid;
			int cost = map[min_id][t].cost;
			if ((!flag[eid])&& (dist[min_id] + cost < dist[eid])) {
					dist[eid] = dist[min_id] + cost;
					path[eid] = min_id;
			}
		}
	}
	cout << "最短路径如下" << endl;
	//输出路径
	for (int i = 0; i < k; i++)
	{
		cout << path[i] << ' ';
	}
}
int main()
{
	vector<edge> map[100];
	int k;//图中结点的个数
	int m;//图中边的个数
	input(k,m,map);
	int start;//开始城市的id
	cout << "请输入开始城市的id" << endl;
	cin >> start;
	searchLoad(map,start, k);
}

2,BFS算法(单源最短路径算法)每个路径长度默认为1才能算

比较简单,注意先访问,再入队。

3,弗洛伊德算法(多源最短路径)

弗洛伊德算法是一个求解图最短路径的一个动态规划算法。
在这里插入图片描述
如上图:将图用邻接表A存储,用path[i][j]存储从i到j的中转站。现在用动态规划的思想来解决这个问题。
动态规划状态转移递归方程
A[i][j]=min{ A[i][j],A[i][k]+A[k][j]},0<=k<=j。
因此,动态规划的核心算法如下

void floyd(int** A, int** path, int n)
{
	for (int k = 0; k < n; k++) {//考虑以vk为中转点
		for (int i = 0; i < n; i++) {//遍历整个矩阵,i为行号,j为列号
			for (int j = 0; j < n; j++){
				if (A[i][j] > A[i][k] + A[k][j]) {//以vk为中转点的路径长度更短
					A[i][j] = A[i][k] + A[k][j];//更新最短路径长度
					path[i][j] = k;//记录中转点
				}
			}
		}
	}
}

该算法因为是求多源最短路径,算法复杂度为n^3,一般算法题目中涉及图的化不好考察。
——如何理解上面的状态转移方程
观察整个代码运行
1,判断从某个结点直接到某个结点的最短路径
2,判断从某个结点使用某个中转点到某个结点的最短路径
3,判断从某个结点使用再一个中转点到某个结点的路径
4,…

表面上看我们仿佛看的是
vi->vj
vi->v0->vj
vi->v1->vj
vi->v2->vj等等的最短路径
实际上就比如vi->v1->vj这条路径中,v1->vj可不是说v1直接到vj,v1->vj也是一条递归的路径。理解了这层递归,相信整个动态规划递归方程不难理解。

标签:Node,right,int,Height,AVL,Dijkstra,Floyd,root,left
来源: https://blog.csdn.net/weixin_42445051/article/details/122750538

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有