数的分离c++代码

#include<iostream>
#include<cmath>
using namespace std;
int digit(long long n,int k){
    int tmp;
    for(int i=1;i<=k;i++){
        tmp=n%10;
        n=n/10;
    }
    return tmp;
} 
int main(){
    long long a;
    int b;
    cin>>a>>b;
    cout<<digit(a,b)<<endl;
}