代码恢复 2024.6.5。
很简单。
直接找循环节就行了。
代码：
点击查看代码
/*
Tips:
你数组开小了吗？
你MLE了吗？
你觉得是贪心，是不是该想想dp？
一个小时没调出来，是不是该考虑换题？
打 cf 不要用 umap！！！
记住，rating 是身外之物。
该冲正解时冲正解！
Problem:
算法：
思路：
*/
#include<bits/stdc++.h>
using namespace std;
//#define map unordered_map
#define forl(i,a,b) for(register long long i=a;i<=b;i++)
#define forr(i,a,b) for(register long long i=a;i>=b;i--)
#define forll(i,a,b,c) for(register long long i=a;i<=b;i+=c)
#define forrr(i,a,b,c) for(register long long i=a;i>=b;i-=c)
#define lc(x) x<<1
#define rc(x) x<<1|1
#define mid ((l+r)>>1)
#define cin(x) scanf("%lld",&x)
#define cout(x) printf("%lld",x)
#define lowbit(x) (x&-x)
#define pb push_back
#define pf push_front
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define endl '\n'
#define QwQ return 0;
#define ll long long
#define ull unsigned long long
#define lcm(x,y) x/__gcd(x,y)*y
#define Sum(x,y) 1ll*(x+y)*(y-x+1)/2
#define aty cout<<"Yes\n";
#define atn cout<<"No\n";
#define cfy cout<<"YES\n";
#define cfn cout<<"NO\n";
#define xxy cout<<"yes\n";
#define xxn cout<<"no\n";
#define printcf(x) x?cout<<"YES\n":cout<<"NO\n";
#define printat(x) x?cout<<"Yes\n":cout<<"No\n";
#define printxx(x) x?cout<<"yes\n":cout<<"no\n";
ll t;
ll n,m,mod;
ll a[100010];
ll k,vis[100010],ans,sum[100010],L;
ll f(ll x,ll y){
return x%y;
}
void solve()
{
cin>>n>>m>>mod;
if(n==1)
{
cout<<m<<endl;
return ;
}
ll num=m;
while(!vis[num])
{
k++;
sum[k]=sum[k-1]+num;
vis[num]=1;
a[num]=k;
num=f(num*num,mod);
}
cout<<(sum[k]-sum[a[num]-1])*((n-a[num]+1)/(k-a[num]+1)) +
sum[(n-a[num]+1)%(k-a[num]+1)+a[num]-1]<<endl;
/* a[++k]=m;
sum[k]=a[k];
vis[a[k]]=1;
forl(i,2,n)
{
a[++k]=f(a[k-1]*a[k-1],mod);
sum[k]=sum[k-1]+a[k];
if(vis[a[k]])
{
L=vis[a[k]];
k--;
break;
}
vis[a[k]]=i;
}
if(!L)
cout<<sum[k]<<endl;
else
{
cout<<(n-k)%(n-L+1)+k<<' '<<L-1<<endl;
cout<<sum[k]+((n-k)/(n-L+1))*(sum[k]-sum[L-1])+sum[(n-k)%(n-L+1)+n-L+1]-sum[L-1]<<endl;
}*/
}
int main()
{
IOS;
t=1;
// cin>>t;
while(t--)
solve();
/******************/
/*while(L<q[i].l) */
/* del(a[L++]);*/
/*while(L>q[i].l) */
/* add(a[--L]);*/
/*while(R<q[i].r) */
/* add(a[++R]);*/
/*while(R>q[i].r) */
/* del(a[R--]);*/
/******************/
QwQ;
}