剑指 Offer 49. 丑数
所有丑数
现在假设有3个数组,分别是:
A:{1* 2,2* 2,3* 2 ,4* 2,5* 2,6* 2,8 * 2,10 * 2…}
B:{1* 3,2* 3,3* 3,4* 3,5* 3,6* 3,8* 3,10* 3…}
C:{1* 5,2* 5,3* 5,4* 5,5* 5,6* 5,8* 5,10* 5…}
那么所有丑数的排列,必定就是上面ABC3个数组的合并结果然后去重得到的,那么这不就转换成了三个有序数组的无重复元素合并的问题了吗?
合并有序数组的一个比较好的方法,就是每个数组都对应一个指针,然后比较这些指针所指的数中哪个最小,就将这个数放到结果数组中,然后该指针向后挪一位。
假设三个数组的指针分别是i,j,k,此时均是指向第一个元素,然后比较A[i],B[j]和C[k],得到的最小的数A[i],就是ugly[1],此时ugly就变成{1,2}了,对应的ABC数组就分别变成了:
A : {1* 2,2* 2…}
B : {1* 3, 2* 3…}
C :{1* 5, 2* 5…}
此时根据合并有序数组的原理,A数组指针i就指向了下一个元素,即’2*2’,而j和k依然分别指向B[0]和C[0],然后进行下一轮合并,就是A[1]和B[0]和C[0]比较,最小值作为ugly[2]…如此循环n次,就可以得到ugly[n]了。
class Solution {
public:
int nthUglyNumber(int n) {
int a=0,b=0,c=0;
int dp[n];
dp[0]=1;
for(int i=1;i<n;++i){
int tmp1=dp[a]*2,tmp2=dp[b]*3,tmp3=dp[c]*5;
dp[i]=min(tmp1,min(tmp2,tmp3));
if(dp[i]==tmp1) ++a;
if(dp[i]==tmp2) ++b;
if(dp[i]==tmp3) ++c;
}
return dp[n-1];
}
};