[URAL 1148] Building Towers

1 篇文章 0 订阅

URAL 1148
题解:
考虑朴素的 dp d p , dp(i,j,k) d p ( i , j , k ) 表示考虑到第i层,当前层放 j j 个块,已经放了k个块的情况数量.
空间约为 8×60×70×2500bytes80MB 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 个数。
总时间复杂度O(h3min(n,h2))
没有任何优化(不考虑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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值