2016湖南湘潭邀请赛 - 2019牛客国庆集训派对day6

2016湖南湘潭邀请赛 - 2019牛客国庆集训派对day6

A - 2016

题意：求一个2阶矩阵的n次幂，对7取模的结果
思路：可以用上面介绍的10进制倍增的方法。也可以用 n 对2016取模，还可以记忆化找循环节。

代码1：记忆化找循环节

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int N=5e5;

string n;
int a,b,c,d;
int m[N+10][2][2];

int main()
{
	cin>>n>>a>>b>>c>>d;
    m[1][0][0]=a,m[1][0][1]=b;
	m[1][1][0]=c,m[1][1][1]=d;

	int mod=0;
	for(int i=2;i<=N;++i)
	{
		mod++;
		m[i][0][0]=(m[i-1][0][0]*a+m[i-1][0][1]*c)%7;
		m[i][0][1]=(m[i-1][0][0]*b+m[i-1][0][1]*d)%7;
		m[i][1][0]=(m[i-1][1][0]*a+m[i-1][1][1]*c)%7;
		m[i][1][1]=(m[i-1][1][0]*b+m[i-1][1][1]*d)%7;
		if(m[i][0][0]==a&&m[i][0][1]==b&&m[i][1][0]==c&&m[i][1][1]==d)
			break;
	} 
    m[0][0][0]=m[mod][0][0];
    m[0][0][1]=m[mod][0][1];
    m[0][1][0]=m[mod][1][0];
    m[0][1][1]=m[mod][1][1];
    
	int n1=0;
	int len=n.size();
	for(int i=0;i<=len-1;++i)
		n1=(n1*10+n[i]-'0')%mod;
	
	printf("%d %d\n%d %d\n",m[n1%mod][0][0],m[n1%mod][0][1],m[n1%mod][1][0],m[n1%mod][1][1]);
	return 0;
}

代码2:10进制倍增

#include <iostream>
#include <algorithm>
#include <cstdio>
#define ll long long 
using namespace std;
const int mod=7;

struct Matrix
{
	static const int N=2;
	ll a[N+1][N+1];
	Matrix()
	{
		clear();
	}
	void init()
	{
		for(int i=1;i<=N;++i)
			a[i][i]=1;
	}
	void clear()
	{
		for(int i=1;i<=N;++i)
			for(int j=1;j<=N;++j)
				a[i][j]=0;
	}
	
	Matrix operator*(const Matrix &b) const
	{
		Matrix ret;
		for(int i=1;i<=N;++i)
			for(int j=1;j<=N;++j)
				for(int k=1;k<=N;++k)
					ret.a[i][j]=(ret.a[i][j]+a[i][k]*b.a[k][j])%mod;
		return ret;
	}	
};

string n;
int a,b,c,d;

int main()
{
	cin>>n>>a>>b>>c>>d; 
	int len=n.size();
	Matrix A[4];
	A[0].a[1][1]=a,A[0].a[1][2]=b;
	A[0].a[2][1]=c,A[0].a[2][2]=d;
	
    Matrix ret;
    ret.init();	
	for(int i=len-1;i>=0;--i)
	{
		A[1].clear(),A[2].clear(),A[3].clear();
		for(int j=1;j<=3;++j)
			A[j]=A[j-1]*A[j-1];
		n[i]-='0';
		for(int j=0;j<4;++j)
		{
			if((n[i]>>j)&1)
				ret=ret*A[j];
		}
		A[0]=A[1]*A[3];	
	}
	cout<<ret.a[1][1]<<" "<<ret.a[1][2]<<"\n";
	cout<<ret.a[2][1]<<" "<<ret.a[2][2]<<"\n";	
	return 0;
}