[Codeforces] number theory (R1900) Part.1

[Codeforces] number theory (R1900) Part.1

题单:https://codeforces.com/problemset/page/1?tags=number%20theory,1601-1900

7C. Line

原题指路:https://codeforces.com/problemset/problem/7/C

题意

给定三个整数$A,B,C(-2\mathrm{e}9\le A,B,C\le 2\mathrm{e}9,A^2+B^2>0)$.在直线$Ax+By+C$上求一个整点$(x,y)s.t.x,y\in [-5\mathrm{e}18,5\mathrm{e}18]$,若存在则输出任一满足条件的整点;否则输出$-1$.

思路

即求不定方程$Ax+By=-C$的一组在$[-5\mathrm{e}18,5\mathrm{e}18]$范围内的整数解,exgcd求即可.

exgcd求出的解必在要求的范围内.

[证] 如上图,显然无解的情况只能是线段$AB$上只有$A$和$B$两个整点,且整点都落在要求的范围之外.

设点$A$与$B$的横坐标、纵坐标之差分别为$\Delta x$和$\Delta y$,则线段$AB$上的整点数为$\gcd (\Delta x,\Delta y)+1$.

为使得要求的范围内无整点,需$\gcd (\Delta x,\Delta y)=1$.

​ ①$\Delta x=1$时,$\Delta y>1\mathrm{e}19$,则斜率$|k|>1\mathrm{e}19$.

​ ②$\Delta y=1$时,$\Delta x>1\mathrm{e}19$,则斜率$|k|<\frac{1}{1\mathrm{e}19}$.

对题目中的直线$Ax+By+C=0$,

​ ①若$A=0$或$B=0$,则直线是垂直于坐标轴轴的直线,显然此时要么无整点,要么有很多在范围内的整点.

​ ②若$B\ne 0$,则直线斜率的绝对值$|k|=\left|\frac{B}{A}\right|\le \frac{2\mathrm{e}9}{1}=2\mathrm{e}9<1\mathrm{e}19,|k|\ge \frac{1}{4\mathrm{e}9}>\frac{1}{\mathrm{e}19}$.

综上,若有解,则exgcd求出的解必在要求的范围内.

代码

ll exgcd(ll a, ll b, ll& x, ll& y) {  // 扩欧
  if (!b) {
    x = 1, y = 0;
    return a;
  }

  ll d = exgcd(b, a % b, y, x);
  y -= a / b * x;
  return d;
}

void solve() {
  ll a, b, c; cin >> a >> b >> c;

  ll x, y;
  ll d = exgcd(a, b, x, y);
  if (c % d) cout << -1;
  else {
    c *= -1;
    x *= c / d, y *= c / d;
    cout << x << ' ' << y;
  }
}

int main() {
  solve();
}

---

---

16C. Monitor

原题指路:https://codeforces.com/problemset/problem/16/C

题意 $(0.5\mathrm{s})$

给定一个长为$a$、宽为$b$的矩形,缩小其长或宽直至长宽比为$x:y$.若有解,则输出缩小后面积最大的矩形对应的长和宽,否则输出"0 0".

第一行输入四个整数$a,b,x,y(1\le a,b,x,y\le 2\mathrm{e}9)$.

思路

显然将$\frac{x}{y}$约分不影响结果,下面讨论中已将其约分.

若初始时$a<x$或$b<y$,显然无解;否则将$\frac{x}{y}$放大$k=\min \left\{\frac{a}{x},\frac{b}{y}\right\}$倍即得最大面积.

代码

void solve() {
  int a, b, x, y; cin >> a >> b >> x >> y;

  int d = gcd(x, y);
  x /= d, y /= d;

  if (a < x || b < y) {
    cout << "0 0";
    return;
  }

  int k = min(a / x, b / y);
  cout << x * k << ' ' << k * y;
}

int main() {
  solve();
}

---

---

66D. Petya and His Friends

原题指路:https://codeforces.com/problemset/problem/66/D

题意 $(2\mathrm{s})$

构造一个长度为$n(2\le n\le 50)$的正整数序列,满足如下三个要求:①对$\forall i,j\in [1,n]$,由$\gcd (a_i,a_j)\ne 1$;②所有元素的$\gcd =1$;③所有元素两两相异.若有解,输出任一组解;否则输出$-1$.

