E. Modular Stability
题意:给你 n 个数,你要从中选 k 个来组成 a 数组,使得非负数 x 对数组 a 中的所有元素以任意的顺序取模后的结果不变,即:
问这样的数组一共能构造多少个?
思路:能看出来数组的关键就在于最小的那一项,如果最小的那一项是其余所有元素的因子的话,那该数组就能被构造,例:
n = 10 , k = 4 ;
a[4] = { 2 , 6 , 8 , 10 }
设最小项为 m , 那么数组中元素就是 m ,2m,3m,4m,……。不论 x 是多少,它都必定被分解为 y * m + c;如此一来,不论取模的顺序如何,最后的结果都必定会等于 c
然后这道题就变成了组合数的板子题啦~
代码附上:
#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
using namespace std;
using ll = long long ;
const ll mod =998244353 ;
const ll maxN = 5e5 + 10;
ll fact[maxN], invfact[maxN], inv[maxN];
ll mult(ll a, ll b)
{
return (a * b) % mod;
}
ll com(ll n,ll k)
{
return mult(fact[n], mult(invfact[k], invfact[n - k]));
}
void init()
{
fact[0] = invfact[0] = fact[1] = invfact[1] = inv[1] = 1;
for (ll i = 2; i < maxN; i++)
{
fact[i] = mult(fact[i - 1], i);
inv[i] = mult(inv[mod % i], mod - mod / i);
invfact[i] = mult(invfact[i - 1], inv[i]);
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
init();
ll n,k,ans=0;
cin>>n>>k;
for(ll i=1; i<=n/k; ++i)
ans=(ans+com(n/i-1,k-1))%mod;
cout<<ans;
return 0;
}