洛谷——P4296 [AHOI2007]密码箱

P4296 [AHOI2007]密码箱

密码x大于等于0，且小于n，而x的平方除以n，得到的余数为1。

求这个密码，$1<=n<=2,000,000,000$

暴力枚举,数据有点儿水$O(nlogn)$，显然过不了$n<=10^9$

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>

#define LL long long
using namespace std;

LL n;
bool flg=false;

int main()
{
    scanf("%lld",&n);

    for(int i=;;i++){
        LL x=i*n+,p=sqrt(x);
        if(p>n) break;
        if(p*p==x){
            flg=true;
            printf("%lld\n",p);
        }
    }
    if(!flg) return puts("None"),;
    return ;
} 

正解：

题目要求$x^x\mod n =1$，也就是求$n|(x^2-1^2)$，即$n|(x-1)\times(x+1)$，既然如此，考虑将$n$拆开，就有$n=a\times b$ $a|(x+1),b|(x-1)$或是$b|(x+1),a|(x-1)$

不妨设$a<b$，枚举$a$，同时枚举两种情况，优先队列+$STL$——$map$判重

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
#include<queue>
#include<vector>

#define LL long long
using namespace std;

LL n;
bool flg=false;

priority_queue<LL,vector<LL>,greater<LL> >Q;
map<LL,bool>M;

int main()
{
    scanf("%lld",&n);
    int p=sqrt(n);
    for(int i=;i<=p;i++){
        if(n%i==){
            int b=n/i;
            for(int x=;x<=n;x+=b)//b|x-1
                if((x+)%i==&&!M[x]) Q.push(x),M[x]=;
            for(int x=b-;x<=n;x+=b)//b|x+1
                if((x-)%i==&&!M[x]) Q.push(x),M[x]=;
        }
    }
    while(!Q.empty()){
        printf("%lld\n",Q.top());
        Q.pop();
    }

    return ;
} 

或是$set$去重

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
#include<queue>
#include<vector>
#include<set>

#define LL long long
using namespace std;

LL n;
bool flg=false;

set<LL>S;

int main()
{
    scanf("%lld",&n);
    int p=sqrt(n);
    for(int i=;i<=p;i++){
        if(n%i==){
            int b=n/i;
            for(int x=;x<=n;x+=b)//b|x-1
                if((x+)%i==)
                    S.insert(x);
            for(int x=b-;x<=n;x+=b)//b|x+1
                if((x-)%i==)
                    S.insert(x);
        }
    }
    while(!S.empty()){
        printf("%lld\n",*S.begin());
        S.erase(S.begin());
    }

    return ;
} 

貌似没有$None$的情况，即数据里不存在$n=1$