题链: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;
}
}