POJ 3208 Apocalypse Someday(디지털 dp)

1560 단어
제목: n번째 출력 연속 6개 6의 수 포함
아이디어:
dp[i][0]는 i 자릿수 중 첫 번째가 6이 아니라 666이 없는 수를 나타낸다. dp[i][1]는 i 자릿수 중 첫 번째가 6개 연속이고 666이 없는 수를 나타낸다. dp[i][2]는 i 자릿수 중 첫 번째가 6개 연속이고 666이 없는 수를 나타낸다. dp[i][3]는 i 자릿수에 666이 함유된 수를 나타낸다.
점차적인 관계를 쓰면 되고, 요구되는 수가 몇 자리인지 다시 확정한 다음, 높은 자리에서 낮은 자리까지 차례로 확정한다.
//132K	16MS
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;

ll dp[15][4];
const int trans[4][10]={
    {0,0,0,0,0,0,0,0,0,0},
    {3,3,3,3,3,3,0,3,3,3},
    {3,3,3,3,3,3,1,3,3,3},
    {3,3,3,3,3,3,2,3,3,3}
};
int main(){
    dp[0][0]=1;
    for(int i=1;i<=15;i++){
        dp[i][0]=(dp[i-1][0]+dp[i-1][1]+dp[i-1][2])*9;
        dp[i][1]=dp[i-1][0];
        dp[i][2]=dp[i-1][1];
        dp[i][3]=dp[i-1][3]*10+dp[i-1][2];
    }
    int T;
    scanf("%d",&T);
    while(T--){
        ll n;
        scanf("%I64d",&n);
        int dig=0;
        int cur=3;
        while(dp[dig][3]<n) dig++;
        while(dig){
            ll s=0;
            for(int i=0;i<10;i++){
                ll tmps=0;
                for(int j=3;j>=trans[cur][i];j--){
                    tmps+=dp[dig-1][j];
                }
                if(s+tmps>=n){
                    printf("%d",i);
                    cur=trans[cur][i];
                    break;
                }
                s+=tmps;
            }
            n-=s;
            dig--;
        }
        putchar('
'); } return 0; }

좋은 웹페이지 즐겨찾기