hdu6395 Sequence(矩阵快速幂+数论分块)

题目

思路来源

https://blog.csdn.net/Originum/article/details/82111066

题解

对于每个P/n向下取整相同值的地方用矩阵快速幂，

[l,r]的值都为v==P/i，那么求r=P/(P/i)，r+1就得到下一个l，是为数论分块

g1,g2为last Fn-1,Fn-2的值，f1,f2为now Fn-1,Fn-2的值，类似滚动dp的存值

end=p/i?min(p/(p/i),n):n;

p/i作分母，所以分0和非0讨论，

由于[l,r]可能包含n，即r可能大于n，所以p/i非0时取min(p/(p/i),n)

p/i==0时，说明p<i，这样[i,n]的值都有p<i即p/i==0，分成一块即可

代码

#include <stdio.h>
#include <memory.h>
#define min(a,b)    (((a) < (b)) ? (a) : (b))
using namespace std;
typedef long long ll;
const int MOD = 1e9+7;
const int MAXN = 3;
 
struct mat {
    ll c[MAXN][MAXN];
    int m, n;
    mat(int a, int b) : m(a), n(b) {
        memset(c, 0, sizeof(c));
    }
    void clear(){
		memset(c, 0, sizeof(c)); 
    }
    mat operator * (const mat& temp) {
        mat ans(m, temp.n);
        for (int i = 0; i < m; i ++)
            for (int j = 0; j < temp.n; j ++)
                for (int k = 0; k < n; k ++)
                    ans.c[i][j] = (ans.c[i][j] + c[i][k] * temp.c[k][j] % MOD) % MOD;
        return ans;
    }
};
 
mat mat_pow(mat M, int n) {
    mat ans(M.m, M.m);
    for (int i = 0; i < M.m; i ++)
        ans.c[i][i] = 1; 
    while (n > 0) {
        if (n & 1) ans = ans * M;
        M = M * M;
        n >>= 1;
    }
    return ans;
}
int T,a,b,c,d,p,n;
int f1,f2,g1,g2;
int end;
mat m(3,3);
int main ()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d%d%d%d",&a,&b,&c,&d,&p,&n);
		if(n==1){printf("%d\n",a);continue;}
		else if(n==2){printf("%d\n",b);continue;}
		g1=b;g2=a;//第一行 第二行 
		for(int i=3;i<=n;++i)
		{
			m.clear();
			m.c[0][0]=d;m.c[0][1]=c;m.c[0][2]=p/i;
			m.c[1][0]=1;m.c[2][2]=1;
			end=p/i?min(p/(p/i),n):n;
			m=mat_pow(m,end-i+1);
			f1=(1ll*m.c[0][0]*g1%MOD+1ll*m.c[0][1]*g2%MOD+m.c[0][2]%MOD)%MOD;
			f2=(1ll*m.c[1][0]*g1%MOD+1ll*m.c[1][1]*g2%MOD+m.c[1][2]%MOD)%MOD;
			g1=f1;g2=f2;i=end;
		}
		printf("%d\n",f1);
	}
	return 0;
}