SDUT2165：Crack Mathmen（快速幂）

题目：http://acm.sdut.edu.cn/sdutoj/showproblem.php?pid=2165&cid=1431

快速幂。

#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <queue>
#include <vector>
#include <algorithm>
#include <queue>
#define inf 0x3f3f3f3f
typedef long long ll;
using namespace std;
int tt,n,cn[];
int qq[],we;
struct node
{
    int w,id;
}q[];
#define mod 997
int er(int lf,int rf,int key)
{
    int l=lf,mid;
    int r=rf;
    while(l<=r)
    {
        mid = l+(r-l)/;
        if(q[mid].w==key)
            return mid;
        else if(q[mid].w>key)
        {
            r=mid-;
        }
        else
        {
            l=mid+;
        }
    }
    return -;
}
int cmp(const void *a,const void *b)
{
    struct node *aa=(struct node *)a;
    struct node *bb=(struct node *)b;
    return aa->w-bb->w;
}
int mult_mod(int a,int b)//计算 (a*b)%c.
{
    a%=mod;//                           利用二分思想减少相乘的时间
    b%=mod;
    ll ret=;
    while(b)
    {
        if(b&)
        {
            ret+=a;
            ret%=mod;
        }
        a<<=;
        if(a>=mod) a%=mod;
        b>>=;
    }
    return ret;
}
int pow_mod(int x,int n)//x^n%n
{
    if(n==) return x%mod;
    x%=mod;
    int tmp=x;
    int ret=;
    while(n)
    {
        if(n&) ret=mult_mod(ret,tmp);
        tmp=mult_mod(tmp,tmp);
        n>>=;
    }
    return ret;
}
void init()
{
    memset(cn,,sizeof(cn));
    tt=;
    for(int i=;i<=;i++)
    {
        q[tt].id=i;
        q[tt++].w=pow_mod(i,n);
    }
    for(int i=;i<=;i++)
    {
         q[tt].id=i;
        q[tt++].w=pow_mod(i,n);
    }
    for(int i=;i<=;i++)
    {
        q[tt].id=i;
        q[tt++].w=pow_mod(i,n);
    }
    qsort(q,tt,sizeof(q[]),cmp);
    cn[]=;
    for(int i=;i<tt;i++)
    {
        if(q[i].w==q[i-].w)
        {
            cn[i]=;
            cn[i-]=;
        }
        else cn[i]=;
    }
}
char s[];
int main()
{
    int T,da;
    bool FF;
    scanf("%d",&T);
    for(int z=;z<=T;z++)
    {
        FF=true;

        scanf("%d",&n);
        init();
        we=;
        scanf("%s",s);
        int l=strlen(s);
        for(int i=;i<l;i=i+)
        {
            da=s[i]-'';
            for(int j=i+;j<=i+;j++)
            {
                da=da*+(s[j]-'');
            }
            int t=er(,tt-,da);
            if(t==-||cn[t]>)
            {
                printf("No Solution\n");

                FF=false;
                break;
            }
            else
            {
                qq[we++]=q[t].id;
            }
        }
        if(FF)
        {
            for(int i=;i<we;i++)
                printf("%c",qq[i]);
            printf("\n");

        }
        // printf("\n");
    }
    return ;
}