杭电ACM1005(矩阵乘法)

#include <iostream>
using namespace std;

int matrix[ 2 ][ 2 ];
int tMatrix[ 2 ][ 2 ];
int res[ 2 ][ 2 ];

int A, B, n;

void calc()
{
	while(n != 0)
	{
		if(n & 1 == 1)
		{
			tMatrix[ 0 ][ 0 ] = matrix[ 0 ][ 0 ] * res[ 0 ][ 0 ] + 
								matrix[ 0 ][ 1 ] * res[ 1 ][ 0 ];
			tMatrix[ 0 ][ 1 ] = matrix[ 0 ][ 0 ] * res[ 0 ][ 1 ] + 
								matrix[ 0 ][ 1 ] * res[ 1 ][ 1 ];
			tMatrix[ 1 ][ 0 ] = matrix[ 1 ][ 0 ] * res[ 0 ][ 0 ] + 
								matrix[ 1 ][ 1 ] * res[ 1 ][ 0 ];
			tMatrix[ 1 ][ 1 ] = matrix[ 1 ][ 0 ] * res[ 0 ][ 1 ] + 
								matrix[ 1 ][ 1 ] * res[ 1 ][ 1 ];
							
			res[ 0 ][ 0 ] = tMatrix[ 0 ][ 0 ] % 7;
			res[ 0 ][ 1 ] = tMatrix[ 0 ][ 1 ] % 7;
			res[ 1 ][ 0 ] = tMatrix[ 1 ][ 0 ] % 7;
			res[ 1 ][ 1 ] = tMatrix[ 1 ][ 1 ] % 7;
		}
		
		tMatrix[ 0 ][ 0 ] = matrix[ 0 ][ 0 ] * matrix[ 0 ][ 0 ] +
						    matrix[ 0 ][ 1 ] * matrix[ 1 ][ 0 ];
		tMatrix[ 0 ][ 1 ] = matrix[ 0 ][ 0 ] * matrix[ 0 ][ 1 ] +
						    matrix[ 0 ][ 1 ] * matrix[ 1 ][ 1 ];
		tMatrix[ 1 ][ 0 ] = matrix[ 1 ][ 0 ] * matrix[ 0 ][ 0 ] + 
						    matrix[ 1 ][ 1 ] * matrix[ 1 ][ 0 ];
		tMatrix[ 1 ][ 1 ] = matrix[ 1 ][ 0 ] * matrix[ 0 ][ 1 ] + 
						    matrix[ 1 ][ 1 ] * matrix[ 1 ][ 1 ];
		
		matrix[ 0 ][ 0 ] = tMatrix[ 0 ][ 0 ] % 7;
		matrix[ 0 ][ 1 ] = tMatrix[ 0 ][ 1 ] % 7;
		matrix[ 1 ][ 0 ] = tMatrix[ 1 ][ 0 ] % 7;
		matrix[ 1 ][ 1 ] = tMatrix[ 1 ][ 1 ] % 7;
		
		n >>= 1;		
	}	
}

int main()
{
	while(cin>>A>>B>>n)
	{
		if(A == 0 && B == 0 && n == 0)
		{
			break;
		}
		matrix[ 0 ][ 0 ] = A;
		matrix[ 0 ][ 1 ] = B;
		matrix[ 1 ][ 0 ] = 1;
		matrix[ 1 ][ 1 ] = 0;
		
		res[ 0 ][ 0 ] = 1;
		res[ 0 ][ 1 ] = 0;
		res[ 1 ][ 0 ] = 0;
		res[ 1 ][ 1 ] = 1;
		
		if(n != 1 && n != 2)
		{
			n = n - 2;
			calc();
		
			cout<<(1 * res[ 0 ][ 0 ] + 1 * res[ 0 ][ 1 ]) % 7<<endl;
		}
		else
		{
			if(n == 1 || n == 2)
			{
				cout<<1<<endl;
			}
		}
	}	
	return 0;
}