BZOJ2742 [HEOI2012]Akai的数学作业

Description

• 这里是广袤无垠的宇宙这里是一泻千里的银河
• 这里是独一无二的太阳系
• 这里是蔚蓝色的地球
• 这里，就是这里，是富饶的中国大陆！
• 这里是神奇的河北大地
• 这里是美丽的唐山
• 这里是神话般的唐山一中
• 这里是Akai曾经的教室
• 黑板上还留有当年Akai做过的数学作业，其实也并不是什么很困难的题目:
  “给出一个一元n次方程：$a_0 + a_1x + a_2 x^2 +…+ a_nx^n= 0$。求此方程的所有有理数解。”
• Akai至今还深刻记得当年熬夜奋战求解的时光，他甚至还能记得浪费了多少草稿纸，
  但是却怎么也想不起来最后的答案是多少了。
• 你能帮助他么？

Input

• 第一行一个整数n。
• 第二行n+1个整数，分别代表 a0 到 an。

Output

• 第一行输出一个整数t，表示有理数解的个数。
• 接下来t行，每行表示一个解，解以分数的形式输出，要求分子和分母互质，且分母必须是正整数特殊的，如果这个解是一个整数，那么直接把这个数输出。
• 等价的解只需要输出一次。
• 所有解按照从小到大的顺序输出。

Sample Input

• 3
  -24 14 29 6

Sample Output

• 3
  -4
  -3/2
  2/3

HINT

• 对于30%的数据，n<=10
• 对于100%的数据，n <= 100，|a i| <= 2*10^7，an≠ 0

---

Solution

• 原方程：$a_1 x + a_2 x ^ 2 + ... + a_n x ^ n = 0$。

• 因为 $x$ 是有理数，设 $x = \frac{p}{q}(p,q均为整数且互质)$，则：
  

  $$\begin{align} a_0 + a_1 \frac{p}{q} + a_2 \frac{p^2}{q^2} + ... + a_n \frac{p^n}{q ^n} &= 0\\ (同乘q^n)\ a_0 q^n + a_1 p q^{n - 1} + a_2 p^2 q^{n - 2} + ... + a_n p^n &= 0\\ \sum \limits_{i = 0}^{n}a_i p^i q^{n - i} &= 0\\ \end{align}$$

• 通过移项，我们可以得到：
  

  $$\begin{align} a_0 q^n &= -a_1 p q^{n - 1} - a_2 p^2 q^{n - 2} - ... - a_n p^n\\ (1)\ a_0 q^n &= p(-a_1 q^{n - 1} - a_2 p q^{n - 2} - ... - a_n p^{n - 1})\\ a_n p^n &= -a_0 q^n - a_1 p q^{n - 1} - a_2 p^2 q^{n - 2} - ... - a_{n-1}p^{n - 1}q\\ (2)\ a_n p^n &= q(-a_0 q^{n - 1} - a_1 p q^{n - 2} - a_2 p^2 q^{n - 3} - a_{n - 1}p^{n - 1})\\ \end{align}$$

• 因为 $p, q, a_i$ 均为整数且 $p, q$ 互质，通过 $(1)(2)$ 我们可以得到：

  • $p$ 为 $a_0$ 的约数，$q$ 为 $a_n$ 的约数（这里的约数包括负数，如 $3$ 的约数为 $\{1, 3, -1, -3\}$）
• 考虑到 $a_i \le 2 \times 10^7$，约数个数不是很多，我们直接找出 $a_0$ 的所有约数作为 $p$，$a_n$ 的所有约数作为 $q$，暴力枚举即可。

• 对于判断一组 $p, q$ 是否合法，我们根据上面的推论：
  

  $$\begin{align} a_0 + a_1 \frac{p}{q} + a_2 \frac{p^2}{q^2} + ... + a_n \frac{p^n}{q ^n} &= 0\\ 即\ \frac{\sum \limits_{i = 0}^{n}a_i p^i q^{n - i}}{q^n} &= 0 (q \neq 0)\\ 所以\ \sum \limits_{i = 0}^{n}a_i p^i q^{n - i} &= 0 \end{align}$$

• 但用高精度的方法不仅麻烦而且容易超时，我们采用取模来避免高精度。

• 就是选用一个较大的质数（例如 $10^9 + 9$），从头到尾带取模运算。
• 如果最后答案为0，我们便可以近似地认为原式为0了（觉得不保险可以多选用几个质数同时检验）。
• 因为约数不可能分解出0，要注意特判一个解 $x = 0$。

