Jumps on a Circle

Jumps on a Circle

题意：题目给出一个有p个点的圆环，圆环上的点遵从0-p-1的顺序，一个人位于0的位置，每次移动距离是一个以1为首项，1为公差的等差数列，移动到p-1的下一个位置是0，问n次移动之后一共能经过多少个不同的点

思路：本题实际上就是问这个等差数列的前缀和数列%p之后有多少个不同的数，而前缀和公式为n*(n+10)/2，本题的n非常大，所以自然要找规律，我们猜测在某一移动之后会循环之前的走法，所以设n*(n+1)/2模上p之后为0，自然可以得到当n为2p时会回到原点，而此时移动距离为p+1，所以之后的所有移动距离%p之后等效于从1开始走，所有遍历至多到2*p-1就可以不用遍历了，（我们注意到移动距离一旦增加到p的倍数，走过的距离就等效于从1开始，而当p时起始距离不能回到原点，所以取2*p），注意此时O(N),N=2e7注意不要用stl，会超时

ac代码：

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
#define pb push_back
#define int long long
#define max(a, b) (((a) > (b)) ? (a) : (b))
#define min(a, b) (((a) < (b)) ? (a) : (b))
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef pair<int,int> pii;
const int N=1e7+10;
int a[N];
int n,p;
void solve()
{
	cin>>n>>p;
	int now=0;
	a[now]=1;
	for(int i=0;i<min(2*n,p);i++)
	{
		now+=i+1;
		a[now%n]=1;
	}
	int res=0;
	for(int i=0;i<n;i++)
	{
		if(a[i]==1)res++;
	}
	cout<<res<<endl;
}
signed main()
{
	Mirai;
	int T=1;
	//cin>>T;
	while(T--)
	{
		solve();
	}
}