sum
Time Limit: 2000/1000ms (Java/Others)
Problem Description:
给定a和n,计算a+aa+aaa+aaaa+...+a...a(n个a) 的和。
Input:
测试数据有多组,以文件结尾。每行输入a,n(1<=a<=10^9,1=<n<=10^18)。
Output:
由于结果可能比较大,所以请输出答案mod 1000000007。
Sample Input:
3 2
Sample Output:
36
Source:
GOJ Monthly Traning,Dec 2014
ps: 这题困扰了一段时间,今天终于把它做出来了。之前虽然也找出了递推关系,但却没找到合适的递推矩阵(矩阵递推的过程出现了小数), 今天重新做了一下这道题,递推矩阵终于有了。
数组递推关系:
(其中,a₁ = a, bit_a 表示a的位数,比如
a =9,则bit_a = 1,a = 88, 则bit_a = 2。)
由数组的递推关系,我们可以得到求和的递推公式:
(其中f(0) = 0)。
接下来,我们将递推矩阵找出来,用快速幂解决就可以了。递推矩阵构造需要一定的技巧,不能出现小数,否则可能会因为精度问题而答案错误。
如下是我构造的递推矩阵:
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<sstream>
#define ll long long
#define mod 1000000007
using namespace std;
void mulit(ll matrix1[3][3], ll matrix2[3][3])
{
ll temp[3][3] = {0};
for(int i = 0; i < 3; i++)
for(int k = 0; k < 3; k++)
for(int j = 0; j < 3; j++)
{
temp[i][j] = (temp[i][j] % mod + (((matrix1[i][k] % mod) * (matrix2[k][j] % mod)) % mod) % mod) % mod;
}
for(int i = 0; i < 3; i++)
for(int j = 0; j < 3; j++)
matrix1[i][j] = temp[i][j];
}
ll res[3][3];
void pow(ll matrix[][3], ll n)
{
memset(res, 0, sizeof res);
for(int i = 0; i < 3; i++)res[i][i] = 1;
while(n > 0)
{
if(n & 1)
{
mulit(res, matrix);
}
mulit(matrix, matrix);
n = n >> 1;
}
}
ll ans[3][1];
void mulit1(ll matrix1[3][3], ll matrix2[3][1])
{
ll temp[3][1] = {0};
for(int i = 0; i < 3; i++)
for(int k = 0; k < 3; k++)
for(int j = 0; j < 1; j++)
temp[i][j] = (temp[i][j] % mod + (((matrix1[i][k] % mod) * (matrix2[k][j] % mod)) % mod) % mod) % mod;
for(int i = 0; i < 3; i++)
for(int j = 0; j < 1; j++)
ans[i][j] = temp[i][j];
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
ll a, n;
while(cin>>a>>n)
{
stringstream ss;
ss<<a;
string bit_a;
ss>>bit_a;
int bit = bit_a.size();
ll fron = 1;
for(int i = 0; i < bit; i++) fron *= 10;
ll f1[3][1] = {a, 2*a, a};
ll f[3][3] = {fron, 1, 0,
0, 2, -1,
0, 1, 0};
pow(f, n - 1);
mulit1(res, f1);
cout<<ans[0][0]<<endl;
}
return 0;
}