pku 1019 數學題（遞推式）

#include<iostream>
using namespace std;

const int hmax=54000;

int array[hmax];
__int64 sum[hmax];

int solve(int m)
{
 int i,j;
    int temp[10];

 for(i=1;i<hmax;i++)
  if(m<=sum[i])break;
 m-=sum[i-1];
 for(i=0;i<hmax;i++)
  if(m<=array[i])break;
 m-=array[i-1];
 
 j=0;
 while(i)
 {
        temp[j]=i%10;
  i/=10;
  j++;
 }

 return temp[j-m];
}
int main()
{
 int i,n,m,t;

    array[0]=0;
    for(i=1;i<=hmax;i++)
 {
        if(i>=10000)array[i]=array[i-1]+5;
  else if(i>=1000)array[i]=array[i-1]+4;
  else if(i>=100)array[i]=array[i-1]+3;
  else if(i>=10)array[i]=array[i-1]+2;
  else array[i]=array[i-1]+1;
 }
 sum[0]=0;
 for(i=0;i<hmax;i++)
  sum[i]=sum[i-1]+array[i];

    while(EOF!=scanf("%d",&t))
 {
     while(t--)
  {
            scanf("%d",&n);
            m=solve(n);
   printf("%d/n",m);
  }
 }
 return 0;
}