剑指offer推荐系统遇到深度学习程序员

java&python版剑指offer(一)

2018-02-12  本文已影响0人  文哥的学习日记

本文按照牛客网的顺序,牛客网剑指offer刷题网址:https://www.nowcoder.com/ta/coding-interviews

本文涉及的题目:
1、二维数组中的查找
2、从尾到头打印链表
3、重建二叉树

1、二维数组中的查找

在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

思路
类似于二分查找,根据题目,如果拿数组中任意一个元素与目标数值进行比较,如果该元素小于目标数值,那么目标数值一定是在该元素的下方或右方,如果大于目标数值,那么目标数值一定在该元素的上方或者左方。

对于二分查找来说,每次比较只能移动一个指针,同时,指针的移动方向必须在查找过程中保持一致。在二维数组的查找中,两个指针是一个上下方向移动,一个是左右方向移动的。两个指针可以从同一个角出发,也可以从不同的角出发。这里我们只考虑从同一个角出发的情况。

假设我们从左上角出发,也就是row=0 和 col=0,如果元素小于目标数值,我们会将row往下移或着col往右移,这样,被屏蔽的区域可能会是目标元素所在的区域。比如row+=1,那么第一行除左上角以外的元素就被忽略了,如果col+=1,那么第一列出左上角以外的元素就被忽略了。因此这样是不可行的。

所以本题从右上角出发寻找解题思路,即开始时col=len(array[0])-1,row=0,col往左移动,row往下移动,当某元素小于目标数值时,col-=1,而当某元素大于目标数值时,row+=1。代码如下:

java

public class Solution {
    public boolean Find(int target, int [][] array) {
        int row = 0;
        int col = array[0].length-1;
        while(row<array.length && col>=0){
            if(array[row][col] == target)
                return true;
            else if(array[row][col] > target)
                col -= 1;
            else
                row += 1;
        } 
        return false;
    }
}

python

# -*- coding:utf-8 -*-
class Solution:
    def Find(self, target, array):
        # write code here
        if not array:
            return False
        row = 0
        col = len(array[0])-1
        while row < len(array) and col >=0:
            if array[row][col] == target:
                return True
            elif array[row][col] < target:
                row = row + 1
            else:
                col = col - 1
        return False
           

2、从尾到头打印链表

问题描述
输入一个链表,从尾到头打印链表每个节点的值。

思路解析
使用栈从头到尾push链表的元素,然后pop所有的元素到一个list中并返回

代码实现
java

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> printListFromTailToHead(ListNode listNode) {
        ArrayList<Integer> arr = new ArrayList<Integer>();
        ListNode p = listNode;
        ArrayList<Integer> stack = new ArrayList<Integer>();
        while(p!=null){
            stack.add(p.val);
            p = p.next;
        }
        int n = stack.size();
        for(int i=n-1;i>=0;i--){
            arr.add(stack.get(i));
        }
        return arr;
        
    }
}

python

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def printListFromTailToHead(self, listNode):
        # write code here
        if not listNode:
            return []
        p = listNode
        stack = []
        res = []
        while p:
            stack.append(p.val)
            p = p.next
        for i in range(len(stack)-1,-1,-1):
            res.append(stack[i])
        return res

3、重建二叉树

问题描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

思路解析
根据前序遍历的结果,对于输入的遍历序列,第一个数一定是根结点,那么我们去中序遍历的结果中找到该根节点,中序遍历结果中根结点左边的是左子树的节点,右边的是右子树的节点,这样很容易使用递归的思路实现二叉树的重建。

代码实现
java

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre == null || in == null || pre.length == 0 || in.length == 0){
            return null;
        }
        return buildTree(pre, in, 0, pre.length - 1, 0, in.length - 1);
    }
    public TreeNode buildTree(int[] pre, int[] in, int preStart, int preEnd, int inStart, int inEnd){
        TreeNode root = new TreeNode(pre[preStart]);
        int rootIn = 0;
        for(; rootIn < in.length; rootIn++){
            if(in[rootIn] == root.val){
                break;
            }
        }
        int leftLength = rootIn - inStart;
        int rightLength = inEnd - rootIn;
        if(leftLength > 0){
            root.left = buildTree(pre, in, preStart + 1, preStart + leftLength, inStart, rootIn - 1);           
        }
        if(rightLength > 0){
            root.right = buildTree(pre, in, preStart + leftLength + 1, preEnd, rootIn + 1, inEnd);   
        }
         
        return root;
    }
}

python

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if not pre or not tin:
            return None
        root = TreeNode(pre[0])
        i = 0
        while i < len(tin):
            if tin[i] == pre[0]:
                break
            i = i + 1
        index = i
        root.left = self.reConstructBinaryTree(pre[1:index+1],tin[:index])
        root.right = self.reConstructBinaryTree(pre[index+1:],tin[index+1:])
        return root
上一篇下一篇

猜你喜欢

热点阅读