拼接最大数

问题描述 :
给定长度分别为 m 和 n 的两个数组,其元素由 0-9 构成,表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数,要求从同一个数组中取出的数字保持其在原数组中的相对顺序。求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。
说明: 请尽可能地优化你算法的时间和空间复杂度。

示例 1:
输入:
nums1 = [3, 4, 6, 5]
nums2 = [9, 1, 2, 5, 8, 3]
k = 5
输出:
[9, 8, 6, 5, 3]

示例 2:
输入:
nums1 = [6, 7]
nums2 = [6, 0, 4]
k = 5
输出:
[6, 7, 6, 0, 4]

示例 3:
输入:
nums1 = [3, 9]
nums2 = [8, 9]
k = 3
输出:
[9, 8, 9]

输入说明 :
首先输入数组长度m,然后输入m个0-9的数字,各数字以空格分隔。
然后输入数组长度n,然后输入n个0-9的数字,各数字以空格分隔。
最后输入整数k。

输出说明 :
输出结果数组,输出内容无空格。

输入范例 :
2
6 7
3
6 0 4
5

输出范例 :
67604

#include<iostream> 
#include<vector>
using namespace std;

vector<int> largestK(vector<int> &nums,int count){//在数组nums中选取count个最大值 
	vector<int> t;
	int popRemain=nums.size()-count;
	for(int i=0;i<nums.size();i++){
		while(popRemain&&!t.empty()&&nums[i]>t.back()){
			popRemain--;
			t.pop_back();
		}
		t.push_back(nums[i]);
	
	}
	vector<int> res(t.begin(),t.begin()+count);
	return res; 
}

vector<int> merge(vector<int> &nums1,vector<int> &nums2){//将两个数组整合起来  使其最大 
	int i=0,j=0;
	vector<int> res;
	while(i<nums1.size()&&j<nums2.size()) {
        if (nums1>nums2) {
            res.push_back(nums1[i]);
            nums1.erase(nums1.begin());
        }else {
            res.push_back(nums2[j]);
            nums2.erase(nums2.begin());
            }
    }
	while(i<nums1.size()){
		res.push_back(nums1[i]);
		i++;
	}
	while(j<nums2.size()){
		res.push_back(nums2[j]);
		j++;
	}
	return res;
} 

vector<int> maxNumber(vector<int> &nums1,vector<int> &nums2,int k){
	vector<int> res;
	for(int i=0;i<=k;i++){
		if(i>nums1.size()){
			break;
		}
		if(k-i<=nums2.size()){
			vector<int> temp1=largestK(nums1,i);//在nums1中选择i个 
			vector<int> temp2=largestK(nums2,k-i);//在nums2中选择k-i个 
			vector<int> temp=merge(temp1,temp2);
			if(res<temp){
				res=temp;
			}
		} 
	}
	return res;
}


int main(){
    int m,n,k,data;
    vector<int> nums1,nums2;
    cin>>m;
    for(int i=0; i<m; i++){
        cin>>data;
        nums1.push_back(data);
    }
    cin>>n;
    for(int i=0; i<n; i++){
        cin>>data;
        nums2.push_back(data);
    }
    cin>>k;
    vector<int> res=maxNumber(nums1,nums2,k);
    for(int i=0; i<res.size(); i++)
        cout<<res[i];
    return 0;
}
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值