public static int squares(int a, int b) {

        int res = 0;
        for (int i = (int) Math.sqrt(b); i >= Math.sqrt(a); i--) {
            if (i * i >= a && i * i <= b) {
                res++;
            }
        }
        return res;

    }

let A = sqrt(Double(a))
        let intA = Int(A)
        let B = sqrt(Double(b))
        let intB = Int(B)
        if a == intA*intA{
        return Int(B-A+1)
        }
        else if b == intB*intB{
            return Int(B-A+1)
        }
        else {
            return intB-intA
        }

return (floor(sqrt(b)) - ceil(sqrt(a)) + 1);

int squares(int a, int b) {
    int i  = 1;
    int count = 0;
    while(i*i <= b){
        if(i*i >= a){
            count++;
        }
        i++;
    }
    return count;
}