牛客刷题 day1（寻找第K大，镜像二叉树）

牛客刷题 day1

寻找第K大

题目

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在

解题思路

像快排一样，但是每次排完需要与k进行比较，具体比较看代码。
1.由于一般排序是从小到大排的，所以注意从小到大排这里是 n-i 与 k进行比较；
2.或者从大到小排，则比较 i+1 与 k的大小；

核心算法

class Solution:
    def findKth(self, a, n, K):
        return self.quickSort(a, 0, n - 1,n, K)

    def quickSort(self, a, low, high, n, k):
        pivot = a[low]
        i = low
        j = high
        while i < j:
            while i < j and a[j] < pivot:
                j -= 1
            while i < j and a[i] > pivot:
                i += 1
            if i < j:
                a[i], a[j] = a[j], a[i]
        if i + 1 == k:
            return a[i]
        elif i + 1 > k:
            return self.quickSort(a, low, i - 1, n, k)
        elif i + 1 < k:
            return self.quickSort(a, i + 1, high, n, k)

镜像二叉树

题目

操作给定的二叉树，将其变换为源二叉树的镜像
比如： 源二叉树
8
/
6 10
/ \ /
5 7 9 11
镜像二叉树
8
/
10 6
/ \ /
11 9 7 5

解题思路

使用递归思想：先镜像当前root左右节点，然后再递归镜像左孩子和右孩子即可。

核心算法

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param pRoot TreeNode类 
# @return TreeNode类
#
class Solution:
    def Mirror(self , pRoot ):
        # write code here
        if not pRoot:
            return 
        self.swap(pRoot)
        return pRoot
    def swap(self,root):
        if root!=None:
            left = root.left
            right = root.right
            root.left = right
            root.right = left
            self.swap(left)
            self.swap(right)