LeetCode 每日一题 321. 拼接最大数(dp或单调栈)

题链:https://leetcode-cn.com/problems/create-maximum-number/

我的思路(超时):力扣有的题没有数据范围,这是很烦的,但是我dp弱,记录一下吧。

dp[i][j][k]代表第一个数组的前i个数第二个数组的前j个数拼接成k个数的序列的最大值。

转移方程:dp[i][j][k]=max{ dp[i-1][j-1][k] ,dp[i-1][j][k-1]+a[i-1] , dp[i][j-1][k-1]+b[j-1] } (PS:i-1,j-1是因为我下标从1开始。)

C++代码:用char数组存储答案。T在第87个测试点。

#define ll long long
char dp[100][200][200][200];
char s[200],ss[200];
class Solution {
public:
    vector<int> maxNumber(vector<int>& a, vector<int>& b, int k) {
        vector<int> ans;
        int n=a.size(),m=b.size();
        if(n>m) swap(a,b),swap(n,m);
        for(int i=0;i<=n;i++)
            for(int j=0;j<=m;j++)
                for(int kk=0;kk<=k;kk++)
                    dp[i][j][kk][0]='\0';
        ss[0]='\0';
        for(int i=0;i<=n;i++)
            for(int j=0;j<=m;j++)
                for(int kk=1;kk<=k;kk++){
                    if(i>0){
                        if(strcmp(dp[i-1][j][kk],dp[i][j][kk])>0)
                            strcpy(dp[i][j][kk],dp[i-1][j][kk]);
                        strcpy(s,dp[i-1][j][kk-1]);  
                        s[kk-1]=('0'+a[i-1]);
                        s[kk]='\0';
                        if(strcmp(s,dp[i][j][kk])>0)
                            strcpy(dp[i][j][kk],s);
                    } 
                    if(j>0){
                        if(strcmp(dp[i][j-1][kk],dp[i][j][kk])>0)
                            strcpy(dp[i][j][kk],dp[i][j-1][kk]);
                        strcpy(s,dp[i][j-1][kk-1]); 
                        s[kk-1]=('0'+b[j-1]);
                        s[kk]='\0';
                        if(strcmp(s,dp[i][j][kk])>0)
                            strcpy(dp[i][j][kk],s);
                    }
                    if(i>0&&j>0){
                        if(strcmp(dp[i-1][j-1][kk],dp[i][j][kk])>0)
                            strcpy(dp[i][j][kk],dp[i-1][j-1][kk]);
                    }
                    if(kk==k){
                        if(strcmp(dp[i][j][k],ss)>0)
                            strcpy(ss,dp[i][j][k]);
                    } 
                }
        for(int i=0;i<k;i++)
            ans.push_back(ss[i]-'0');
        return ans;
    }

};

Java代码:用大数存储答案。也是T在第87个测试点。

import java.math.BigInteger;
class Solution {
    public int[] maxNumber(int[] a, int[] b, int k) {
        int n=a.length,m=b.length;
    	BigInteger[][][] dp= new BigInteger[n+1][m+1][k+10];
    	BigInteger[]aa= new BigInteger[n];
     	BigInteger[]bb= new BigInteger[m];
        for(int i=0;i<n;i++)
        	aa[i]=BigInteger.valueOf(a[i]);
        for(int i=0;i<m;i++)
        	bb[i]=BigInteger.valueOf(b[i]);
    	int [] ans = new int [k];
        for(int i=0;i<=n;i++)
            for(int j=0;j<=m;j++)
                for(int kk=0;kk<=k;kk++)
                    dp[i][j][kk]=BigInteger.ZERO;
        BigInteger s=BigInteger.ZERO,ss=BigInteger.ZERO,ten = new BigInteger("10");
        for(int i=0;i<=n;i++)
            for(int j=0;j<=m;j++)
                for(int kk=1;kk<=k;kk++){
                    if(i>0){
                        dp[i][j][kk]=dp[i-1][j][kk];
                        s=dp[i-1][j][kk-1];
                        s=s.multiply(ten).add(aa[i-1]);
                        if(dp[i][j][kk].compareTo(s)<0)
                            dp[i][j][kk]=s;
                    } 
                    if(j>0){
                        if(i==0)
                            dp[i][j][kk]=dp[i][j-1][kk];
                        else if(dp[i][j][kk].compareTo(dp[i][j-1][kk])<0) 
                            dp[i][j][kk]=dp[i][j-1][kk];
                        s=dp[i][j-1][kk-1];
                        s=s.multiply(ten).add(bb[j-1]);
                        if(dp[i][j][kk].compareTo(s)<0)
                            dp[i][j][kk]=s;
                    }
                    if(i>0&&j>0 && dp[i][j][kk].compareTo(dp[i-1][j-1][kk])<0 )
                        dp[i][j][kk]=dp[i-1][j-1][kk];
                    if(kk==k && ss.compareTo(dp[i][j][k])<0 ) ss=dp[i][j][k];
                }
        String sss = ss.toString();
        for(int i=0;i<k;i++)
        	ans[i]=(sss.charAt(i)-'0');
    	return ans;
    }
}