思路I

(1)$n=2$时,显然无解.

(2)$n\ge 3$时,取前三个数为$ab,ac,bc$,如$6,15,10$,则它们的$\gcd =1$,剩下的数只需保证不与前三个数互素,显然可取$10$的倍数.

代码I

void solve() {
  int n; cin >> n;

  if (n == 2) {
    cout << -1;
    return;
  }

  cout << 6 << endl;
  cout << 15 << endl;
  for (int i = 1; i <= n - 2; i++) cout << 10 * i << endl;
}

int main() {
  solve();
}

---

思路II

预处理出前$50$个素数$p_1,\cdots ,p_{50}$,令$A=\prod _{i=1}^{50}{p}_i$,取$\frac{A}{p_1},\cdots ,\frac{A}{p_{50}}$即可.本做法需要高精度.

---

---

111B. Petya and Divisors

原题指路:https://codeforces.com/problemset/problem/111/B

题意 $(5\mathrm{s})$

给定$n$个询问$x_i{y}_i$,对每个询问,求$x_i$有多少个约数不整除$x_{i-y_i},x_{i-y_i+1},\cdots ,x_{i-1}$,初始时$y_i=0$.

第一行输入一个整数$n(1\le n\le 1\mathrm{e}5)$.接下来$n$行每行输入两个整数$x_i,y_i(1\le x_i\le 1\mathrm{e}5,0\le y_i\le i-1,1\le i\le n)$.

思路

用map记录每个约数最后出现的位置即可.

代码

umap<int, int> mp;  // 记录每个约数最后出现的位置

bool insert(int pos, int lim, int d) {  // 记录约数最后出现的位置,返回是否合法
  if (mp[d] >= pos - lim) {  // 约数在范围内出现过
    mp[d] = pos;  // 更新约数最后出现的位置
    return false;
  }

  mp[d] = pos;  // 更新约数最后出现的位置
  return true;
}

void solve() {
  int n; cin >> n;
  for (int i = 1; i <= n; i++) {
    int x, y; cin >> x >> y;

    int ans = 0;
    for (int d = 1; (ll)d * d <= x; d++) {
      if (x % d == 0) {
        ans += insert(i, y, d);
        if (d != x / d) ans += insert(i, y, x / d);
      }
    }
    cout << ans << endl;
  }
}

int main() {
  solve();
}

---

---

117B. Very Interesting Game

原题指路:https://codeforces.com/problemset/problem/117/B

题意 $(2\mathrm{s})$

给定三个整数$a,b,c(0\le a,b\le 1\mathrm{e}9,1\le c\le 1\mathrm{e}7)$.A写一个$9$位的十进制数$s_1$(允许有前导零,下同)使得它不超过$a$;B写一个$9$位的十进制数$s_2$使得它不超过$b$.若$s_1$与$s_2$拼接而成的数能被$c$整除则B胜;否则A胜.两人都采取最优策略,若B胜,输出$2$;否则先输出$1$,再输出A可以选择的字典序最小的$s_1$.

思路

设A选择数$x\le a$,考察$r=10^9\cdot x\mathrm{mod}c$.遍历所有$x$,若$c-r\le b$,则B必胜.显然至多需遍历$c$次.

代码

const int BASE = 1e9;

void solve() {
  int a, b, c; cin >> a >> b >> c;

  for (int x = 0; x <= min(a, c); x++) {  // 枚举A选择的数
    int r = ((-(ll)BASE * x) % c + c) % c;
    if (0 <= r && r <= b) continue;

    printf("1 %09d", x);
    return;
  }
  printf("2");
}

int main() {
  solve();
}

---

---

121C. Lucky Permutation

原题指路:https://codeforces.com/problemset/problem/121/C

题意 $(2\mathrm{s})$

只包含数码$4$和$7$的数字是好的.对$1\sim n(1\le n\le 1\mathrm{e}9)$的所有全排列,求字典序第$k(1\le k\le 1\mathrm{e}9)$小的全排列中有多少个好的数字的下标也是好的,排列的下标从$1$开始.若不存在第$k$小的全排列,输出$-1$.

