【ACWing】1308. 方程的解

记录算法题解

于 2023-06-17 16:37:10 发布

阅读量152

点赞数

分类专栏： AC 数学文章标签：算法 c++ 动态规划

本文链接：https://blog.csdn.net/qq_46105170/article/details/131261494

版权

AC 数学专栏收录该内容

109 篇文章 7 订阅

订阅专栏

文章描述了一种处理不定方程a_1+a_2+...+a_k=g(x)，其中g(x)=x^xmod1000的问题。通过快速幂计算x^x模1000的结果，然后利用隔板法求解正整数解的个数，使用递推公式进行高精度计算。代码示例展示了如何实现这一过程，时间复杂度为O(logx+k)，空间复杂度为O(1)。

摘要由CSDN通过智能技术生成

题目地址：

https://www.acwing.com/problem/content/1310/

佳佳碰到了一个难题，请你来帮忙解决。对于不定方程 $a_1+a_2+⋯+a_{k−1}+a_k=g(x)$ ，其中 $k \geq 1$ 且 $k∈N^+$ ， $x$ 是正整数， $g(x)=x^x\mod1000$ （即 $x^x$ 除以 $1000$ 的余数）， $x, k$ 是给定的数。我们要求的是这个不定方程的正整数解组数。举例来说，当 $k = 3, x = 2$ 时，方程的解分别为： $\left\{ \begin{array}{l} a_1=1 \\ a_2=1 \\ a_3=2 \\ \end{array} \right.$ ， $\left\{ \begin{array}{l} a_1=1 \\ a_2=2 \\ a_3=1 \\ \end{array} \right.$ 和 $\left\{ \begin{array}{l} a_1=2 \\ a_2=1 \\ a_3=1 \\ \end{array} \right.$ 。

输入格式：
有且只有一行，为用空格隔开的两个正整数，依次为 $k, x$ 。

输出格式：
有且只有一行，为方程的正整数解组数。

数据范围：
$1 \leq k \leq 100$
$1≤x<2^{31}$
$k \leq g (x)$

先用快速幂求 $n=x^x\mod 1000$ ，接着问题转化为求 $a_1+...+a_k=n$ 的正整数解的个数，可以用隔板法求出答案为 $n-1\choose k-1$ 。组合数有递推公式 ${n\choose k}={n-1\choose k-1}+{n-1\choose k}$ 。令 $f[n][k]={n\choose k}, k\le n$ ，已知 $f [0] [0] = 1$ ，一路递推即可。最终答案大概是 ${1000\choose 100}$ 这个级别，需要高精度。代码如下：

#include <iostream>
using namespace std;

const int N = 150, mod = 1000;
int k, x;
int f[1000][100][N];

int qmi(int a, int b) {
  int res = 1;
  while (b) {
    if (b & 1) res = res * a % mod;
    b >>= 1;
    a = a * a % mod;
  }
  return res;
}

// 高精度 a = b + c
void add(int a[N], int b[N], int c[N]) {
  for (int i = 0, t = 0; i < N; ++i) {
    t += b[i] + c[i];
    a[i] = t % 10;
    t /= 10;
  }
}

int main() {
  scanf("%d%d", &k, &x);
  int n = qmi(x % mod, x);

  for (int i = 0; i < n; ++i) {
    for (int j = 0; j <= i && j < k; ++j) {
      if (!j) f[i][j][0] = 1;
      else add(f[i][j], f[i - 1][j], f[i - 1][j - 1]);
    }
  }

  // 不要打印开头0
  int m = N - 1;
  while (!f[n - 1][k - 1][m]) m--;
  while (m >= 0) printf("%d", f[n - 1][k - 1][m--]);
  puts("");
}