hdu 6468 zyb的面试

Posted 2020-11-25 foreveroier

题目传送门
解题思路:将这些 1~n 的数，分为一个树，然后再往下找子节点，用搜索来遍历这棵树，最后找到第k个数

#include<bits/stdc++.h>
using namespace std;
inline int read()
{
    int x=0,k=1;char c=getchar();
    while(c<'0' || c>'9'){if(c=='-') k=0;c=getchar();}
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return k?x:-x;
}
int T,n,k,s;
bool ck;
void dfs(int x)
{
    if(ck) return;//如果已经找到了，那么直接返回 
    s++;//次数加 1，(s 表示第几个数) 
    if(s==k)//如果是第 k 个数了 
    {
        printf("%d
",x);//那么输出当前数 
        ck=1;//标记为找到 
        return;//返回 
    }
    for(int i=0;i<=9;i++)//因为是个位，并且前面有十位所以是从 0 至 9 
    {
        int newx=(x<<3)+(x<<1)+i;//新的 x 设为原来的 (x<<3)+(x<<1)+i(等同于 x*10+i) 
        if(newx>n) return;//如果这个新的 x 比 n 大，那么就返回 
        dfs(newx);//如果这个新的 x 小于等于 n，那么就继续搜索 
    }
}
int main()
{
    T=read();//输入 
    while(T--)
    {
        n=read(),k=read();//输入 
        ck=s=0;//将次数和判断归 0 
        for(int i=1;i<=9 && !ck;i++)//因为是首位，所以是 1 至 9，并且要判断是否已经找到该数 
            dfs(i);//搜索当前 i 
    }
    return 0;
}