思路

不存在字典序第$k$小的全排列当且仅当$k>n!$,而$13!>1\mathrm{e}9$,故可暴力检查$k$是否$\le n!$,若否则无解.$n\ge 13$时必有解.

预处理出$1\mathrm{e}9$内的好的数字.因$13!>1\mathrm{e}9$,则字典序前$k$小的全排列中至多后$13$位改变,且这后$13$位是$1\sim n$中最大的$13$个数,用逆Cantor展开还原序列即可.

代码

const int MAXN = 15;
int fac[MAXN];
vi res;  // 将1~n中最大的lim个数加入
int ans;

void dfs(ll cur, int lim) {  // 预处理[1,lim]范围内好的数的个数
  if (cur > lim) return;

  ans += cur != 0;  // 去掉初始的0
  dfs(cur * 10 + 4, lim), dfs(cur * 10 + 7, lim);
}

int find(int k) {  // 求剩下的元素中的第(k+1)小的元素
  sort(all(res));
  int tmp = res[k];
  res.erase(res.begin() + k);
  return tmp;
}

bool check(int n) {  // 判断一个数是否是好的
  string s = to_string(n);
  for (auto ch : s)
    if (ch != '4' && ch != '7') return false;
  return true;
}

void solve() {
  int n, k; cin >> n >> k;
  k--;  // 逆Cantor展开第一步的-1

  int lim = -1;  // >k的最小的阶乘
  fac[0] = 1;
  for (int i = 1; i <= n; i++) {
    fac[i] = i * fac[i - 1];
    res.push_back(n - i + 1);  // 将1~n中最大的lim个数加入集合

    if (fac[i] > k) {
      lim = i;
      break;
    }
  }

  if (lim == -1) {  // k>n!
    cout << -1;
    return;
  }

  dfs(0, n - lim);

  for (int i = lim; i >= 1; i--) {  // 逆Cantor展开还原后lim个数
    int x = find(k / fac[i - 1]), pos = n - i + 1;  // 全排列中倒数第i个数及其下标
    if (check(x) && check(pos)) ans++;
    k %= fac[i - 1];
  }
  cout << ans;
}

int main() {
  solve();
}

---

---

134B. Pairs of Numbers

原题指路:https://codeforces.com/problemset/problem/134/B

题意

初始时给定一个数对$(a,b)$,它可产生两个数对$(a+b,b)$或$(a,a+b)$.给定一个整数$n(1\le n\le 1\mathrm{e}6)$,问从$(1,1)$开始至少需要多少步可使得数对中出现至少一个$n$.

思路

首先一定有解,至少可以做$n$次$+1$.

考虑DFS,显然直接DFS会TLE.考虑从结果开始DFS,当$a=b=1$时更新答案.

代码I

int ans = INF;

void dfs(int a, int b, int step) {  // 当前数对为(a,b),步数为step
  if (a < 1 || b < 1 || step >= ans) return;  // 非法解或已不是最优解
  if (a == 1 && b == 1) ans = min(ans, step);  // 找到合法解

  if (a - b > 0) dfs(a - b, b, step + 1);
  if (b - a > 0) dfs(a, b - a, step + 1);
}

void solve() {
  int n; cin >> n;

  for (int i = 1; i <= n; i++) dfs(i, n, 0);
  cout << ans;
}

int main() {
  solve();
}

---

代码II

int ans = INF;

int dfs(int a, int b) {  // 当前数对为(a,b)
  if (a == 1 && b == 1) return 0;

  if (a > b) swap(a, b);  // 保证a≤b
  if (a < 1 || b < 1 || a == b) return INF;  // 非法解
  return (b - 1) / a + dfs((b - 1) % a + 1, a);
}

void solve() {
  int n; cin >> n;

  for (int i = 1; i <= n; i++) ans = min(ans, dfs(i, n));
  cout << ans;
}

int main() {
  solve();
}

---

---

222C. Reducing Fractions

原题指路:https://codeforces.com/problemset/problem/222/C

题意 $(2\mathrm{s})$

