18361 数绵羊

18361 数绵羊

时间限制:1000MS 代码长度限制:10KB
提交次数:7 通过次数:4

题型: 编程题 语言: 不限定
Description
小粥今晚依然因相思而失眠，所以开始数绵羊：
一只绵羊，两只绵羊，三只绵羊，四只绵羊，五只绵羊，六只绵羊，七只绵羊，
八只绵羊，九只绵羊，十只绵羊，十二只绵羊，十一只绵羊，十三只绵羊，十四只绵羊，十五只绵羊。。。

什么，你说他数错了？绝对是因为你不够聪明所以看不出来他在数什么。

他数绵羊的顺序满足以下条件：

(1). 所数的数字为正数整；

(2). 在满足(1)的前提下，二进制表示中最高位1越小的越早被数到；

(3). 在满足(1),(2)的前提下，二进制表示中1的个数越少的越早被数到；

(4). 在满足(1),(2),(3)的前提下，二进制表示中字典序越小的越早被数到；

现在小粥想有一些问题，他想知道他自己数的第n个数是多少，或者是某个数字x是第几个被数到的数字。

如果不解决这些问题的话小粥就要数绵羊数到天亮了，所以希望你们帮帮失眠的小粥解决这个问题，好让小粥安心睡觉吧。（说好的相思呢？

输入格式
第一行包含一个数字T(0 <= T <= 100000)，表示下面总共有T个问题。

接下来T行每行包含一个字符（N或X）和一个数字a，以空格分隔。

如果行首是N，则表示这个问题是问第a个数是多少；

如果行首是X，则表示这个问题是问a是第几个被数到的数。(0 < a <= 2147483647)

输出格式
输出T行，每行包含对应问题的结果。

输入样例
13
N 1
N 2
N 3
N 4
N 5
N 6
N 7
N 8
N 9
N 10
N 11
N 1234567
X 1330566

输出样例
1
2
3
4
5
6
7
8
9
10
12
1330566
1234567

之前不会做这个题，现在隔了半年突然记起这道题，一看这不就是排列组合吗？o(1)算法。
举例一个7位数1100101（2），问他应该出现的位置，按照排列他出现的位置应该大于10000000（2），之后看除了最高位会有多少个1，这里是三个1，那么在他前面一定有除最高位外1的个数为1的数，2的数，因为除最高位外剩下有6个位，那么个数就是C(6,1)+C(6,2)(5个里面取一个以及5个取两个)，我们加上，之后接着问题就转换为寻找100101这个排列在6个位置里面有3个1的字典序的大小，那我们从首位开始，如果这个位置出现1，那在他前面的数还能多出多少个呢？能多出在这个最高为选0，其他1自由排列的个数，也就是C(5,3)，也就是10+后面5个位有3个1的数的个数，后面继续找，再次出现1的话，证明他还能比一些数大，具体是什么数呢？1000+后面3个位置两个1的个数对吧，后面类推，反过来也是一样的道理。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=2e5+10;
typedef long long ll;
int cnt[33][33];//排列组合
string str;
int n;
void init()
{
    for(int i=0;i<=32;i++)
    {
        for(int j=0;j<=i;j++)
        {
            if(j==0||j==i)
            {
                cnt[i][j]=1;
            }
            else cnt[i][j]=cnt[i-1][j]+cnt[i-1][j-1];
        }
    }
}
int gethighbit()//获取最高位
{
    int mmax=0;
    for(int i=0;i<=31;i++)
    {
        if((n>>i)&1)
        {
            mmax=i;
        }
    }
    return mmax;
}
int getone()//除了最高位有几个1
{
    int x=0;
    for(int i=0;i<gethighbit();i++)
    {
        if((n>>i)&1)
        {
            x++;
        }
    }
    return x;
}
void solve2()
{
    int mmax=gethighbit();//他的原数的位数
    int ans=(1<<mmax)-1;
    int tp=0;//这个数有多少个1
    while(ans+cnt[mmax][tp]<n)
    {
        ans+=cnt[mmax][tp];
        tp++;
    }
    int pos=n-ans;//tp个1的第pos个排列
    ans=(1<<mmax);
    for(int i=mmax-1;i>=0;i--)
    {
        //cout<<pos<<' '<<i<<' '<<tp<<' '<<cnt[i][tp]<<' '<<ans<<endl;
        if(tp==0)
        {
            break;
        }
        else if(i+1==tp)
        {
            for(int j=0;j<=i;j++)
            {
                ans+=(1<<j);
            }
            break;
        }
        else if(pos>cnt[i][tp])
        {
            ans+=(1<<i);
            pos-=cnt[i][tp];
            tp--;
        }
    }
    cout<<ans<<endl;
}
void solve1()
{
    int mmax=gethighbit();//最高位1
    int ans=(1<<mmax)-1;
    int x=getone();//除了最高位有几个1
    for(int i=0;i<x;i++)//加上小于他的1的个数的个数
    {
        ans+=cnt[mmax][i];
    }
    for(int i=mmax-1;i>=1;i--)
    {
        if((n>>i)&1)
        {
            ans+=cnt[i][x];
            x--;
        }
    }
    ans++;
    cout<<ans<<endl;
}
int main ()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    init();
    int T;
    cin>>T;
    while(T--)
    {
        cin>>str>>n;
        if(str[0]=='X')
        {
            solve1();
        }
        else solve2();
    }
    return 0;
}