mmdbcn 发表于 2015-12-3 11:16:24

leetcode 【 Merge Sorted Array 】python 实现

  题目:
  Given two sorted integer arrays A and B, merge B into A as one sorted array.
  Note:
You may assume that A has enough space (size that is greater or equal to m + n) to hold additional elements from B. The number of elements initialized in A and B are m and nrespectively.
  
  代码:oj测试通过 Runtime: 58 ms



1 class Solution:
2   # @param Aa list of integers
3   # @param man integer, length of A
4   # @param Ba list of integers
5   # @param nan integer, length of B
6   # @return nothing
7   def merge(self, A, m, B, n):
8         
9         index_total = m + n - 1
10         indexA = m-1
11         indexB = n-1
12         
13         while indexA >= 0 and indexB >= 0 :
14             if A > B:
15               A = A
16               indexA -= 1
17             else:
18               A = B
19               indexB -= 1
20             index_total -= 1
21         
22         if indexA >= 0:
23             while indexA >= 0:
24               A = A
25               indexA -= 1
26               index_total -= 1
27         else:
28             while indexB >= 0:
29               A = B
30               indexB -= 1
31               index_total -= 1
  
  思路:
  采用最基础的两路归并思想。针对A B两个数组维护两个指针。
  这里用到一个技巧是:从后往前遍历。
  这样可以不用再开辟一个m+n空间的数组。
页: [1]
查看完整版本: leetcode 【 Merge Sorted Array 】python 实现