给定两个集合$S_1,S_2$,其中$S_1$包含$n(1\le n\le 1\mathrm{e}5)$个元素$a_1,\cdots ,a_n(1\le a_i\le 1\mathrm{e}7)$,$S_2$包含$m(1\le m\le 1\mathrm{e}5)$个元素$b_1,\cdots ,b_m(1\le b_i\le 1\mathrm{e}7)$.$S_1$中的元素之积为分数的分子,$S_2$中的元素之积为分母.求一个两个分别于$S_1,S_2$等价的集合$T_1,T_2$,使得$T_1$中的元素之积为分子,$T_2$中的元素之积为分母,且得到的结果为既约分数.

思路

模拟约分过程即可.

注意分别对每个数做素因数分解,时间复杂度 O ( max ⁡ { n , m } max ⁡ { a , b } ) O\left(\max\{n,m\}\sqrt{\max\{a,b\}}\right) O(max{n,m}max{a,b} ​),会TLE.

可用线性筛或埃筛预处理出每个数的最小素因子,素因数分解时每次除以其最小素因子,时间复杂度 O ( max ⁡ { n , m } log ⁡ max ⁡ { a , b } ) O\left(\max\{n,m\}\log{\max\{a,b\}}\right) O(max{n,m}logmax{a,b}).

代码

const int MAXN = 1e7 + 5;
int n, m;
int primes[MAXN];  // 记录每个数的最小素因子
int up[MAXN], down[MAXN];  // 分子、分母中每个素因子的次数

void init() {  // 预处理出每个数的最小素因子
  for (int i = 2; i < MAXN; i++) {
    if (!primes[i]) {
      primes[i] = i;

      for (int j = 2 * i; j < MAXN; j += i) primes[j] = i;
    }
  }
}

void solve() {
  init();

  cin >> n >> m;
  vi a(n);
  for (auto& ai : a) {
    cin >> ai;
    for (int j = ai; j > 1; j /= primes[j]) up[primes[j]]++;  // 素因数分解
  }
  vi b(m);
  for (auto& bi : b) {
    cin >> bi;
    for (int j = bi; j > 1; j /= primes[j]) down[primes[j]]++;  // 素因数分解
  }

  cout << n << ' ' << m << endl;
  for (auto ai : a) {
    int res = 1;
    for (int j = ai; j > 1; j /= primes[j]) {
      if (down[primes[j]]) down[primes[j]]--;  // 可以约分
      else res *= primes[j];  // 不能约分
    }
    cout << res << ' ';
  }
  cout << endl;
  for (auto bi : b) {
    int res = 1;
    for (int j = bi; j > 1; j /= primes[j]) {
      if (up[primes[j]]) up[primes[j]]--;  // 可以约分
      else res *= primes[j];  // 不能约分
    }
    cout << res << ' ';
  }
}

int main() {
  solve();
}

---

---

223C. Partial Sums

原题指路:https://codeforces.com/problemset/problem/223/C

题意 $(4\mathrm{s})$

给定一个长度为$n$的序列$a_1,\cdots ,a_n$.现有操作,分为如下三步:①求$a[]$的前缀和数组$pre[]$,即$pre[1]=a[1],pre[i]=pre[i-1]+a[i](i\ge 2)$;②将$pre[]$的元素都对$1\mathrm{e}9+7$取模;③将$pre[]$作为新的$a[]$.问$k$次操作后得到的序列$a[]$.

第一行输入两个整数$n,k(1\le n\le 2000,0\le k\le 1\mathrm{e}9)$.第二行输入$n$各整数$a_1,\cdots ,a_n(0\le a_i\le 1\mathrm{e}9)$.

思路I

设$B_{ij}=[i\ge j]$,则$pr{e}_i=\sum _{j=1}^n{B}_{ij}{a}_j$.将$a[]$和$pre[]$视为列向量,则每次操作等价于$\overrightarrow{a}$左乘转移矩阵$B=(B_{ij}{)}_{n\times n}$.用矩阵快速幂求,时间复杂度$O(n^3\log k)$,会TLE.

