HDU 1588 构造矩阵求和两种方法

最新推荐文章于 2021-04-20 10:56:04 发布

lansetiankong_yiyi

最新推荐文章于 2021-04-20 10:56:04 发布

阅读量2.3k

点赞数

分类专栏：构造矩阵乘法加法文章标签： function input output numbers matrix each

本文链接：https://blog.csdn.net/lansetiankong_yiyi/article/details/5828239

版权

构造矩阵乘法同时被 2 个专栏收录

7 篇文章 0 订阅

订阅专栏

加法

1 篇文章 0 订阅

订阅专栏

Gauss Fibonacci

Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 609 Accepted Submission(s): 277

Problem Description

Without expecting, Angel replied quickly.She says: "I'v heard that you'r a very clever boy. So if you wanna me be your GF, you should solve the problem called GF~. "
How good an opportunity that Gardon can not give up! The "Problem GF" told by Angel is actually "Gauss Fibonacci".
As we know ,Gauss is the famous mathematician who worked out the sum from 1 to 100 very quickly, and Fibonacci is the crazy man who invented some numbers.

Arithmetic progression:
g(i)=k*i+b;
We assume k and b are both non-nagetive integers.

Fibonacci Numbers:
f(0)=0
f(1)=1
f(n)=f(n-1)+f(n-2) (n>=2)

The Gauss Fibonacci problem is described as follows:
Given k,b,n ,calculate the sum of every f(g(i)) for 0<=i<n
The answer may be very large, so you should divide this answer by M and just output the remainder instead.

Input

The input contains serveral lines. For each line there are four non-nagetive integers: k,b,n,M
Each of them will not exceed 1,000,000,000.

Output

For each line input, out the value described above.

Sample Input

Sample Output

  
  
   
   21
12
  
  
  
  
   
   /*******************************************************************************************
	首先我们看g=k*i+b; 是一个等差数列

	如果能推出f(g)这个函数也是一个等差或者等比数列，就可以得出一个公式

	f(k*i+b)  0<=i<n

	建立一个二分矩阵A=[1 1,1 0]
	
	f(b)=A^b  

	则:

	f(b)=A^b

	f(k+b)=A^k+b

	f(2*k+b)=A^2*k+b
	.
	.
	.
	f((n-1)*k+b)=A^(n-1)*k+b

	我们就可以得出一个等比数列:

	首项：A^b
	
	公比：A^k

	项数：n

	(res是单位矩阵)

	运用等比数列求和公式得出：sum=A^b*(res+A^k+(A^k)^2+(A^k)^3+...+(A^k)^(n-1))

	需要注意的一点是：当b=0的情况
***************************************************************************************/

/***************************************************************************************

例如:A+A^2+A^3+A^4+A^5+A^6=(A+A^2+A^3)+A^3*(A+A^2+A^3)  用递归写

***************************************************************************************/

#include<stdio.h>
struct node
{
	int num[3][3];
};
int m;
node res,tp;
node mut(node a,node b)  //矩阵乘法函数
{
	node c;
	int i,j,k;
	for(i=1;i<3;i++)
		for(j=1;j<3;j++)
		{
			c.num[i][j]=0;
			for(k=1;k<3;k++)
				c.num[i][j]+=(a.num[i][k]*b.num[k][j])%m;
			c.num[i][j]%=m;
		}
		return c; 
}
node abc(node a,int temp) //运用的快速幂的函数
{
	tp=a;
	a=res;
	while(temp)
	{
		if(temp&1)
			a=mut(a,tp);
		tp=mut(tp,tp);
		temp>>=1;
	}
	return a;
}
node add(node a,node b) //矩阵加法函数
{
	node c;                                                                                                                                                                   

                                                                                                            
	int i,j;
	for(i=1;i<3;i++)
		for(j=1;j<3;j++)
			c.num[i][j]=(a.num[i][j]+b.num[i][j])%m;
		return c;
}
node function(int k,node p) //用二分法求矩阵和,递归实现
{
	if(k==1)
		return p;
	else if(k&1)
		return add(function(k-1,p),abc(p,k));
	else
		return mut(function(k>>1,p),add(abc(p,k>>1),res)); 
}

int main()
{
	int k,n,b;
	node a,ak,ab,ans;
	a.num[1][1]=0;a.num[1][2]=1;
	a.num[2][1]=1;a.num[2][2]=1;
	res.num[1][1]=1;res.num[2][2]=1;
	res.num[1][2]=0;res.num[2][1]=0;
	while(scanf("%d%d%d%d",&k,&b,&n,&m)!=EOF)
	{
	ak=abc(a,k); //保存a的k次方
	ab=abc(a,b); //保存a的b次方
	ans=add(res,function(n-1,ak)); // 通过这一步可以算出A的K次方+A的2K次方+3K次方+。。。+(n-1)K次方
	ans=mut(ab,ans);               // 最后再加一个单位阵res ,然后这一步再与A的b次方相乘就得到答案了
	printf("%d/n",ans.num[2][1]);
	}
	return 0;
} 
  
  
  
  
   
   第二种方法比较巧妙，构造了一个矩阵直接求和
  
  
  
  
  
  
   
   首先，我们将问题整理一下，就是对等差数列 ai=k*i+b，求所有的f(ai)之和除以M的余数

当0<=i<N

