digit解法

各位大佬好像没多少用int digit的，我就发个简单的题解吧。

#include<iostream>
using namespace std;

int digit(int a, int b) {
	for (int i = 0; i < b-1; i++)
	{
		a /= 10;
	}
	a %= 10;
	return a;
}

int main() {
	int a, b;
	cin >> a >> b;
	cout << digit(a, b);
}