HDU 4506 小明系列故事——师兄帮帮忙

Problem Description

小明自从告别了ACM/ICPC之后，就开始潜心研究数学问题了，一则可以为接下来的考研做准备，再者可以借此机会帮助一些同学，尤其是漂亮的师妹。这不，班里唯一的女生又拿一道数学题来请教小明，小明当然很高兴的就接受了。不过等他仔细读题以后，发现自己也不会做，这下小明囧了：如果回复说自己不懂，岂不是很没面子？
　　所以，他现在私下求你帮忙解决这道题目，题目是这样的：
　　给你n个数字，分别是a₁,a₂,a₃,a₄,a₅……a_n，这些数字每过一个单位时间就会改变，假设上一个单位时间的数字为a₁’,a₂’,a₃’……a_n’，那么这个单位时间的数字a[i] = a[i - 1]’ * K(i == 1的时候a[1] = a[n]’ * K)，其中K为给定的系数。
　　现在的问题就是求第t单位时间的时候这n个数字变成了什么了？由于数字可能会很大，所以只要你输出数字对10⁹ + 7取余以后的结果。

Input

输入数据第一行是一个正整数T，表示有T组测试数据；
　　每组数据有两行，第一行包含输入三个整数n, t, k，其中n代表数字个数，t代表第t个单位时间，k代表系数；第二行输入n个数字ai，代表每个数字开始的时候是多少。

[Technical Specification]
　　T <= 100
　　1 <= n <= 10⁴
　　0 <= t <= 10⁹　　其中 t = 0 表示初始状态
　　1 <= k <= 10⁹
　　1 <= ai<= 10⁹

Output

对于每组数据请输出第t单位时间后这n个数字变成了什么，输出的时候每两个数字之间输出一个空格，行末不要输出多余的空格，具体见样例。

Sample Input

2
3 2 5
1 2 3
3 0 5
1 2 3

Sample Output

50 75 25
1 2 3

题解

题意：

• 共T组数据，给定n，t，k，a₀,……，a_n-1，每次将数组a的每个数乘以k并赋给后一个数，最后一个给第一个，输出操作t次后的数组a。（可以理解为最后 a_i*k^t %(10⁹+7) 放在数组的某个位置。）

数学思想：

• 由于所给数据而可能非常大，幂的次数很大，所以要用到快速幂。快速幂之前学习过。另外，对于 a_i*k^t %(10⁹+7) 最终应放的位置，也需要用到取模。 其位置为 (i + t) % n，由于 t 的范围很大，可以这样计算 (i + t%n) % n。

思路：

• 读入数据存入数组a，先用快速幂算法计算k^t %(10⁹+7)的结果记为sum，找到数据最终位置，将计算结果存入数组b中，因为原值被覆盖，所以计算结果不能存入数组a。输出b的前n个值。

Code

#include<iostream>
#define ll long long
#define m 1000000007
using namespace std;


ll pow_1(ll a, ll b, ll p)//计算a^b%p
{
	ll ans = 1;
	while(b)
	{
		if(b & 1)	ans = ans * a % p;	
		a = a * a % p;
		b >>= 1;
	}
	return ans;
}

ll a[100009], b[100009];

int main()
{
    ll T, n, t, k, i, sum;
    cin >> T;
    for(;T > 0; T--)
    {
        cin >> n >> t >> k;
        for(i = 0;i < n; i++)
            cin >> a[i];
        sum = pow_1(k, t, m);
        for(i = 0; i < n; i++)
             b[(i + t % n) % n] = ((a[i] % m) * (sum % m)) % m;
        for(i = 0; i < n - 1; i++)
            printf("%lld ",b[i]);
        printf("%lld\n", b[n - 1]);
     
    }
    return 0;
}