POJ 3208 Apocalypse Someday

Description
The number 666 is considered to be the occult “number of the beast” and is a well used number in all major apocalypse themed blockbuster movies. However the number 666 can’t always be used in the script so numbers such as 1666 are used instead. Let us call the numbers containing at least three contiguous sixes beastly numbers. The first few beastly numbers are 666, 1666, 2666, 3666, 4666, 5666…
Given a 1-based index n, your program should return the nth beastly number.

---

【题目分析】
数位DP，放一个传送门。%%%

---

【代码】

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int f[11][4],ans[11];
int main()
{
    f[0][0]=1;
    for (int i=0;i<=15;i++)
      for (int j=0;j<=3;j++)
      {
        f[i+1][j==3?3:j+1]+=f[i][j];
        f[i+1][j==3?3:0]+=f[i][j]*9;
      }

    int tt,x;
    scanf("%d",&tt);
    while (tt--)
    {
        scanf("%d",&x);
        int len=3,cnt=0,now=0;
        while (f[len][3]<x) len++;
        for (int i=len;i>=1;--i)
        {
            for (int j=0;j<=9;++j)
            {
                int y=cnt;
                if (cnt<3) {if (j==6) cnt++; else cnt=0;}
                int tot=now;
                for (int k=3;k>=0;--k) if (cnt+k>=3) tot+=f[i-1][k];
                if (tot>=x) {ans[i]=j;break;}
                now=tot;
                cnt=y;
            }
        }
        for (int i=len;i>=1;--i) printf("%d",ans[i]); printf("\n");
    }
}