zoukankan      html  css  js  c++  java
  • [SDOI2017]序列计数

    首先将问题转换成一个容斥的形式,把质数的限制去掉。

    显然有一个(Theta(NMP))的dp转移对吧。

    我们不难发现每一次(dp)的转移都是一样的,所以直接矩阵快速幂优化即可。

    /*
      mail: mleautomaton@foxmail.com
      author: MLEAutoMaton
      This Code is made by MLEAutoMaton
    */
    #include<stdio.h>
    #include<stdlib.h>
    #include<string.h>
    #include<math.h>
    #include<algorithm>
    #include<queue>
    #include<set>
    #include<map>
    #include<iostream>
    using namespace std;
    #define ll long long
    #define re register
    #define file(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout)
    inline int gi(){
    	int f=1,sum=0;char ch=getchar();
    	while(ch>'9' || ch<'0'){if(ch=='-')f=-1;ch=getchar();}
    	while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
    	return f*sum;
    }
    const int N=20000010,P=110,Mod=20170408;
    int n,m,p;
    struct matrix{
    	int a[P][P];
    	void clear(){memset(a,0,sizeof(a));}
    	int*operator[](int x){return a[x];}
    	matrix operator*(matrix b){
    		matrix c;
    		for(int i=0;i<p;i++)
    			for(int j=0;j<p;j++){
    				c[i][j]=0;
    				for(int k=0;k<p;k++)
    					c[i][j]=(c[i][j]+1ll*a[i][k]*b[k][j]%Mod)%Mod;
    			}
    		return c;
    	}
    }S,T;
    int ha[N];
    bool is_prime[N];
    int prime[5000010],s;
    void init(){
    	is_prime[1]=1;
    	for(int i=2;i<=m;i++){
    		if(!is_prime[i])prime[++s]=i;
    		for(int j=1;j<=s && i*prime[j]<=m;j++){
    			is_prime[i*prime[j]]=1;
    			if(i%prime[j]==0)break;
    		}
    	}
    }
    int main(){
    #ifndef ONLINE_JUDGE
    	freopen("in.in","r",stdin);
    #endif
    	n=gi();m=gi();p=gi();init();
    	for(int i=1;i<=m;i++)ha[i%p]++;
    	for(int i=0;i<p;i++)S[0][i]=ha[i]%Mod;
    	for(int i=0;i<p;i++)
    		for(int j=0;j<p;j++)
    			T[i][(i+j)%p]=(T[i][(i+j)%p]+ha[j])%Mod;
    	int b=n-1;
    	while(b){
    		if(b&1)S=S*T;
    		T=T*T;b>>=1;
    	}
    	int ans=S[0][0];
    	memset(ha,0,sizeof(ha));T.clear();S.clear();
    	for(int i=1;i<=m;i++)if(is_prime[i])ha[i%p]++;
    	for(int i=0;i<=min(m,p-1);i++)S[0][i]=ha[i]%Mod;
    	for(int i=0;i<p;i++)
    		for(int j=0;j<p;j++)
    			T[i][(i+j)%p]=(T[i][(i+j)%p]+ha[j])%Mod;
    	b=n-1;
    	while(b){
    		if(b&1)S=S*T;
    		T=T*T;b>>=1;
    	}
    	printf("%d
    ",(ans-S[0][0]+Mod)%Mod);
    	return 0;
    }
    
  • 相关阅读:
    实体枚举字段注释反向生成数据库注释sql
    系统间数据存储和交互思路
    复选框与bitmap算法实践
    Entity Framework Core配置DbContext的两种方式
    C#语法糖——持续更新
    抽丝剥茧读源码——Microsoft.Extensions.Configuration(2)
    抽丝剥茧读源码——Microsoft.Extensions.Configuration(1)
    算法分享之关于atcoderbeginner166E的讲解
    关于coder168E问题的分析与解答(C语言)
    atcoder168D题
  • 原文地址:https://www.cnblogs.com/mleautomaton/p/11418663.html
Copyright © 2011-2022 走看看