k路有序数组归并Java实现

最新推荐文章于 2022-03-21 23:42:23 发布

zfyseu1

最新推荐文章于 2022-03-21 23:42:23 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/zfyseu1/article/details/77930908

算法专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种数组合并算法，包括基本的两数组合并方法merge及多数组合并方法k_path_merge。通过递归方式将多个数组逐步合并为一个有序数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

public static int[] merge(int[] A,int[] B)
	{
		int[] result=new int[A.length+B.length];
		
		int lpos=0,rpos=0;
		for(int i=0;i<result.length;i++)
		{
			if(lpos<A.length&&rpos<B.length)
			{
				if(A[lpos]<=B[rpos])
				{
					result[i]=A[lpos++];
				}
				else
				{
					result[i]=B[rpos++];
				}
				
			}
			else if(lpos<A.length)
			{
				result[i]=A[lpos++];
			}
			else if(rpos<B.length)
			{
				result[i]=B[rpos++];
			}
		}
		return result;
	}
	
	public static int[] k_path_merge(ArrayList<int[]> nums)
	{	
		int m=nums.size();
		int len=0;
		for(int i=0;i<m;i++)
		{
			len+=nums.get(i).length;
		}
		int[] result=new int[len];
		
		int k=m;
		while(k>1)
		{
			ArrayList<int[]> temp=new ArrayList<int[]>();
			for(int i=0;i<k;i+=2)
			{
				if(i+1<k)
				{
					int[] current=merge(nums.get(i),nums.get(i+1));
					temp.add(current);
				}
				else
				{
					temp.add(nums.get(i));
				}
			}
			nums.clear();
			for(int j=0;j<temp.size();j++)
			{
				nums.add(temp.get(j));
			}
			k=nums.size();
		}
		return nums.get(0);
	}