「SCOI2014」方伯伯的商场之旅 解题报告

时间:2023-01-26 18:59:55

「SCOI2014」方伯伯的商场之旅

我一开始的想法会被两个相同的集合位置去重给搞死,不过应该还是可以写的,讨论起来老麻烦。

可以先钦定在\(1\)号点集合,然后往后调整一部分。

具体一点,通过前缀和减去后缀和的正负性移动

写的时候把\(sum\)压进去搞会非常简单


Code

#include <cstdio>
#include <cstring>
#define ll long long
ll dp[25][3000];int bit[25],k;
ll dfs(int dep,int sum,int limit)
{
    if(dep==0) return sum;
    if(!limit&&~dp[dep][sum]) return dp[dep][sum];
    ll ret=0;
    for(int i=0,up=limit?bit[dep]:k-1;i<=up;i++)
        ret+=dfs(dep-1,sum+(dep-1)*i,limit&i==up);
    return limit?ret:dp[dep][sum]=ret;
}
ll dfs(int dep,int pos,int sum,int limit)
{
    if(sum<0) return 0;
    if(dep==0) return sum;
    if(!limit&&~dp[dep][sum]) return dp[dep][sum];
    ll ret=0;
    for(int i=0,up=limit?bit[dep]:k-1;i<=up;i++)
        ret+=dfs(dep-1,pos,sum+(dep<pos?-1ll:1ll)*i,limit&i==up);
    return limit?ret:dp[dep][sum]=ret;
}
ll cal(ll x)
{
    int cnt=0;
    while(x)
    {
        bit[++cnt]=x%k;
        x/=k;
    }
    memset(dp,-1,sizeof dp);
    ll ret=dfs(cnt,0,1);
    for(int i=2;i<=cnt;i++)
        memset(dp,-1,sizeof dp),ret-=dfs(cnt,i,0,1);
    return ret;
}
int main()
{
    ll l,r;
    scanf("%lld%lld%d",&l,&r,&k);
    printf("%lld\n",cal(r)-cal(l-1));
    return 0;
}

2019.2.23