c++用函数写，90分，第三个点WA，求助

https://www.luogu.com.cn/problem/P1150

#include<bits/stdc++.h>
using namespace std;
int check(int n,int k)
{
    int num=0,x=0;//x为烟蒂数量 
    num+=n;//吸完第一批演烟 
    x+=n;//第一批烟的烟蒂数 
    while(x>k)
    {
        num+=x/k;
        x=x%k+x/k;
    }
    return num;
}
int main() 
{
int n,k,num=0;
cin>>n>>k;
num=check(n,k);
cout<<num;
    return 0;
}

不知道为什么第三个点会WA了，有哪位dalao可以帮助我一下，感激不尽