Вот такой код. Хочу найти ошибку. Но найти не могу. Для всех перебранных случаев все вычисляется как надо. Не пойму как пройти тест.
#include <iostream>

using namespace std;

int foo(short pos);

int main()
{
    long i;
    short int y;
    cin>>i;
    for(int cnt_1 = 0; i > cnt_1; cnt_1++){
        cin>>y;
        cout<<foo(y)<<" ";
    }
}

int foo(short int pos)
{
    short int res(1);
    short int i(0);
    do
    {
        res = res + i;
        i++;
    }
    while(res<pos);

    if(res != pos)
        return 0;
    else
        return 1;
}

Какая тут ошибка?