CodeForces 258 C.Little Elephant and LCM（数论+二分+组合数学）

Description

给出一个长度为$n$的序列$a_i$，要求找满足$lcm(b_1,b_2,...,b_n)=max(b_1,b_2,...,b_n),1\le b_i\le a_i$的$b$序列个数

Input

第一行一整数$n$表示序列长度，之后$n$个正整数$a_i(1\le n,a_i\le 10^5)$

Output

输出满足条件的$b$序列个数，结果模$10^9+7$

Sample Input

4
1 4 3 2

Sample Output

15

Solution

一个序列的最大公倍数等于其最大值说明这个序列所有元素均为最大值的因子，那么我们考虑枚举最大值为$m$，那么每个$b_i$必须是$m$的因子，其某些$b_i$必须等于$m$，假设$m$的因子从小到大拿出来是$c_1,c_2,...,c_d$，对于第$i$个数字，最多可以取到$a_i$，假设$c_k\le a_i<c_{k+1}$，那么从$1$~$a_i$中可以取到$k$个不同的$m$的因子，对$a$序列排序，每次二分搜出介于$[c_k,c_{k+1})$之间的$a_i$数量$cnt_k$，记大于等于$m$的$a_i$数量为$cnt$，那么对于$a_i<m$的位置，其取值方案就是$\prod\limits_{k=1}^{d-1}k^{cnt_k}$，而对于这$cnt$个不小于$m$的位置，必须要保证其中至少有一个位置取到$m$，故方案数为$m^{cnt}-(m-1)^{cnt}$，即最大值为$m$的合法序列有$(m^{cnt}-(m-1)^{cnt})\cdot\prod\limits_{k=1}^{d-1}k^{cnt_k}$，枚举$m$累加答案即可，时间复杂度$O(nlog^2n)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
#define mod 1000000007
vector<int>f[maxn];
void init(int n=1e5)
{
    for(int i=1;i<=n;i++)
        for(int j=i;j<=n;j+=i)
            f[j].push_back(i);
}
int Pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=(ll)ans*a%mod;
        a=(ll)a*a%mod;
        b>>=1; 
    }
    return ans;
}
int n,a[maxn];
int main()
{
    init();
    while(~scanf("%d",&n))
    {
        int m=0;
        for(int i=0;i<n;i++)scanf("%d",&a[i]),m=max(m,a[i]);
        sort(a,a+n);
        int ans=0;
        for(int i=1;i<=m;i++)
        {
            int res=1,size=f[i].size();
            for(int j=0;j<size-1;j++)
            {
                int t1=f[i][j],t2=f[i][j+1];
                int num=lower_bound(a,a+n,t2)-lower_bound(a,a+n,t1);
                res=(ll)res*Pow(j+1,num)%mod;
            }
            int num=lower_bound(a,a+n,m+1)-lower_bound(a,a+n,f[i][size-1]);
            res=(ll)res*(Pow(size,num)-Pow(size-1,num)+mod)%mod;
            ans+=res;
            if(ans>=mod)ans-=mod;
        }
        printf("%d\n",ans);
    }
    return 0;
}