CodeForces 478 E.Wavy numbers（折半枚举+map）

Description
如果一个数除了最高位和最低位外的每一位都同时大于或小于相邻的两位，且如果只有两位数那么这两位不同的话，就称这个数是波浪数，问可以被n整除的第k个波浪数
Input
两个整数n和k(1<=n,k<=1e14)
Output
输出可以被n整除的第k个波浪数，如果结果超过1e14则输出-1
Sample Input
123 4
Sample Output
1845
Solution
折半枚举，先找出1e7中所有的波浪数，并用M[i][0/1][j]记录最高位是i且最高位小于/大于次高位且模n为j的波浪数的数目（为了省时间用unordered_map），如果1e7以内的波浪数中模n为0的数已经够k个则直接输出答案即可，如果不够就枚举高七位，首先高七位要是一个波浪数，而低七位需要满足连接处也是波浪且与高七位组成的数模n为0，根据这两个限制从M中找到合适的累加，如果枚举到某个高七位，累加的答案已经超过k，那么答案的高七位就是这七位，而低七位再次枚举即可
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<unordered_map>
using namespace std;
typedef long long ll;
unordered_map<ll,int>M[2][10];//M[0/1][i]表示以i开头最高位小于/大于次高位
int a[14];
const ll C=10000000;
int deal(int x)
{
    int res=0;
    while(x)a[res++]=x%10,x/=10;
    return res;
}
void Solve(ll n,ll k)
{
    for(int i=1;i<C;i++)
    {
        int res=deal(i),flag=1;
        for(int j=1;j<res-1;j++)
            flag&=((a[j]>a[j-1]&&a[j]>a[j+1])||(a[j]<a[j-1]&&a[j]<a[j+1]));
        if(res==2)flag&=(a[0]!=a[1]);
        if(!flag)continue;
        k-=(i%n==0);
        if(!k)
        {
            printf("%d\n",i);
            exit(0);
        }
        if(res==6&&a[res-1]>a[res-2])M[0][0][i%n]++;
        else if(res==7)M[a[res-1]>a[res-2]][a[res-1]][i%n]++;
    }
    for(int i=1;i<C;i++)
    {
        int res=deal(i),flag=1;
        for(int j=1;j<res-1;j++)
            flag&=((a[j]>a[j-1]&&a[j]>a[j+1])||(a[j]<a[j-1]&&a[j]<a[j+1]));
        if(res==2)flag&=(a[0]!=a[1]);
        if(!flag)continue;
        ll num=0,m=(n-C*i%n)%n;
        if(res==1)
        {
            for(int j=0;j<a[0];j++)
                if(M[0][j].find(m)!=M[0][j].end())
                    num+=M[0][j][m];
            for(int j=a[0]+1;j<10;j++)
                if(M[1][j].find(m)!=M[1][j].end())
                    num+=M[1][j][m];
        }
        else
        {
            if(a[0]>a[1])
            {
                for(int j=0;j<a[0];j++)
                    if(M[0][j].find(m)!=M[0][j].end())
                        num+=M[0][j][m];
            }
            else
            {
                for(int j=a[0]+1;j<10;j++)
                    if(M[1][j].find(m)!=M[1][j].end())
                        num+=M[1][j][m];
            }
        }
        if(num<k)k-=num;
        else
        {
            printf("%d",i);
            int len=7,kk=i;
            while(kk)a[len++]=kk%10,kk/=10;
            for(int j=1e5;j<1e7;j++)
            {
                ll temp=C*i+j;
                if(temp%n)continue;
                int res=deal(j),flag=1;
                while(res<7)a[res++]=0;
                res=len;
                for(int k=1;k<res-1;k++)
                    flag&=((a[k]>a[k-1]&&a[k]>a[k+1])||(a[k]<a[k-1]&&a[k]<a[k+1]));
                if(!flag)continue;
                k--;
                if(!k)
                {
                    printf("%07d\n",j);
                    exit(0);
                }
            }
        }
    }
    printf("-1\n");
}
int main()
{
    ll n,k;
    scanf("%I64d%I64d",&n,&k);
    Solve(n,k);
    return 0;
}