leetcode刷题记录。321.拼接最大数
思路
具体算法:
从 nums1 中 取 min(i, len(nums1))min(i,len(nums1)) 个数形成新的数组 A(取的逻辑同第一题),其中 i 等于 0,1,2, … k。
从 nums2 中 对应取 min(j, len(nums2))min(j,len(nums2)) 个数形成新的数组 B(取的逻辑同第一题),其中 j 等于 k - i。
注意我这里取数的上限,即不能超过数组本身的长度,这点容易大家忽略
将 A 和 B 按照上面的 merge 方法合并
上面我们暴力了 k 种组合情况,我们只需要将 k 种情况取出最大值即可。
代码
int* removeKdigits(int* num,int numsize,int k)
{
int* stk=(int*)malloc(sizeof(int)*(numsize+1));
int top=0;
for(int i=0;i<numsize;i++)
{
while(top>0&&num[i]>stk[top]&&k)
{
k--;
top--;
}
stk[++top]=num[i];
}
top-=k;
int* ret=(int*)malloc(sizeof(int)*(top));
for(int i=1;i<=top;i++)
{
ret[i-1]=stk[i];
}
return ret;
}
int* maxNumber(int* nums1, int nums1Size, int* nums2, int nums2Size, int k, int* returnSize)
{
bool isfirst=true;
int* temp=(int*)malloc(sizeof(int)*(k));
int* ret=(int*)malloc(sizeof(int)*(k));
for(int i=0;i<=k;i++)
{
int n1num=i>nums1Size? nums1Size:i;
int n2num=k-n1num>nums2Size? nums2Size:k-n1num;
if(n1num+n2num<k)
{
continue;
}
int k1=nums1Size-n1num;
int k2=nums2Size-n2num;
int* str1=removeKdigits(nums1,nums1Size,k1);
int* str2=removeKdigits(nums2,nums2Size,k2);
int x=0,y=0,z=0;
while(x<n1num&&y<n2num)
{
if(str1[x]>str2[y])
{
temp[z++]=str1[x++];
}
else if(str1[x]<str2[y])
{
temp[z++]=str2[y++];
}
else
{
int xtemp=x+1,ytemp=y+1;
while(xtemp<n1num&&ytemp<n2num&&str1[xtemp]==str2[ytemp])
{
xtemp++;
ytemp++;
}
if(xtemp==n1num&&ytemp<n2num)
{
temp[z++]=str2[y++];
}
else if(xtemp<n1num&&ytemp==n2num)
{
temp[z++]=str1[x++];
}
else if(xtemp==n1num&&ytemp==n2num)
{
temp[z++]=str2[y++];
}
else
{
if(str1[xtemp]>str2[ytemp])
{
temp[z++]=str1[x++];
}
else
{
temp[z++]=str2[y++];
}
}
}
}
while(x<n1num)
{
temp[z++]=str1[x++];
}
while(y<n2num)
{
temp[z++]=str2[y++];
}
if(isfirst)
{
for(int i=0;i<k;i++)
{
ret[i]=temp[i];
}
isfirst=false;
}
else
{
for(int i=0;i<k;i++)
{
if(ret[i]==temp[i])
{
continue;
}
else if(ret[i]>temp[i])
{
break;
}
else if(ret[i]<temp[i])
{
for(int j=0;j<k;j++)
{
ret[j]=temp[j];
}
break;
}
}
}
}
*returnSize=k;
return ret;
}