官方题解:菜了菜了,真没想到用单调栈。解法为,从第一个数组中选x个数(长度为x的最大子序列),从第二个数组中选k-x个数(长度为k-x的最大子序列 ,然后合并成一个最大的序列,求所有可能序列的最大序列。枚举x,用单调栈求对应的最大子序列。具体求法为,维护一个有k个元素单调递减的栈。每次要从单调栈中弹出元素时,要考虑栈里的数和剩下的数是否够k个?大于等于k个则可以弹出,否则不能弹放入栈时,栈内的数小于k个就放;否则就不放(单调递减的栈,现在栈顶一定大于等于这个元素,除非不够k个,否则没必要放)。然后考虑合并。假设我们现在该合并第一个最大子序列的第i个数和第一个最大子序列的第j个数。如果两个数不相等,我们肯定要先放较大的那一个;如果相等,我们要看看它们后面的数(i+1,j+1)直到两个数不同或者有一个序列走到头。如果有一个序列走到头,我们要先放没走到头的那一个。比如,9,8,6,5,7和6,5如果先放走到头的就是9866557,而如果先放没走到头的就是9866575。(因为我们要考虑没走到头的序列的下一个数是否有可能更大。)

class Solution {
public:
    void print(vector<int>& a){
        for(int i=0;i<a.size();i++)
            cout<<a[i]<<",";
        cout<<endl;
    }
    vector<int> getmax(vector<int>& a,int n,int len){
        vector<int> st(len,0);
        int top=-1;
        for(int i=0;i<n;i++){
            //top+1:栈内的个数
            //n-1-i:剩下的数
            while(top!=-1 && st[top]<a[i] && top+1+n-1-i>=len )
                top--;
            if(top<len-1) st[++top]=a[i];//不够k个才放
        }
        return st;
    }
    bool cmp(vector<int>& a,int n,int i,vector<int>& b,int m,int j){
        while(i<n&&j<m){
            //不相等放较大的
            if(a[i]>b[j]) return 1;
            else if(a[i]<b[j]) return 0; 
            //相等看后面的数
            i++;
            j++; 
        }
        //有一个走到头,则放没走到头的
        return i<n;
    }
    vector<int> merge(vector<int>& a, int n,vector<int>& b, int m){
        vector<int> c(n+m,0);
        int i=0,j=0,k=0;
        while(i<n&&j<m){
            if(cmp(a,n,i,b,m,j))
                c[k++]=a[i++];
            else 
                c[k++]=b[j++];
        }                  
        while(i<n) c[k++]=a[i++];
        while(j<m) c[k++]=b[j++];
        return c;
    }
    vector<int> maxNumber(vector<int>& a, vector<int>& b, int k) {
        int n = a.size() , m = b.size();
        int nn = min(n,k),y;
        vector<int> ans(k,0);
        for(int x=0;x<=nn;x++){
            y=k-x;
            if(y<0||y>m) continue;
            vector<int> v1= getmax(a,n,x);
            vector<int> v2= getmax(b,m,y);
            vector<int> c=merge(v1,x,v2,y);
            if(c>ans) ans=c;
        }
        return ans;
    }
};

真dp思路:太菜了太菜了,两个序列分别dp,这样就不会T了,然后再按上面说的合并即可。

dp1[i][j]代表第一个数组的前i个数拼接成j个数的序列的最大值。

dp2[i][j]代表第二个数组的前i个数拼接成j个数的序列的最大值。

转移方程:dp[i][j]=max{ dp[i-1][j] ,dp[i-1][j-1]+a[i-1] } (PS:i-1,j-1是因为我下标从1开始。)

c++会T,java可以过。

import java.math.BigInteger;

public class Solution {
    public int[] maxNumber(int[] a, int[] b, int k) {
        int n=a.length,m=b.length;
    	BigInteger[][] dp1= new BigInteger[n+1][k+1];
    	BigInteger[][] dp2= new BigInteger[m+1][k+1];
    
    	BigInteger[]aa= new BigInteger[n];
     	BigInteger[]bb= new BigInteger[m];
        for(int i=0;i<n;i++)
        	aa[i]=BigInteger.valueOf(a[i]);
        for(int i=0;i<m;i++)
        	bb[i]=BigInteger.valueOf(b[i]);
        for(int i=0;i<=k;i++)
            dp1[0][i]=dp2[0][i]=BigInteger.ZERO;
        BigInteger ten = new BigInteger("10"),s=BigInteger.ZERO,ss=BigInteger.ZERO;
        for(int i=1;i<=n;i++){
            dp1[i][0]=BigInteger.ZERO;
            for(int j=1;j<=k;j++){
                dp1[i][j]=dp1[i-1][j];
                s=dp1[i-1][j-1].multiply(ten).add(aa[i-1]);
                if(dp1[i][j].compareTo(s)<0)
                	dp1[i][j]=s;
            }
        }
        for(int i=1;i<=m;i++){
            dp2[i][0]=BigInteger.ZERO;
            for(int j=1;j<=k;j++){
                dp2[i][j]=dp2[i-1][j];
                s=dp2[i-1][j-1].multiply(ten).add(bb[i-1]);
                if(dp2[i][j].compareTo(s)<0)
                	dp2[i][j]=s;              
            }
        }
        int nn = Math.min(n,k),mm;
        BigInteger temp;
        for(int i=0;i<=nn;i++){
            mm=k-i;
            if(mm<0||mm>m) continue;
            temp = merge(dp1[n][i].toString(),i,dp2[m][mm].toString(),mm);
            if(ss.compareTo(temp)<0) ss=temp;
        }
        String sss = ss.toString();
        int [] ans = new int[k]; 
        for(int i=0;i<k;i++)
        	ans[i]=(sss.charAt(i)-'0');
    	return ans;
    }
    boolean cmp(String a,int i,int n,String b,int j,int m){
        while(i<n&&j<m){
            if(a.charAt(i)>b.charAt(j)) return true;
            else if(a.charAt(i)<b.charAt(j)) return false;
            i++;
            j++;
        }
        return (i<n);
    }
    BigInteger merge(String a,int n,String b,int m){
        BigInteger c=BigInteger.ZERO;
        BigInteger ten = new BigInteger("10");
        int i=0,j=0;
        while(i<n&&j<m){
            if(cmp(a,i,n,b,j,m))
                c=c.multiply(ten).add(BigInteger.valueOf(a.charAt(i++)-'0'));
            else c=c.multiply(ten).add(BigInteger.valueOf(b.charAt(j++)-'0'));
        }
        while(i<n) c=c.multiply(ten).add(BigInteger.valueOf(a.charAt(i++)-'0'));
        while(j<m) c=c.multiply(ten).add(BigInteger.valueOf(b.charAt(j++)-'0'));
        return c;
    }
}

 

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值