7-28每日一题：第k个除数二叉树中所有距离为K的结点

最新推荐文章于 2023-03-03 15:58:26 发布

Shirandexiaowo

最新推荐文章于 2023-03-03 15:58:26 发布

阅读量75

点赞数

分类专栏：每日一题文章标签：二叉树 dfs leetcode

本文链接：https://blog.csdn.net/MATLAB2020ab/article/details/119190070

版权

每日一题专栏收录该内容

144 篇文章 0 订阅

订阅专栏

本文介绍了两种算法问题的解决方案。第一部分是Acwing平台上的第k个除数问题，通过试除法找到所有质因数并排序，以O(nlogn)的时间复杂度找到第k个除数。第二部分是LeetCode的二叉树中所有距离为K的节点问题，通过遍历构建父节点映射，然后进行深度优先搜索找到目标深度的节点。

摘要由CSDN通过智能技术生成

Acwing第k个除数

思路

试除法求所有约数，对约数排序
时间复杂度： $O(\sqrt{n}logn)$

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;
typedef long long LL;
LL n, k;
int main()
{
    cin >> n >> k;
    vector<LL> d;
    for (LL i = 1; i * i <= n; i ++ )
    {
        if (n % i == 0)
        {
            d.push_back(i);
            if (n / i != i) d.push_back(n / i);
        }
    }
    if (k > d.size()) puts("-1");
    else 
    {
        nth_element(d.begin(),d.begin() + k - 1,d.end());
        // sort(d.begin(),d.end());
        cout << d[k - 1] << endl;
    }
    return 0;
}

Leetcode每日一题：二叉树中所有距离为K的结点

思路：

遍历，保存每个结点的父节点 $O (n)$
从 $t a r g e t$ 开始搜索直到深度为 $k$ 为止，将答案返回

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    unordered_map<int, TreeNode*> pa;
    vector<int> res;

    void find_parents(TreeNode* root)
    {
        if (root == nullptr)
            return ;
        if (root->left)
        {
            pa[root->left->val] = root;
            find_parents(root->left);
        }

        if (root->right)
        {
            pa[root->right->val] = root;
            find_parents(root->right);
        }
    }

    void dfs(TreeNode* node, TreeNode* from, int depth, int k)
    {
        if (node == nullptr) 
            return ;
        if (depth == k)
        {
            res.push_back(node->val);
            return ;
        }
        if (node->left != from)
            dfs(node->left, node, depth + 1, k);
        if (node->right != from)
            dfs(node->right, node, depth + 1, k);
        if (pa[node->val] != from)
            dfs(pa[node->val], node, depth + 1, k);
    }
public:
    vector<int> distanceK(TreeNode* root, TreeNode* target, int k) {
       find_parents(root);

       dfs(target, nullptr, 0, k);

       return res; 
    }
};