南昌网络赛 H The Nth Item

南昌网络赛The Nth Item 暴力快速幂+unordered_map记忆化

注意：记忆化不能写到快速幂求解函数里，不断调用函数会造成很大的时间浪费

#include<bits/stdc++.h>
using namespace std;
#define sc(x) scanf("%lld",&x);
#define si signed
#define int long long
#define P pair<int,int>
#define fi first
#define se second
#define read(A) for(int i=0;i<n;i++) scanf("%lld",&A[i]);
#define push_back pb
int A[][]={{,},{,}};
int B[][];
int C[][];
int D[][];
int Q,N;
#define mod 998244353
#define maxn 100000+10
void mut(int A[][],int B[][])
{
    C[][]=((A[][]*B[][])%mod+(A[][]*B[][])%mod)%mod;
    C[][]=((A[][]*B[][])%mod+(A[][]*B[][])%mod)%mod;
    C[][]=((A[][]*B[][])%mod+(A[][]*B[][])%mod)%mod;
    C[][]=((A[][]*B[][])%mod+(A[][]*B[][])%mod)%mod;
    for(int i=;i<=;i++){
        for(int j=;j<=;j++){
            A[i][j]=C[i][j];
        }
    }
}
unordered_map<int,int> mp;
int qp(int n)
{
    if(n==)return ;
    if(n==)return ;
    n-=;
    int Ans[][]={{,},{,}};
    for(int i=;i<;i++){
        for(int j=;j<;j++){
            B[i][j]=A[i][j];
        }
    }
    while(n){
        if(n&){
           mut(Ans,B);
        }
        mut(B,B);
        n>>=;
    }
    return Ans[][];
}

si main()
{
    scanf("%lld%lld",&Q,&N);
    int ans=;
    while(Q--){
        int a;
        if(mp[N%]){
         a=mp[N%];
        }
        else{
            mp[N%]=a=qp(N%);
        }
        ans^=a;
        N=((N^(a*a)));

    }
    cout<<ans<<'\n';
}