编写函数int take(int n,int k),该函数返回正整数n的第k位数字（从右数起）

例如：n=301259,则take(n,1)返回数字9；take(n,4)返回数字1

int take(int n,int k){
    int res;
    while(k--){
        res = n%10;
        n /= 10;
    }
    return res;
}