编写函数digit(int n, int k),返回正整数n中的第k位数字（从右边算起）··········

编写函数digit(int n, int k),返回正整数n中的第k位数字（从右边算起）。例如，digit(256,1)返回6，digit(256,3)返回2。如果k大于n的数字位数，则返回-1，如digit(256,4)

#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=1;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("正整数%d中的第%d位数字为:%d",n,k,digit(n,k));
    return 0;
}

如有戳无，欢迎指正