URAL 1148
题解:
考虑朴素的
dp
d
p
,
dp(i,j,k)
d
p
(
i
,
j
,
k
)
表示考虑到第i层,当前层放
j
j
个块,已经放了个块的情况数量.
空间约为
8×60×70×2500bytes≈80MB
8
×
60
×
70
×
2500
b
y
t
e
s
≈
80
M
B
远超出空间限制。
由于需要求出字典序为第k的解,所以不能直接使用滚动数组。
此时考虑用时间换空间,由于
dp
d
p
过程很难求逆(或者可能不能求逆)
此时只要每次都使用滚动数组计算出对于
i=k
i
=
k
时的
dp
d
p
值, 并确定所有询问的答案的第
k
k
个数。
总时间复杂度
没有任何优化(不考虑dp的废状态)的情况下
800ms
800
m
s
就能通过此题.
#include<bits/stdc++.h>
#define LL long long
#define ull unsigned long long
#define ULL ull
#define mp make_pair
#define pii pair<int,int>
#define piii pair<int, pii >
#define pll pair <ll,ll>
#define pb push_back
#define big 20160116
#define INF 2147483647
#define pq priority_queue
using namespace std;
inline int read(){
int x=0,f=1;
char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
namespace Mymath{
LL qp(LL x,LL p,LL mod){
LL ans=1;
while (p){
if (p&1) ans=ans*x%mod;
x=x*x%mod;
p>>=1;
}
return ans;
}
LL inv(LL x,LL mod){
return qp(x,mod-2,mod);
}
LL C(LL N,LL K,LL fact[],LL mod){
return fact[N]*inv(fact[K],mod)%mod*inv(fact[N-K],mod)%mod;
}
template <typename Tp> Tp gcd(Tp A,Tp B){
if (B==0) return A;
return gcd(B,A%B);
}
template <typename Tp> Tp lcm(Tp A,Tp B){
return A*B/gcd(A,B);
}
};
namespace fwt{
using namespace Mymath;
void FWT(int a[],int n,LL mod)
{
for(int d=1;d<n;d<<=1)
for(int m=d<<1,i=0;i<n;i+=m)
for(int j=0;j<d;j++)
{
int x=a[i+j],y=a[i+j+d];
a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod;
//xor:a[i+j]=x+y,a[i+j+d]=x-y;
//and:a[i+j]=x+y;
//or:a[i+j+d]=x+y;
}
}
void UFWT(int a[],int n,LL mod)
{
LL rev=inv(2,mod);
for(int d=1;d<n;d<<=1)
for(int m=d<<1,i=0;i<n;i+=m)
for(int j=0;j<d;j++)
{
int x=a[i+j],y=a[i+j+d];
a[i+j]=1LL*(x+y)*rev%mod,a[i+j+d]=(1LL*(x-y)*rev%mod+mod)%mod;
//xor:a[i+j]=(x+y)/2,a[i+j+d]=(x-y)/2;
//and:a[i+j]=x-y;
//or:a[i+j+d]=y-x;
}
}
void solve(int a[],int b[],int n,LL mod)
{
FWT(a,n,mod);
FWT(b,n,mod);
for(int i=0;i<n;i++) a[i]=1LL*a[i]*b[i]%mod;
UFWT(a,n,mod);
}
};
const int Maxn=61;
LL dp[2][72][2500];
int n,h,m;
void calc(int h){
memset(dp,0,sizeof(dp));
for (int i=1;i<=70;i++){
dp[1][i][i]=1;
}
for (int ii=2;ii<=h;ii++){
int i=ii&1;
for (int j=1;j<=70;j++){
for (int k=j;k<=n;k++){
dp[i][j][k]=dp[i^1][j-1][k-j]+dp[i^1][j+1][k-j];
}
}
}
}
vector<pair<LL,pair<int,int> > >v;
vector<int> ans[2333];
void Pr(int id){
for (int i=0;i<=70;i++){
for (int j=1;j<2500;j++){
dp[id][i][j]+=dp[id][i][j-1];
}
}
}
int main(){
scanf("%d %d %d",&n,&h,&m);
if (n>=2500){
n=2499;
}
calc(h);
LL res=0;
for (int i=1;i<=n;i++){
res+=dp[h&1][m][i];
}
printf("%I64d\n",res);
//return 0;
while (1){
LL x;
scanf("%I64d",&x);
if (x==-1) break;
v.pb(mp(x,mp(m,n)));
}
int q=v.size();
for (int i=0;i<q;i++) ans[i].pb(m);
for (int i=h-1;i>=1;i--){
Pr(i&1);
for (int j=0;j<v.size();j++){
int rm=v[j].second.first,rn=v[j].second.second;
LL sb=v[j].first;
int h1=i&1;
if (sb>dp[h1][rm-1][rn-rm]){
sb-=dp[h1][rm-1][rn-rm];
rn-=rm;
rm++;
v[j].second=mp(rm,rn);
ans[j].pb(rm);
}
else{
rn-=rm;
rm--;
v[j].second=mp(rm,rn);
ans[j].pb(rm);
}
v[j].first=sb;
}
calc(i);
}
for (int i=0;i<q;i++){
for (int j=0;j<ans[i].size();j++){
printf("%d ",ans[i][j]);
}
printf("\n");
}
return 0;
}