大家有没有想到，因为ai是等差数列，倘若f(ai)也是个等什么序列，那说不定就有公式求了

f(ai)显然不是等差数列，直接看上去也不是等比数列

但是如果把f(ai)换成我们刚才所说的Fibonacci矩阵呢？

是的，可是我们对矩阵是直接求幂即可，由于矩阵加法的性质，我们要求A^ai的右上角元素之和，只要求A^ai之和的右上角元素

就矩阵这个东西来说，完全可以看作一个等比数列，
首项是：A^b
公比是：A^k
项数是：N

呵呵，我们可以把问题进一步简化

因为矩阵的加法对乘法也符合分配律，我们提出一个A^b来，形成这样的式子：
A^b*( I + A^k + (A^k)^2 + .... + (A^k)^(N-1) )

A^b 和 A^k 显然都可以用我们之前说过的方法计算出来，这剩下一部分累加怎么解决呢

简单起见，设A^k=B
要求 G(N)=I + ... + B^(N-1)，设i=N/2
若N为偶数，G(N)=G(i)+G(i)*B^i  意思就是假设N=6 那么G(N)=(I+B1+B2)*(I+B3)=I+B1+B2+B3+B4+B5  B1指B的一次方， I是单位阵
若N为奇数，G(N)=I+ G(i)*B + G(i) * (B^(i+1)) G(N)=I+(I+B1+B2)*(B1+B4)=I+B1+B2+B3+B4+B5+B6

呵呵，这个方法就是比赛当时ACRush用的
而农夫用的则是更精妙的方法，大家可想知道

我们来设置这样一个矩阵
B I
O I
其中O是零矩阵，I是单位矩阵

将它乘方，得到
B^2 I+B
O   I
乘三方，得到
B^3 I+B+B^2
O   I
乘四方，得到
B^4 I+B+B^2+B^3
O   I

既然已经转换成矩阵的幂了，继续用我们的二分或者二进制法，直接求出幂就可以了


斐波那契数列f(n)=f(n-1)+f(n-2),f(0)=1,f(1)=1

     | 1  1 |  mat[0][0]相当于f(2)    mat[0][1]相当于f(1)  

mat= | 1  0 |  mat[1][0]相当于f(1)    mat[1][1]相当于f(0)


   | 1       1|        | f(2)    f(1)|

mat= |1        0|   =    | f(1)    f(0)|

     

                | 1       1| ^b          | f(b+1)    f(b)|

         mat^b =|1       0 |          =  | f(b)    f(b-1)|




f(b) = matrix[0][1];





#include<stdio.h>
struct node
{
	long long num[5][5];  //注意这里要用long long 或者__int64 
};
int m;
node res,tp;   // res为单位阵 

node mut(node a,node b,int d)  //矩阵乘法
{
	node c;
	int i,j,k;
	for(i=1;i<=d;i++)
		for(j=1;j<=d;j++)
		{
			c.num[i][j]=0;
			for(k=1;k<=d;k++)
				c.num[i][j]+=(a.num[i][k]*b.num[k][j])%m;
			c.num[i][j]%=m;
		}
		return c;
}
	
node abc(node a,int temp,int d)
{
	if(temp==0)
		return res;  // 这里要千万注意啊，如果传进来的b等于0是返回单位阵，因为a^b=a^0=单位阵，晕HDOJ真郁闷，我错的答案也能AC
 	tp=a;
	a=res;
	while(temp)
	{
		if(temp&1)
			a=mut(a,tp,d);
		tp=mut(tp,tp,d);
		temp>>=1;
	}
	return a;
}

int main()
{
	int i,j;
	int k,b,n;
	node a;
	a.num[1][1]=0;a.num[1][2]=1;
	a.num[2][1]=1;a.num[2][2]=1;
	for(i=1;i<=4;i++)
		for(j=1;j<=4;j++)
			if(i==j)
				res.num[i][j]=1;
			else
				res.num[i][j]=0;
	while(scanf("%d%d%d%d",&k,&b,&n,&m)!=EOF)
	{
		node ak,ab;
		ak=abc(a,k,2);
		ab=abc(a,b,2);
		node ans;
		for(i=1;i<=2;i++)
			for(j=1;j<=2;j++)
				ans.num[i][j]=ak.num[i][j];
			for(i=3;i<=4;i++)
				for(j=1;j<=2;j++)
					ans.num[i][j]=0;
				ans.num[1][3]=ans.num[2][4]=1;
				ans.num[1][4]=ans.num[2][3]=0;
				ans.num[3][3]=ans.num[4][4]=1;
				ans.num[3][4]=ans.num[4][3]=0;
			ans=abc(ans,n,4);
			node cn;
			cn.num[1][1]=ans.num[1][3];  // 提取右上角的4位数
			cn.num[1][2]=ans.num[1][4];
			cn.num[2][1]=ans.num[2][3];
			cn.num[2][2]=ans.num[2][4];
			cn=mut(ab,cn,2);            // 再乘上ab就是答案了
			printf("%d/n",cn.num[2][1]); // 最后取num[2][1]
	}
	return 0;
}

lansetiankong_yiyi

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
2
评论
HDU 1588 构造矩阵求和两种方法

Gauss FibonacciTime Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 609 Accepted Submission(s): 277Problem DescriptionWithout expecting, Angel replied quickly.She says: "I'v heard that you'r a very clev
复制链接

扫一扫