题解:ABC321C - 321-like Searcher
·题目
链接:Atcoder。
链接:洛谷。
·难度
算法难度:B。
思维难度:B。
调码难度:B。
综合评价:见洛谷链接。
·算法
动态规划+深度优先搜索。
·思路
首先想到爆搜,能够获得0分的好成绩(abc没有部分分)。
于是考虑计算出所有“i位数,最高位为j,满足此条件的321-like数的数量”,以此设计状态f[i][j],初始值自然是f[1][j]都为1。需要注意的是初始值也要包括f[1][0],以便后续转移,可是不计入答案总数。
转移很简单,对于f[i][j],用所有f[i-1][k](k=0、1、2、……、j-1)求和算出。
然后按照先从小到大遍历i,在从小到大遍历j的顺序遍历所有状态(i用到1~14,j用到1~9即可),求出前缀和,判断所求是否在该状态范围内,若在该范围内就dfs,遍历到答案即可。
·代价
由于是常数,不估计时间复杂度,但是一定够用。
·细节
f的双下标定义范围:25、10。
·代码
#include<bits/stdc++.h>
using namespace std;
long long f[15][10]={},ans=0,k=0,n=0,sum=0;
inline void dfs(long long d,long long num);
int main(){
scanf("%lld",&k);
for(long long i=0;i<10;i++){
f[1][i]=1;
}
for(long long i=2;i<15;i++){
for(long long j=1;j<10;j++){
for(long long k=0;k<j;k++){
f[i][j]+=f[i-1][k];
}
}
}
//预处理出所有状态
for(long long i=1;i<15;i++){
for(long long j=1;j<10;j++){
if(sum+f[i][j]>=k){
//如果答案在该范围内,进行dfs
n=j;
//n就负责记录目前为止遍历到的数
dfs(i-1,j);
//第一个参数表示该处理第几位了,第二个参数是上一位是多少
printf("%lld\n",ans);
//输出答案
return 0;
}
sum+=f[i][j];
//总和累加
}
}
return 0;
}
inline void dfs(long long d,long long num){
if(ans!=0){
return;
}
//已经找到答案就不用接着找了
if(d==0){
//已经遍历完整个数
sum++;
//总和累加
if(sum==k){
ans=n;
//数量够了就是答案
return;
}
}
for(long long i=0;i<num;i++){
n=n*10+i;
dfs(d-1,i);
n/=10;
}
//遍历这一位,同时往下递归,注意回溯
return;
}
·注意
开long long!!!……(省略一古戈尔个感叹号)