#include "nmod.h"
#include "nmod_poly.h"
ulong
_nmod_poly_div_root(nn_ptr Q, nn_srcptr A, slong len, ulong c, nmod_t mod)
{
ulong r, t;
slong i;
if (len < 2)
return UWORD(0);
t = A[len - 2];
r = Q[len - 2] = A[len - 1];
for (i = len - 2; i > 0; i--)
{
r = nmod_add(nmod_mul(r, c, mod), t, mod);
t = A[i-1];
Q[i-1] = r;
}
r = nmod_add(nmod_mul(r, c, mod), t, mod);
return r;
}
ulong
nmod_poly_div_root(nmod_poly_t Q,
const nmod_poly_t A, ulong c)
{
ulong rem;
slong len = A->length;
if (len == 0)
{
nmod_poly_zero(Q);
return UWORD(0);
}
if (len == 1)
{
rem = A->coeffs[0];
nmod_poly_zero(Q);
return rem;
}
if (c == 0)
{
rem = A->coeffs[0];
nmod_poly_shift_right(Q, A, 1);
return rem;
}
nmod_poly_fit_length(Q, len - 1);
rem = _nmod_poly_div_root(Q->coeffs, A->coeffs, len, c, Q->mod);
Q->length = len - 1;
return rem;
}