洛谷3702,递推+矩阵乘法优化

传送门
这题一开始想出来了一个二元一次递推式,有点麻烦。
一看题解,发现可以转为求全集和答案的补集,挺妙的。
一开始筛素数的数组只开到 106 ,RE两个点,开到 5106 就AC了,感觉很玄学,毕竟值域 m2107

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mo=20170408;
struct matrix{
    ll a[110][110];
    int n;
    void clear(){
        memset(a,0,sizeof a);
    }
    matrix operator*(const matrix&x){
        matrix ans;
        ans.clear();
        for(int i=0;i<n;++i)
            for(int j=0;j<n;++j)
                for(int k=0;k<n;++k)ans.a[i][k]=(ans.a[i][k]+a[i][j]*x.a[j][k])%mo;
        ans.n=n;
        return ans;
    }
}a,b,c,d;
matrix pow(matrix a,int b){
    if(b==1)return a;
    matrix u=pow(a,b>>1);
    if(b&1)return u*u*a;
        else return u*u;
}
ll n,m,p,x[110],y[110],pp[5000010],xb,i,j,z,ans;
bool bb[20000010];
int main(){
    cin>>n>>m>>p;
    y[0]=m/p;
    for(i=1;i<=m%p;++i)y[i]=m/p+1;
    for(;i<p;++i)y[i]=m/p;
    for(i=2;i<=m;++i){
        if(!bb[i])pp[++xb]=i,++x[i%p];
        for(j=1;j<=xb;++j){
            z=pp[j]*i;
            if(z>m)break;
            bb[z]=1;
            if(i%pp[j]==0)break;
        }
    }
    if(n==1)return printf("%d\n",!bb[p] && p<=m),0;
    d.n=b.n=p;
    for(i=0;i<p;++i)
        for(j=0;j<p;++j){
            z=(j-i+p)%p;
            b.a[i][j]=y[z]-x[z];
            d.a[i][j]=y[z];
        }
    a=pow(b,n-1);
    c=pow(d,n-1);
    for(j=0;j<p;++j)
        ans=(ans+mo+y[j]*c.a[j][0]%mo-(y[j]-x[j])*a.a[j][0]%mo)%mo;
    printf("%lld\n",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值