• 最后还有一个小坑，若$a_0 = 0$，则由 $(1)$ 得：
  

  $$\begin{align} p(-a_1 q^{n - 1} - a_2 p q^{n - 2} - ... - a_n p^{n - 1}) &= 0(p \neq 0)\\ p = 0\ 或\ -a_1 q^{n - 1} - a_2 p q^{n - 2} - ... - a_n p^{n - 1} &= 0\\ 所以\ -a_1 q^{n - 1} - a_2 p q^{n - 2} - ... - a_n p^{n - 1} &= 0\\ a_1 q^{n - 1} + a_2 p q^{n - 2} + ... + a_n p^{n - 1} &= 0\\ \end{align}$$

• 如果只靠分解约数 $p$ 只能为0，而对于这个式子 $p, q$ 可能有其它解，因此我们还要加一个判断。

• 实际上我们会发现这是一个和原来情况类似的问题，只是少了一个项，且剩余的项被约去一个 $p$。
• 那么对于接下来的 $a_1 = 0, a_2 = 0, ...$ 也是一样的情况。
• 最终我们会得到：对于第一个不为0的系数 $a_k$，$p$ 为 $a_k$ 的约数，$q$ 为 $a_n$ 的约数，判断 $p, q$ 合法的式子为 $\sum \limits_{i = k}^n a_i p^{i - k} q^{n - i}$（幸亏题目中有 $a_n \neq 0$，不然也要搞这么麻烦）。

Code

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

typedef long long ll;
const int N = 605, M = 105; 
const int Mod = 1e9 + 9;

struct Ans
{
    int p, q;

    friend inline bool operator < (const Ans &x, const Ans &y)
    {
        return x.p < y.p || x.p == y.p && x.q < y.q;
    }

    inline bool operator != (const Ans &x)
    {
        return p != x.p || q != x.q;
    }
}g[N * N], f[N * N];

inline int Gcd(int x, int y)
{
    int r = x % y;
    while (r) x = y, y = r, r = x % y;
    return y;
}

inline bool cmp(const Ans &x, const Ans &y)
{
    ll tmp = x.q * y.q / Gcd(x.q, y.q);
    return tmp * x.p / x.q < tmp * y.p / y.q;
}

int a[N], b[N], c[N], p[M], q[M];
int sum, n, G, F;

inline void Change(int x, int *b)
{
    int y = x < 0 ? -x : x; 
    for (int i = 1, im = ceil(sqrt(y)); i <= im; ++i)
        if (y % i == 0)
            b[++b[0]] = i, b[++b[0]] = x / i;
    for (int i = 1; i <= b[0]; ++i)
        b[i + b[0]] = -b[i]; b[0] <<= 1;
}

int main()
{
//  freopen("akai.in", "r", stdin);
//  freopen("akai.out", "w", stdout);

    scanf("%d%d", &n, &a[0]);
    for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);

    int len = 0; while (!a[len]) ++len;
    Change(a[len], b); Change(a[n], c); 

    for (int i = 1; i <= b[0]; ++i)
        for (int j = 1; j <= c[0]; ++j)
        {
            ++G; 
            int tmp = Gcd(b[i], c[j]);
            g[G].p = b[i] / tmp; 
            g[G].q = c[j] / tmp;
            if (g[G].p > 0 && g[G].q < 0)
                g[G].p = -g[G].p, g[G].q = -g[G].q;
        }

    sort(g + 1, g + G + 1);
    for (int i = 1; i <= G; ++i)
        if (g[i] != g[i - 1]) f[++F] = g[i];

    p[0] = q[0] = 1; G = 0;
    for (int i = 1; i <= F; ++i)
    {
        for (int k = 1; k <= n; ++k)
        {
            p[k] = (ll)p[k - 1] * f[i].p % Mod; 
            q[k] = (ll)q[k - 1] * f[i].q % Mod;
        }

        sum = 0;
        for (int k = len; k <= n; ++k)
            sum = ((ll)a[k] * p[k - len] % Mod * q[n - k] % Mod + sum) % Mod;
        if (!sum) g[++G] = f[i];
    }

    if (!a[0]) g[++G].p = 0, g[G].q = 1;
    sort(g + 1, g + G + 1, cmp);

    printf("%d\n", G);
    for (int i = 1; i <= G; ++i)
        if (g[i].q == 1 || g[i].p == 0) printf("%d\n", g[i].p);
            else printf("%d/%d\n", g[i].p, g[i].q);

//  fclose(stdin); fclose(stdout);
    return 0;
}