Vijos训练计划 1-1 计数问题

刚刚参加了Vijos的训练计划，很长时间不学习了，有一些东西都已经忘记了，但是昨天看到学长写的代码，发现自己要走的路还很长，这道题时训练计划的第一题，所以难度比较简单，下面附上代码。

#include<bits/stdc++.h>
/*
    作者:Unis
    日期:2018.3.23
    题目:计数问题
*/
using namespace std;

int JudgeIndex(int x){
    int t_x = x;
    int countx = 0;
    do{

        t_x /= 10;
        countx++;

    }while(t_x != 0);

    return countx;
}


int main() {
    //10 = 1和0 11 = 1 和 1
    int x,n;
    int flag[10];
    int t,e,countx=0;
    while(scanf("%d %d",&n,&x) != EOF){

        memset(flag,0,sizeof(flag));
        //testing
//        for(int i = 0;i < 22;i++){
//            printf("%d\n",i);
//        }

        for(int i = 1;i <= n;i++){
            t = i,e = i;

            if(i >= 10){

                countx = JudgeIndex(i);
                //printf("index=%d\n",countx);

                do{
        
                    e = t % 10;
                    t = t / 10;
                    flag[e]++;
                    countx--;

                }while(countx != 0);

            }
            else{
                flag[i]++;
            }

            //printf("flag = %d\n",flag[i]);

        }

        printf("%d\n" ,flag[x]);
    }
	return 0;
}