POJ 3208 Apocalypse Someday

题意：

　　　将含有连续的三个6的数称为不吉利数，比如666，1666，6662，但是6266吉利。则666为第一个不吉利数，输入整数n，求第n个不吉利数。(n <= 5*10^7)

解法：

　　　如果是给出n，求n以内的不吉利数有多少个，就是一个普通的数位DP。所以，就二分答案，对于每一个mid，求一下小于等于mid的数中不吉利数的个数。

tag:二分法，数位DP

 /*
  * Author:  Plumrain
  * Created Time:  2013-12-14 20:55
  * File Name: DP-POJ-3208.cpp
  */
 #include <iostream>
 #include <cstdio>
 #include <cstring>

 using namespace std;

 #define CLR(x) memset(x, 0, sizeof(x))
 typedef long long int64;
 int64 d1[][][], d2[][][];

 int64 get_num(int64 x)
 {
     int len = , dit[];
     bool xxx = ;
     while (x){
         dit[len++] = x % ;
         x /= ;
         if (len >  && dit[len-] ==  && dit[len-] ==  && dit[len-] == )
             xxx = ;
     }
     dit[len] = ; dit[len+] = ;

     int64 ret = ;
     if (xxx) ret = ;
     bool flag = ;
     for (int i = len-; i >= ; -- i){
         for (int j = ; j < ; ++ j)
             for (int k = ; k < ; ++ k)
                 ret += dit[i] * d2[i][j][k];

         if (flag){
             for (int j = ; j < ; ++ j)
                 for (int k = ; k < ; ++ k)
                     ret += dit[i] * d1[i][j][k];
         }
         else {
             if (dit[i+] ==   && dit[i+] ==  && dit[i] > ){
                 for (int j = ; j < ; ++ j)
                     for (int k = ; k < ; ++ k)
                         ret += d1[i][j][k];
             }
             else if (dit[i+] ==  && dit[i] > ){
                 for (int j = ; j < ; ++ j)
                     ret += d1[i][][j];
             }
             else if (dit[i] > ) ret += d1[i][][];
         }

         if (dit[i] ==  && dit[i+] ==  && dit[i+] == )
             flag = ;
     }
     return ret;
 }

 int64 gao(int64 n)
 {
     int64 l = , r = ;
     while (l <= r){
         int64 mid = (l + r) >> ;
         if (get_num(mid) < n) l = mid + ;
         else r = mid - ;
     }
     return l;
 }

 int main()
 {
     CLR (d1); CLR (d2);
     d1[][][] = ;
     for (int i = ; i < ; ++ i){
         d1[][i][] = ;
         for (int j = ; j < ; ++ j)
             d1[][i][j] = ;
     }

     for (int i = ; i < ; ++ i){
         d2[i][][] += d1[i-][][];
         d1[i][][] -= d1[i-][][];
         for (int j = ; j < ; ++ j)
             for (int k = ; k < ; ++ k)
                 for (int l = ; l < ; ++ l){
                     d2[i][j][k] += d2[i-][k][l];
                     d1[i][j][k] += d1[i-][k][l];
                 }
     }

     int T;
     scanf ("%d", &T);
     while (T--){
         int64 n;
         cin >> n;
         cout << gao(n) << endl;
     }
     return ;
 }