您的位置:首页 > 其它

个人模板 快速幂取模,矩阵快速幂,快速乘

2017-03-27 17:54 357 查看
#include<bits/stdc++.h>
using namespace std;
const int MOD = 1007;
int xx(int a,int n,int MOD) //计算a的n次方对MOD取模
{
int ret=1;
int tmp=a%MOD;
while(n)
{
if(n&1)
ret=ret*tmp%MOD;
tmp=tmp*tmp%MOD;
n>>=1;
}
return ret;
}
int main()
{
int m,n;
while(scanf("%d%d",&m,&n)==2)
{
printf("%d\n",xx(m,n,MOD));
}
}
#include <bits/stdc++.h>
using namespace std;
const int N = 4 ;
int n;
struct xx
{
int a

;
} ori,res;
xx mul(xx x,xx y)
{
xx temp;
memset(temp.a,0,sizeof(temp.a));
for(int i=0; i<N; i++)
{
for(int j=0; j<N; j++)
{
for(int k=0; k<N; k++)
{
temp.a[i][j]+=x.a[i][k]*y.a[k][j];
}
}
}
return temp;
}
void init()
{
int x;
for(int i=0; i<N; i++)
{
for(int j=0; j<N; j++)
{
scanf("%d",&x);
ori.a[i][j]=x;
}
}
for(int i=0; i<N; i++)
{
for(int j=0; j<N; j++)
{
res.a[i][j]=i==j?1:0;
}
}
}
xx calc(long long k)
{
    while(k)
    {
        if(k%2==1)                      ///这么写比普通的快速幂好,亲身体会,HDU 5667用普通写法 TLE,这么写就过了
        {
            res=mul(ori,res);
            k-=1;
        }
        else
        {
            ori=mul(ori,ori);
            k/=2;
        }
    }
    return res.a[0][0];
}
int main()
{
while(cin>>n)
{
init();
calc(n);
}
}
long long mul(long long a,long long b)
{
long long res=0;
while(b)
{
if(b&1) res=(a+res)%Mod;
b>>=1;
a=(a+a)%Mod;
}
return res;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: