Primitive Root

watersail / 2024-11-05 / 原文

  • P为质数的条件是为了迎合题目的原根背景,不一定是解题的突破口,要把握题目的整体感受
  • 把区间拆分的想法一跃而过,为什么不多想想呢?是因为之前两道题都是拆分多个区间吗?
  • 使用区间差分的思想简化除法运算
  • 如果要用lowbit运算拆分区间,则输出的区间倒序排列,对应的是树状数组的查询过程
点击查看代码
#include <bits/stdc++.h>
using namespace std;
long long p,m,ans;
long long lowbit(long long n)
{
	return n&(-n);
}
long long calc(long long m)
{
	if(m==0)
	{
		return 0;
	}
	return (m-1)/p+1;
}
long long calc(long long l,long long r)
{
	long long tmp=lowbit(r)-1;
	tmp=~tmp;
	long long f=(p-1)&tmp;
	l=l^f;
	return calc(l+lowbit(r)-1)-calc(l)+(l%p==1);
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T;
	cin>>T;
	while(T--)
	{
		cin>>p>>m;
		ans=((m^(p-1))%p==1);
		while(m)
		{
			ans+=calc(m-lowbit(m),m);
			m-=lowbit(m);
		}
		cout<<ans<<"\n";
	}
	return 0;
}