递归求任意整数的根

#include <iostream>
#include <string>
#include <cmath>
using namespace std;
int gen(int n)
{
    if(n<10)
    {
        return n;
    }
    else
    {
        int x=0;
        do
        {
            x+=n%10;
            n=n/10;
        }
        while(n>0);
        return gen(x);
    }
}
int main()
{
    int a;
    cin>>a;
    cout<<gen(a);
}