sdut 2605 A^X mod P (分解优化)

题意：

题目意思很简单，就是求（A^f[1]+A^f[2]+。。。+A^f[n]）%P

题解：

一开始直接扫描一遍结果无情TL，用快速幂计算幂值有很多重复的计算，因此想办法将结果保存在数组里面，dp的思想。显然f[i]=fix*k+j，这样分解是对的，那么选取一个合适的fix，这样数组可以存下需要的解。不妨令fix=31623

，那么A^(fix*k+j)%m,就可以分解成(A^k)^fix*A^j，用dpk[i]保存(A^k)^i,dpj[i]保存A^i,那么

A^f[i]=dpk[i/fix]*dpj[i%fix];

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define B(x) (1<<(x))
using namespace std;
typedef long long ll;
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
void cmax(ll& a,ll b){ if(b>a)a=b; }
void cmin(ll& a,ll b){ if(b<a)a=b; }
void add(int& a,int b,int mod){ a=(a+b)%mod; }
void add(ll& a,ll b,ll mod){ a=(a+b)%mod; }
const int oo=0x3f3f3f3f;
const int MOD=2012;
const int maxn=31623;
const int fix=maxn;
int T,A,K,a,b,m,P,n;
ll dpk[maxn+1],dpj[maxn+1];

void Init(){
    dpk[0]=dpj[0]=1;
    for(int i=1;i<=maxn;i++){
        dpj[i]=(dpj[i-1]*A)%P;
    }
    dpk[1]=dpj[maxn];
    for(int i=2;i<=maxn;i++){
        dpk[i]=(dpk[i-1]*dpk[1])%P;
    }
}

int main(){
    //freopen("E:\\read.txt","r",stdin);
    scanf("%d",&T);
    ll ans,f;
    for(int cas=1;cas<=T;cas++){
        scanf("%d%d%d%d%d%d%d",&n,&A,&K,&a,&b,&m,&P);
        Init();
        f=K;
        ans=0;
        for(int i=1;i<=n;i++){
            ans=(ans+dpk[f/fix]*dpj[f%fix])%P;
            f=(a*f+b)%m;
        }
        printf("Case #%d: %lld\n",cas,ans);
    }
    return 0;
}