【1164】digi函数

【问题描述】
       在程序中定义一函数digit(n,k)，它能分离出整数n从右边数第k个数字，如digit(31859,3)=8，digit(2076,5)=0。
【输入格式】：
       正整数n和k。
【输出格式】：
       一个数字
【输入样例】：
       31859 3
【输出样例】：
       8
【参考程序】

#include <cstdio>
#include <iostream>
using namespace std;

long digit(int n, int k) {
	k--;
	if (k == 0) {
		return n % 10;
	} else {
		return digit(n/10, k);
	}
} 

int main() {
	int n, k;
	cin >> n >> k;
	cout << digit(n, k);
	
	return 0;
}