https://atcoder.jp/contests/abc115/tasks/abc115_d
题意:第i级汉堡组成为’B’+第i-1级汉堡+‘P’+第i-1级汉堡+‘B’,‘B’指面包,‘P’指肉,1级汉堡为‘P’,求第n级汉堡的前k层中有多少肉(‘P’)。
思路:先从1到n找出第一个层数比X大的级别a,之后级别再高对答案的影响只有最前面那几个“B”,那就是暴力找a中的前m-(n-a)层里有几个肉,如果正在查找的个数包含一些整层还可以直接调用之前算的肉层数。
#include<cstdio>
#include<string>
#include<cstring>
#include<utility>
#include<cmath>
#include<map>
#include<queue>
#include<set>
#include<algorithm>
#include<vector>
#include<iostream>
#define ll long long
#define pii pair<int,int>
#define mp make_pair
#define fi first
#define se second
using namespace std;
ll a[100],b[100];
ll dfs(ll x,int y)
{
if(x<=0)
{
return 0;
}
if(y==0)
{
return 1;
}
if(a[y]==x)
{
return b[y];
}
ll ans=0;//a[y]>x
int i;
if(x>=a[y-1]+1)//b ***** p ***** b
{
ans+=b[y-1];
if(x>=a[y-1]+2)
{
ans++;
if(x>a[y-1]+2)
{
x-=a[y-1]+2;
for(i=0;i<y;i++)
{
if(a[i]+(y-i-1)>=x)
{
break;
}
}
ans+=dfs(x-(y-i-1),i);
}
}
return ans;
}
else
{
for(i=0;i<y;i++)
{
if(a[i]+(y-i)>=x)
{
break;
}
}
return dfs(x-(y-i),i);
}
}
int main()
{
int i,j,k,n,x,y;
ll m,an=1,aa=1;
scanf("%d%lld",&n,&m);
for(i=0;i<=n;i++)
{
a[i]=an;
b[i]=aa;
if(a[i]>=m)
{
break;
}
an=an*2+3;
aa=aa*2+1;
}
// printf("%d",m-n+i);
cout<<dfs(m-(n-i),i);
return 0;
}