AtCoder Beginner Contest 161 F - Division or Substraction (数论)

题目传送门

题意:给定n,求[2,n]范围内有多少个数能通过题目两种运算(n=n-k,n/=k)(n>=k时)使得n最后结果为1.

思路:

AC代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main(){
	ll n,ans=2;//初始值包括n和n-1两种. 
	cin>>n;
	if(n==2) ans=1;//特判 
	for(ll k=2,i=n;k*k<=n;k++,i=n){ 
		while(i>=k) i%k?i%=k:i/=k;//这里是计算[2,sqrt[n]的情况. 
		if(i==1) ans++;
		if((n-1)%k==0&&k*k!=(n-1)) ans++;//这里是计算 [sqrt(n),n]n-1的因数情况,这里k*k!=(n-1)是去重 
	}
	cout<<ans<<endl;
	return 0;
}