考虑优化,注意到$(B^k{)}_{i,j}=(B^k{)}_{i-j+1,1}$,即矩阵$B^k$中与主对角线平行的对角线上的元素都相等,这可用数学归纳法证明.这表明:用序列$(b^k{)}_i=(B^k{)}_{i,1}$即可确定矩阵$B^k$,即矩阵$B^k$第一列的元素.矩阵$B^k{B}^l$的第一列的元素$(b^{k+l})=\sum _{j=1}^i(b^k{)}_j(b^l{)}_{i-j+1}$.用矩阵快速幂求,时间复杂度$O(n^2\log k)$,可过.

事实上本题还可进一步优化.下面用数学归纳法证明$(b^k{)}_i=C_{k+i-1}^{k-1}$.

[证] $(b^{k+1}{)}_i=\sum _{j=1}^k{C}_{k+j-1}^{k-1}=C_k^k+C_k^{k-1}+\sum _{j=3}^i{C}_{k+j-1}^{k-1}=C_k^{k+1}+C_{k+1}^{k-1}+\sum _{j=4}^i{C}_{k+j-1}^k=\cdots =C_{k+i}^k$.

注意到$\frac{(b^k{)}_{i+1}}{(b^k{)}_i}=\frac{(k+i)}{(i+1)}$,故$(b^k{)}_i=\frac{k+i-1}{i}(b^k{)}_{i-1}$,递推求即可.

另一做法:$pr{e}_i^k=\sum _{j=1}^i{C}_{k+i-j}^{i-j}{a}_j$.因$k$的范围较大,无法预处理出所有组合数.注意到$i-j\le n\le 2000$,可只处理出所有$(i-j)$可能的取值对应的组合数,总时间复杂度$O(n^2)$.

代码

const int MAXN = 2005;
const int MOD = 1e9 + 7;
int n, k;
int a[MAXN];
int inv[MAXN];
int C[MAXN];  // C[i-j]表示组合数C(k+i-j,i-j)

void init() {
  inv[1] = 1;
  for (int i = 2; i < MAXN; i++) inv[i] = (ll)(MOD - MOD / i) * inv[MOD % i] % MOD;

  C[0] = C[1] = 1;
  for (int i = 1; i <= n; C[++i] = 1)
    for (int j = 1; j <= i; j++) C[i] = (ll)C[i] * (k + j - 1) % MOD * inv[j] % MOD;
}

void solve() {
  cin >> n >> k;
  for (int i = 1; i <= n; i++) cin >> a[i];

  init();

  for (int i = 1, ans = 0; i <= n; i++, ans = 0) {
    for (int j = 1; j <= i; j++) ans = ((ll)ans + (ll)C[i - j] * a[j] % MOD) % MOD;
    cout << ans << ' ';
  }
}

int main() {
  solve();
}

---

思路II

[引理] 对形式幂级数$A(x)$和$p\in \mathbb{R}$,若$F(x)=[A(x){]}^p$,则$A(x)F^{\prime }(x)=pF(x)A^{\prime }(x)$.

[证] $\ln F(x)=p\ln A(x)$,两边对$x$求导,移项即证.

对序列$a_0,\cdots ,a_{n-1}$,设其生成函数为$A(x)$,则求 { a n } \{a_n\} {an​}的前缀和 { b n } \{b_n\} {bn​}等价于给$A(x)$乘上$\frac{1}{1-x}$.

[证] $A(x)$的常数项$a_0$求前缀和得到$b_0=a_0$,即$A(x)$乘上$1$.

$A(x)$的一次项$a_1$求前缀和得到$b_1=a_0+a_1$,即$A(x)$乘上$x$.

同理易得求前缀和的过程即$A(x)$乘上$1+x+x^2+\cdots =\frac{1}{1-x}(|x|<1)$.

设$F(x)=\sum _n^{\infty }{f}_n{x}^n$,则$F(x)=\frac{1}{(1-x{)}^k}$.

由引理:$(1-x)F^{\prime }(x)=-kF(x)\cdot (-1)=kF(x)$,即$\sum _{n=0}^{\infty }(n+1)f_{n+1}{x}^n-\sum _{n=1}^{\infty }n{f}_n{x}^n=k\sum _{n=0}^{\infty }{f}_n{x}^n$.

