[机房练习赛4.6] comb 卢卡斯定理

Problem 3. comb

Input file: comb.in
Output file: comb.out
Time limit: 1 second
Memory limit: 256 MB
一天，Mr. Hu 对组合数产生了兴趣，他想要知道满足下面条件的数i 有多少个：
gcd(
(
n
i
)
; p) = p (0 i n)
其中p 是素数。
Input
第1 行，2 个整数：n p。
Output
输出所求。
Sample
comb.in
5 2
comb.out
1
Note
样例中，一共有5 个组合数：1; 5; 10; 5; 1，其中和2 公约数为2 的数只有一个10，故输出1。
• 对于30% 的数据，满足1 n; p 103
• 对于100% 的数据，满足1 n; p 1018
Page 3

`

#include<iostream>
#include<cstdio>
using namespace std;
int main() {
    freopen("comb.in","r",stdin); freopen("comb.out","w",stdout);
    long long n, p, ans1, ans2;
    scanf("%I64d%I64d", &n, &p);
    ans1 = n + 1;ans2 = 1;
    while( n ) {  ans2 *= n % p + 1; n /= p; }
    printf("%I64d", ans1-ans2-1 );
    return 0;
}