合并有序数组java

题目：

给定两个排序后的数组 A 和 B，其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法，将 B 合并入 A 并排序。（不使用额外空间）

初始化 A 和 B 的元素数量分别为 m 和 n。

输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6],       n = 3

输出: [1,2,2,3,5,6]

1、暴力法（能AC）
直接将B中数据插入A中后部，然后sort排序。

import java.util.*;

public void merge(int[] A, int m, int[] B, int n) {
	int j=0;
	for(int i=m;i<A.length;i++){
		if(j<n){
			A[i] = B[j++];
		}else{
			break;
		}
	}
	Arrays.sort(A);
}

2、直接归并。此题可以使用归并排序合并部分。

public void merge(int[] A, int m, int[] B, int n) {
	int i = m-1;
	int j = n-1;
	while(i>=0&&j>=0){
		if(A[i]>B[j]){
			A[i+j+1] = A[i--];
		}else{
			A[i+j+1] = B[j--];
		}
	}
	//i=0或者j=0时跳出while
	//i=0时 A中m个元素全都放在了对应的位置。B中剩余的是数组下标与A对应的小元素，合并入A中即可。
	//j=0时 B中n个元素全部放在了A中，合并结束。
	while(j>=0){
		A[j] = B[j--];
	}
}

3、A中数据不是分布在数组的前部。

例如：
 A = {-3, 5, INVALID_NUM, 7, INVALID_NUM, 10, INVALID_NUM, 11, INVALID_NUM}
 B = {-1, 2, 6, 12}
最后：A = {-3, -1, 2, 5, 6, 7, 10, 11, 12}

思路：
1、可以先把A中无效数字移动到后部。
（原地算法将指定元素移动到数组末尾，保持非指定元素的相对位置）
2、归并A 、B

public void merge(int[] A, int m, int[] B, int n) {

	//移动INVALID_NUM元素到数组末尾
	int index = 0;
	for(int i=0;i<A.length;i++){
		if(A[i]!=INVALID_NUM){
			A[index++] = A[i];
		}
	}
	for(int i=index;i<A.length;i++){
		A[i] = INVALID_NUM;
	}
	
	//归并
	int i = m-1;
	int j = n-1;
	while(i>=0&&j>=0){
		if(A[i]>B[j]){
			A[i+j+1] = A[i--];
		}else{
			A[i+j+1] = B[j--]
		}
	}
	while(j>=0){
		A[j] = B[j--];
	}
	
}

参考链接：移动元素到数组后部 java

ex 腾讯面试题

输入 
[-1, 0, 1, 1, -1] 

输出 
所有的1放前面，所有的0放后面。 
[1, 1, -1, -1, 0] 

分析： 将所有1移动到前部，0移动到后部。可以借鉴快速排序中间的思路，使用双指针方式。

public void move_num(int[] num){
	int left = 0;
	int right = num.length-1;
	int i = 0;
	while(i<=right){
		if(num[i]==1){
			swap(num,left++,i); 
		}else if(num[i]==0){
			swap(num,right--,i);
		}else{
			++i;
		}
	}
}