LeetCode刷题——题目2038、440（字典树思想）

Joey's_Code

已于 2022-03-24 00:01:37 修改

阅读量551

点赞数

分类专栏： LeetCode刷题记录文章标签：算法 c++ leetcode 数据结构

于 2022-03-23 23:52:15 首次发布

本文链接：https://blog.csdn.net/m0_46392078/article/details/123674886

版权

LeetCode刷题记录专栏收录该内容

9 篇文章 0 订阅

订阅专栏

2038：如果相邻两个颜色均相等则删除当前颜色

解题：

class Solution {
public:
    bool winnerOfGame(string colors) {
        //主要是计算最后谁走的回合数多；alice想要赢，次数必须大于bob
        int count=0;
        if(colors.size()<=2) return false;

        for(int i=2;i<colors.size();i++){
            if(colors[i]=='A' && colors[i-1]=='A'  && colors[i-2]=='A'){
                count++;
            }
            if(colors[i]=='B' && colors[i-1]=='B' && colors[i-2]=='B'){
                count--;
            }
        }
        return count>0;
    }
};

通过结果：

代码优化

class Solution {
public:
    bool winnerOfGame(string colors) {
        int count=0;
        for(int i=1;i<colors.size()-1;i++){
            if(colors[i]==colors[i-1] && colors[i-1]==colors[i+1]){
                count = count + ( colors[i]=='A'?1:-1 );
            }
        }
        return count>0;
    }
};

通过结果：

440.字典序的第K小数字

解题：

字典序是基于字母顺序排列的单词按照字母顺序排列。对于数字1、2、3……n的排列，不同排列的先后关系是从左到右逐个比较对应的数字的先来来决定的：

比如对于12和3来说，3大于12，因为3大于12的第一位数；

对于12和13来说，13>12，因为第一位数是相等的，但是2<3，所以13>12;

class Solution {
public:
    long getCount(int prefix,int n){
        long current=prefix;
        long next_prefix=current+1;
        int count=0;
        while(current<=n){  //不能超过上界
            count += (next_prefix>(long)(n+1)?n+1:next_prefix) - current;
            current*=10;
            next_prefix*=10;
        }
        return count;
    } 
    int findKthNumber(int n, int k) {
        int prefix=1;
        int p_current=1;
        while(p_current<k){
            int count=getCount(prefix,n);
            if(p_current+count>k){
                prefix=prefix*10;  //前缀的子串
                p_current++;  //p向后移动一个
            }
            else{
                prefix++;  //当前前缀的下一个前缀
                p_current=p_current+count;  //跳过上一个前缀的所有拥有的数的个数
            }
        }
        return prefix;
    }
};

代码不是自己想出来的，参考了大佬的思想，使用了字典树的思想，比如1，是第二层的第一个树结点，将以1为前缀的子串作为自己的孩子节点，2、3、4等作为自己的兄弟结点，以1-13为例：

通过对前缀进行*10的操作进入下一层。通过p_current控制当前所读位置，对我们假象中的字典树进行遍历（因为这里是不需要构造一个真正的树的，只是借用这个思想），直到p=k，输出此时树上“结点”的值即可，也就是prefix，看图能发现这其实是树的先序遍历。

根据这个特点，先梳理主要函数的思路：

①初始化当前的前缀是1，初始化当前p的位置是1；

②进行循环，循环条件为p_current<k，p相当于是在计算我们到底已经遍历过多少个数字，直到我们确定p_current=k，输出当前prefix的值即可。

③在while循环中，首先对当前的前缀prefix进行计算，它有多少个前缀，其中包括他本身，该数字记为count；如果p_current+count>k，也就是在当前前缀的基础上我们可以遍历多少个数，如果遍历数字大于k，那么k就是在这个前缀及其子串中，于是我们将prefix*10，这样就进入树的下一层了。

注意：首先需要明白，如果count=0，但是k=1怎么办，这是只会出现在最开始的情况，说明此时n=1，这个情况下根本不会进入while循环，直接到最后一步： return prefix（此时prefix=1）；其次需要注意，为什么不能是p+count>=k，这样k也应该在我们此次可遍历的范围，但是，如果真的这样了，那么在这个if判断中，p_current++之后，p_current就永远无法到达真正的k值，加完之后直接结束while循环，输出的prefix的值就是本来该输出的值的前缀的下一个前缀。（好像说的比较乱，可以修改代码进行测试，这样比较直观）。

接if后的else，这个else有两个作用，一个是比如当前缀=1的时候，k就不在这个范围里面，那么直接进入前缀2再去找看有多少个；一个是如果我们在if中确定过比如前缀1的所有数字中就有我们需要的第k个数，那我们就可以通过if进入下一层的第2个结点，以13为例，这个结点就是11（因为10在if判断中就已经试探过了！注意里面的p_current++就是因为这个才设置的++；前面说过，能进入循环一定确定k>=2）；如果此时k=2，循环会在第一次if就结束，所以这里就假设k=4，（11、12、13的getCount都是1），进入不了第2次的if语句，这是最棒的地方，因为我们使用else就一直在试探1的孩子结点们，从10，到11,12……，直到我们试探出来p_current=k，p_current在试探了11之后，发现他是1，那就p_current+1=3，然后下一个前缀12，还是1，p_current+1=4，OK！结束循环，输出prefix=12。

④getCounth（）的思路，这个就比较简单了，使用*10的思想不断深入层数

 count += (next_prefix>(long)(n+1)?n+1:next_prefix) - current;

这句话是最关键的，把它写的更仔细：

count = count + min( next_prefix , n+1 ) -current;

next_prefix是下一个前缀的起点，current是现在前缀的起点，所以很明显两个相减的话就是目前子节点的数量；举例上面的n=13来说，当next_prefix=1的时候，我们记录1这一层的就1这一个，1的下一个前缀是2，那就2-1，得到1，也就是此时记录下了1这个结点，此时count=1；继续深入*10，进入1的下一层，此时如果我们不使用n+1来约束溢出可以发现会直接超出边界，毕竟next_prefix是到不了20的。我们用边界限制（n+1）减去当前前缀的值，得到这一层的结点数（14-10）=4，再加上上一层的1，得到5，完毕；

继续举例：那当n=101的时候，20-10，就是前缀1的下一层的，10个结点，然后current再进入下一层，此时next_prefix=200，超界线了，所以是101+1-current=101-100=1。

通过结果：