将整数按权重排序-记忆性dfs

题目描述：

我们将整数 x 的 权重 定义为按照下述规则将 x 变成 1 所需要的步数：

如果 x 是偶数，那么 x = x / 2

如果 x 是奇数，那么 x = 3 * x + 1

比方说，x=3 的权重为 7 。因为 3 需要 7 步变成 1 （3 --> 10 --> 5 --> 16 --> 8 --> 4 --> 2 --> 1）。

给你三个整数 lo， hi 和 k 。你的任务是将区间 [lo, hi] 之间的整数按照它们的权重 升序排序 ，如果大于等于 2 个整数有 相同 的权重，那么按照数字自身的数值 升序排序 。

请你返回区间 [lo, hi] 之间的整数按权重排序后的第 k 个数。

注意，题目保证对于任意整数 x （lo <= x <= hi） ，它变成 1 所需要的步数是一个 32 位有符号整数。

链接: https://leetcode-cn.com/problems/sort-integers-by-the-power-value/.

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6;
typedef long long ll;
int ch[100086];
int p[100086];
int dfs(int a)
{
    int ans = 1;
    if (a == 1)
        return 0;
    if (ch[a])
        return ch[a];
    ans += (a % 2 == 0) ? dfs(a / 2)  : dfs(3 * a + 1) ;
    ch[a] = ans;
    return ans;
}
int main()
{
    int a, b, k;
    cin >> a >> b >> k;
    for (int i = a; i <= b; i++)
    {
        p[i] = dfs(i);
    }
    int len = b - a + 1;
    cout << len << endl;
    sort(p + a, p + a + len + 1);
    for (int i = a; i <= b; i++)
    {
        if (p[i] == p[a + k - 1])
        {
            cout << i << endl;
            break;
        }
    }
    return 0;
}