POJ 3208-Apocalypse Someday(数位dp)

题意:给定n,输出第n大包含666的数字。

分析:dp[i][j][k][l]表示 长度为i,当前位是否是6,前一位是否6,是否已经包含666,表示的数量,再用二分找出第n大的这样的数字。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
const ll INFll =0xfffffffffLL;
const int INF= 0x7ffffff;
const int mod =;
ll dp[][][][],n;
int bit[];
ll dfs(int i,int j,int k,int l,int e){
if(i==)
return l;
if(!e&&dp[i][j][k][l]!=-)
return dp[i][j][k][l];
int u=e?bit[i]:;
ll num=;
for(int v=;v<=u;++v){
if(l==)
num+=dfs(i-,v==,j,,(e&&v==u));
else
num+=dfs(i-,v==,j,j&&k&&v==,(e&&v==u));
}
return e?num:dp[i][j][k][l]=num;
}
ll getnum(ll x){
int len=;
while(x){
bit[++len]=x%;
x/=;
}
return dfs(len,,,,);
}
int main()
{
memset(dp,-,sizeof(dp));
int t;
scanf("%d",&t);
ll n;
while(t--){
scanf("%I64d",&n);
ll l=,r=INFll,mid;
while(l<=r){
mid=(l+r)>>;
if(getnum(mid)<n)l=mid+;
else r=mid-;
}
printf("%I64d\n",l);
}
return ;
}
上一篇:[Effective Java]第三章 对所有对象都通用的方法


下一篇:RVIZ实现模拟控制小车