C++实验返回一个正整数n的第k位数码

编写一个函数，用于返回一个正整数n的第k位数码。注意，数码的位次顺序为从右到左，从0开始.

#include<stdio.h>
int digit(int n,int k){
    int i,a,count=0,copy_n;
    copy_n=n;
    while(copy_n!=0){
        copy_n/=10;
        count++;
    }
    if(k>count){
        return -1;}
    else{
        for(i=0;i<=k;i++){
            a=n%10;
            n/=10;}
        return a;
    }}
int main(){
    int n,k;
    scanf("%d %d",&n,&k);
    if(digit(n,k)==-1)
        printf("k大于数字位数!");
    else
        printf("n%d的第%d位数:%d",n,k,digit(n,k));
    return 0;
}