感觉随机抽的题目从第二周以来越来越简单了。。。。以下为题目地址:
https://leetcode.com/problems/nth-magical-number/description/
在我看来,这题其实是一道数学题。它说的是,给出两个2到40000的正整数,能被这两个数中任意一个数整除的数叫做魔数。 然后给定一个数字N(小于1000000000),求第N个魔数是多少。
这题很简单,我们可以很容易看出来,这些数的排列是以最小公倍数为周期的。首先求出两个数的最小公倍数,这里先求最大公约数,然后用两个数的积除于最大公约数,就得到了最小公倍数。w然后计算在一个最小公倍数的周期内有多少魔数K。然后计算N里面包含了多少个最小公倍数周期(N/K),得出的结果乘以最小公倍数,就可以得到N包含的完整的最小公倍数周期到达了多少。接着计算不完整的那个周期里第(f=N%K)个数,并和上面得到的答案加起来。
这里计算第f个数用了一些技巧。就是第f个数一定大于等于第floor(f*B/(A+B))个被A整除的数大,也大于等于第floor(f*A/(A+B))个被B整除的数。这是因为,f前面至少有floor(f*A/(A+B))个被B整除的数与floor(f*B/(A+B))个被A整除的数。因为f*B*A/(A+B)与f*A*B/(A+B)相等,所以我们可以这样划分。于是我们可以看floor(f*B/(A+B))+floor(f*A/(A+B))是不是小于f,如果等于,则取这两个数中比较大的。否则,取比这两个大的数中的下一个数。
下面给出程序代码:
#include <iostream>
#include <cmath>
#include <algorithm>
#define modn 1000000007
using namespace std;
class Solution {
public:
int nthMagicalNumber(int N, int A, int B) {
int gcdab = gcd(A,B);
int maxn = (A / gcdab) * B;
int num = A / gcdab + B / gcdab - 1;
long long f = (long long)N / num * maxn;
int ans = f % modn;
long long f2 = N % num;
double d1 = (f2*B)/(A+B);
double d2 = (f2*A)/(A+B);
int c1 = floor(d1);
int c2 = floor(d2);
int temp = 0;
if(c1+c2<f2)
temp = min((c1+1)*A,(c2+1)*B);
else
temp = max(c1*A,c2*B);
ans+=temp;
ans = ans % modn;
return ans;
}
int gcd(int x, int y)
{
if(!x||!y) return x>y?x:y;
for(int t;t=x%y;x=y,y=t);
return y;
}
};
这道题其实主要的难点在于容易溢出。其实最容易的方法,是把所有变量用long long存,然后所有步骤都模1000000007。但是我的代码里面只在可能会溢出的地方做了处理。
下面给出运行结果: