51Nod 1014 X^2 Mod P 二分查找+余数的性质

X*X mod P = A，其中P为质数。给出P和A，求<=P的所有X。

收起

输入

两个数P A，中间用空格隔开。(1 <= A < P <= 1000000, P为质数)

输出

输出符合条件的X，且0 <= X <= P，如果有多个，按照升序排列，中间用空格隔开。
如果没有符合条件的X，输出：No Solution

输入样例

13 3

输出样例

4 9

 思路：

此题我先求出0-1000000的平方，储存到数组中。

因为X^2%p=a;

所以等式可化为y%p=a+p*z%p;

因此只需要验证y是否可以开方即可。

这就需要调用二分查找来实现验证。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
const int maxn=1000000;
typedef long long ll;
ll is[maxn+5];
ll p,a;
vector<int> v;
void Pow ()
{
    for (ll i=0;i<=maxn;i++)
    {
        is[i]=i*i;
    }
}
void Find(ll target)
{
    int l=0,r=maxn;
    while (l<=r)
    {
        int mid=(l+r)>>1;
        if(is[mid]==target)
        {
            v.push_back(mid);
            break;
        }
        else if(is[mid]<target)
        {
            l=mid+1;
        }
        else
        {
            r=mid-1;
        }
    }
}
int main()
{
    Pow();
    scanf("%lld%lld",&p,&a);
    for (ll i=a;i<=p*p;i+=p)
    {
         Find(i);
    }
    if(v.size()==0)
    {
        printf("No Solution\n");
    }
    else
    {
        for (int i=0;i<v.size();i++)
        {
            printf("%d%c",v[i],i==v.size()-1? '\n':' ');
        }
    }
    return 0;
}