「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