则$\{\begin{array}{l}{f}_0=F(0)=1\\ f_{0+1}=F(0+1)=k{f}_0\Rightarrow f_1=k\\ (n+1)f_{n+1}-n{f}_n=k{f}_n(n\ge 1)\end{array}$,故$f_n=\frac{n+k-1}{n}{f}_{n-1}$,由此可见$f_n$是组合数.

$O(n)$递推出 { f n } \{f_n\} {fn​}后$O(n^2)$地求$A(x)$与$F(x)$的卷积即可.因$n\le 2000$,暴力求即可.

---

---

279E. Beautiful Decomposition

原题指路:https://codeforces.com/problemset/problem/279/E

题意 $(2\mathrm{s})$

给定一个长度不超过$1\mathrm{e}6$的不含前导零的二进制数$n$,求至少要用多少个形如$2^k$或$-2^k$的数使得它们之和为$n$.

思路I

每个$2$的幂次至多使用一次.

[证] ①加两次$2^k$相当于加一次$2^{k+1}$;②减两次$2^k$相当于减一次$2^{k+1}$;③加一次并减一次$2^k$相当于无操作.

设$n$的最高位的权值为$2^k$.注意到$2^0+\cdots +2^{k-1}<2^k$,则为使得$s$的最高位为$1$,需加上某个$2^x(x\ge k)$.注意到加上$x\ge k+2$不是最优的,如加上$2^{k+2}$后还需减去一个$2^{k+1}$,不如只加上一个$2^{k+1}$.故$x=k$或$k+1$.

①若选择加上$2^k$,则需填补后面的数位.

②若加上$2^{k+1}$,则需填补$m=2^{k+1}-n$后面的数位,即将$n$从该位开始的后缀的数码翻转.

​ 显然求$m$无需确定$k$值,直接翻转初始时的$n$各数码即可.

设$v[0]=n,v[1]=m,dp[i][j]$表示构造$v[i]$从下标$j$开始的后缀所需的最少$2$的幂次的个数.

下面字符串下标从$1$开始,若$n[i]=0$,则$m[i]=1$.

状态转移方程:

①$n[i]=0$时, d p [ 0 ] [ i ] = d p [ 0 ] [ i − 1 ] , d [ 1 ] [ i ] = min ⁡ { d p [ 0 ] [ i − 1 ] , d p [ 1 ] [ i − 1 ] } + 1 dp[0][i]=dp[0][i-1],d[1][i]=\min\{dp[0][i-1],dp[1][i-1]\}+1 dp[0][i]=dp[0][i−1],d[1][i]=min{dp[0][i−1],dp[1][i−1]}+1.

②$n[i]=1$时, d p [ 1 ] [ i ] = d p [ 1 ] [ i − 1 ] , d p [ 0 ] [ i ] = min ⁡ { d p [ 0 ] [ i − 1 ] , d p [ 1 ] [ i − 1 ] } + 1 dp[1][i]=dp[1][i-1],dp[0][i]=\min\{dp[0][i-1],dp[1][i-1]\}+1 dp[1][i]=dp[1][i−1],dp[0][i]=min{dp[0][i−1],dp[1][i−1]}+1.

初始条件$dp[0][1]=[n[1]=1],dp[1][1]=1$,后者是因为翻转整个串需要一个$2$的幂次.

代码I

const int MAXN = 1e6 + 5;
char s[MAXN];
int dp[2][MAXN];  // v[0]=n,v[1]=m,dp[i][j]表示构造v[i]从下标j开始的后缀所需的最少2的幂次的个数

void solve() {
  cin >> s + 1;

  int n = strlen(s + 1);
  dp[0][1] = s[1] == '1', dp[1][1] = 1;  // 初始条件
  for (int i = 2; i <= n; i++) {
    if (s[i] == '0') {
      dp[0][i] = dp[0][i - 1];
      dp[1][i] = min(dp[0][i - 1], dp[1][i - 1]) + 1;
    }
    else {
      dp[1][i] = dp[1][i - 1];
      dp[0][i] = min(dp[0][i - 1], dp[1][i - 1]) + 1;
    }
  }
  cout << dp[0][n];
}

int main() {
  solve();
}

---

---