bzoj 3328 PYXFIB —— 单位根反演

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3328

单位根反演，主要用到了 \( [k|n] = \frac{1}{k} \sum\limits_{i=0}{k-1} w_{k}^{in} \)推导见：https://www.cnblogs.com/galaxies/p/bzoj3328.html构造 \( F(x) \) 是为了凑成二项式定理的形式，其实也不难想；又忘记在定义构造函数时写 N 了！把 =N(...) 写成 =(...) 不仅能编译过，还能过样例...以后写构造函数先检查这个！

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
ll rd()
{
  ll ret=,f=; char ch=getchar();
  while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return f?ret:-ret;
}
int K,mod,pt[],cnt;
ll n;
struct N{
  int a,b,c,d;
  N(int a=,int b=,int c=,int d=):a(a),b(b),c(c),d(d) {}
  void init(){a=; b=; c=; d=;}
  N operator * (const N &y) const
  {
    N ret;
    ret.a=((ll)a*y.a+(ll)b*y.c)%mod;
    ret.b=((ll)a*y.b+(ll)b*y.d)%mod;
    ret.c=((ll)c*y.a+(ll)d*y.c)%mod;
    ret.d=((ll)c*y.b+(ll)d*y.d)%mod;
    return ret;
  }
};
ll pw(ll a,ll b){ll ret=; for(;b;b>>=,a=a*a%mod)if(b&)ret=ret*a%mod; return ret;}//ll b
int upt(int x){while(x>=mod)x-=mod; while(x<)x+=mod; return x;}
int G(int p)
{
  int x=p-; cnt=;
  for(int i=;i*i<=x;i++)
    {
      if(x%i)continue;
      pt[++cnt]=i; if(i*i!=x)pt[++cnt]=x/i;
    }
  for(int a=;a<p;a++)
    {
      bool fl=;
      for(int i=;i<=cnt;i++)
    if(pw(a,pt[i])==){fl=; break;}
      if(fl)return a;
    }
}
N pww(N a,ll b)//ll b
{
  N ret; ret.init();
  for(;b;b>>=,a=a*a)if(b&)ret=ret*a;
  return ret;
}
int F(int x)
{
  N A=N(x+,,,x); A=pww(A,n);//N()
  x=pw(x,mod-); x=pw(x,n);
  return (ll)x*A.a%mod;
}
int main()
{
  int T=rd();
  while(T--)
    {
      n=rd(); K=rd(); mod=rd();
      int w=pw(G(mod),(mod-)/K),inv=pw(w,mod-);
      int ans=;
      for(int i=,t=;i<K;i++,t=(ll)t*inv%mod)ans=upt(ans+F(t));
      ans=(ll)ans*pw(K,mod-)%mod;
      printf("%d\n",upt(ans));
    }
  return ;
}