60pts，报两处WA，求调！QWQ！

#include<bits/stdc++.h>

using namespace std;
int sum;
int ws(int n) {
	while (n != 0) {
		n /= 10;
		sum++;
	}
	return sum;
}
void digit(int n,int k) {
	if (k == 1) {
		cout << n / (int)pow(10,ws(n) - 1);
		return;
	}
	cout << n % (int)pow(10,k) / (int)pow(10,k-1);
}


int main() {
	int n,k;
	sum = 0;
	cin >> n >> k;
	digit(n,k);
	return 0;
}