2021-08-17大一暑假第三周周赛 E 快速幂

快速幂

题目链接

题目：
给 a,b ，每次 a,b 会变为 a+b,a−b ，问 k 次之后变成了哪两个数，对 998244353 取模，多组数据。

输入：
第一行一个正整数 T(1≤T≤100000) ，代表测试组数。

接下来 T 行每行三个数 a,b,k(0≤a,b<998244353,0≤k≤109) 。

输出：
T 行每行两个整数，代表每一组数据 a,b 最后变成了什么。

样例：
输入：
2
2 1 5
5 5 100
输出：
12 4
329904920 329904920
————————————————————————————————————————
题意：
每个样例给定两个数a,b；接下来有k次变化，每次变化都变成a+b,a-b；求k次变化后a,b的值。

思路：
直接暴力的话由于数据太大肯定会爆，所以找规律，发现：
a b k
a+b a-b 1
2a 2b 2
2a+2b 2a-2b 3
4a 4b 4
以此类推…
根据以上可以得出
当k为偶数时，变化后的结果为(2^ k/2)*a，( 2^k/2) b
当k为奇数时，变化后的结果为（2^ k/2) * （a+b），(2^ k /2)（a-b)
但是，如果两者相减得负数怎么办呢？
负数取余就等于（负数+mod）%mod即可

————————————————————————————————————————
代码：

#include<stdio.h>
#include<math.h>
#include<iostream>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<string.h>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
const int mod=998244353;
long long ksm(long long x)//快速幂求k次时是2的几次方
{
    long long s=1,n=2;
    while(x)
    {
        if(x%2==1)
            s=s*n%mod;
        n=n*n%mod;
        x/=2;
    }
    return s;
}
int main()
{
    int i,j,z,t;
    long long a,b,k,c;
    cin>>t;
    while(t--)
    {
        scanf("%lld%lld%lld",&a,&b,&k);
        long long s,s1,sum,sum1;
        s=((ksm(k/2)%mod)*a%mod)%mod;
        s1=((ksm(k/2)%mod)*b%mod)%mod;
        sum=(s%mod+s1%mod)%mod;
        sum1=((s-s1)%mod+mod)%mod;//注意，如果相减得负数，就加上余数在进行取余
        if(k%2)//奇数
            printf("%lld %lld\n",sum,sum1);
        else//偶数
            printf("%lld %lld\n",s,s1);
    }
    return 0;
}