[Codeforces] combinatorics (R1600) Part.4

[Codeforces] combinatorics (R1600) Part.4

题单:https://codeforces.com/problemset?tags=combinatorics,1201-1600

735C. Tennis Championship

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

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

求一个包含$n(2\le n\le 1\mathrm{e}18)$个节点的、每个节点的两棵子树高度之差不超过$1$的二叉树的最大高度(高度从$0$开始).

思路

$f(h)$表示高度为$h$的满足条件的二叉树所包含的最小节点数.考虑其根节点的两棵子树,它们的高度只能为$(h-1)$或$(h-2)$,则$f(h)=f(h-1)+f(h-2)$.

设$S_n=\sum _{i=1}^{n}f(i)$,问题转化为求最小的$anss.t.S_{ans}\ge n$.

代码

vl fib, ans;

void init() {
  fib.push_back(1), fib.push_back(1);
  for (int i = 2; fib.back() < 1e18; i++)
    fib.push_back(fib[fib.size() - 2] + fib[fib.size() - 1]);

  ans.push_back(0);
  for (int i = 0; ans.back() < 1e18; i++)
    ans.push_back(fib[i] + ans.back());
}

void solve() {
  init();

  ll n; cin >> n;
  cout << lower_bound(all(ans), n) - ans.begin() - 1;
}

int main() {
  solve();
}

---

---

804B. Minimum number of steps

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

题意

给定一个长度不超过$1\mathrm{e}6$的只包含字符’a’和’b’的字符串$s$.现有操作:选择$s$中的一个子串"ab",将其替换为"bba",无法操作时停止.求最小操作次数,答案对$1\mathrm{e}9+7$取模.

思路

注意到操作一次会减少一对’a’在’b’之前的字符对,同时使得’b’的个数翻倍,则最终字符串形如$b\cdots ba\cdots a$.

注意到前面的子串"ab"操作一次后产生的’a’可能会与后面的’b’构成子串"ab",故可从后往前考虑,遇到一个字符’b’时令其$cnt++$;遇到一个字符’a’时$ans+=cnt,cnt\ast =2$.

代码

const int MOD = 1e9 + 7;

void solve() {
  string s; cin >> s;

  reverse(all(s));
  int cnt = 0;  // 字符'b'的个数
  int ans=0;
  for (auto ch : s) {
    if (ch == 'b') cnt++;
    else {
      ans = ((ll)ans + cnt) % MOD;
      cnt = (ll)cnt * 2 % MOD;
    }
  }
  cout << ans;
}

int main() {
  solve();
}

---

---

817B. Makes And The Product

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

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

给定一个长度为$n(3\le n\le 1\mathrm{e}5)$的整数序列$a_1,\cdots ,a_n(1\le a_i\le 1\mathrm{e}9)$.求有多少对$(i,j,k)s.t.i<j<k,a_i\cdot a_j\cdot a_k$最小.

思路

显然应贪心地选$a[]$中前三小的元素,不妨设为$x,y,z$,它们的出现次数分别为$cn{t}_1,cn{t}_2,cn{t}_3$.

①$x\ne y\ne z$时,$ans=cn{t}_1\cdot cn{t}_2\cdot cn{t}_3$.

②$x=y=z$时,$ans=C_{cn{t}_1}^3$.

③$x=y\ne z$时,$ans=C_{cn{t}_1}^2\cdot cn{t}_3$.

④$x\ne y=z$时,$ans=cn{t}_1\cdot C_{cn{t}_2}^2$.

代码

map<int, int> cnt;

ll C(int n, int m) {  // 组合数C(n,m)
  ll res = 1;
  for (int i = n - m + 1; i <= n; i++) res = res * i / (i - n + m);
  return res;
}

void solve() {
  int n; cin >> n;
  vi a(n);
  for (auto& ai : a) {
    cin >> ai;
    cnt[ai]++;
  }

  sort(all(a));
  if (a[0] != a[1] && a[1] != a[2]) cout << (ll)cnt[a[0]] * cnt[a[1]] * cnt[a[2]];
  else if (a[0] == a[1] && a[1] == a[2]) cout << C(cnt[a[0]], 3);
  else if (a[0] == a[1] && a[1] != a[2]) cout << C(cnt[a[0]], 2) * cnt[a[2]];
  else cout << C(cnt[a[1]], 2) * cnt[a[0]];
}

int main() {
    solve();
}

---

---

840A. Leha and Function

原题指路:https://codeforces.com/problemset/problem/840/A

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

考察从$n$个元素$[1,\cdots ,n]$中选$k$个元素构成一个子集,定义$f(n,k)$表示所有子集中最小元素的期望.

给定两长度为$n(1\le n\le 2\mathrm{e}5)$的序列$a_1,\cdots ,a_n$和$b_1,\cdots ,b_n(1\le a_i,b_i\le 1\mathrm{e}9)$.将序列$a[]$重排为序列$a^{\prime }[]s.t.\sum _{i=1}^{n}f(a_i,b_i)$最大,输出任一满足条件的序列$a^{\prime }[]$.

思路

从$n$个元素中选$k$个元素的方案数为$C_n^k$.枚举$i$,为使得其是子集中最小的元素,应在比它大的$(n-i)$个元素中选$(k-1)$个元素,故$f(n,k)=\frac{\sum _{i=1}^{n-k+1}i\cdot C_{n-i}^{k-1}}{C_n^k}$.

注意到$\frac{f(n,k)}{f(n,k-1)}=\frac{k}{k+1}$,则$f(n,k)=\frac{n+1}{k+1}$.为使得$f(n,k)$尽量大,应让较大的$n$与较小的$k$配对.

将序列$a[]$降序排列、$b[]$升序排列后配对即可.

代码

void solve() {
  int n; cin >> n;
  vi a(n);
  vii b(n);
  for (auto& ai : a) cin >> ai;
  for (int i = 0; i < n; i++) {
    cin >> b[i].first;
    b[i].second = i;
  }

  sort(all(a), greater<int>()), sort(all(b));

  vi ans(n);
  for (int i = 0; i < n; i++) ans[b[i].second] = a[i];
  for (int i = 0; i < n; i++) cout << ans[i] << " \n"[i == n - 1];
}

int main() {
  solve();
}

---

---

844B. Rectangles

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

题意

给定一个$n\times m(1\le n,m\le 50)$的$0-1$矩阵.将元素分为若干个非空集合,使得集合内的数字相等,且任意两个集合中的元素同行或同列,求集合数.

思路

注意到符合要求的集合只能包含一行或一列,统计每行、每列的$0$、$1$的个数,不妨设第$i$行的$0$的个数为$ro{w}_0$,则该行对答案的贡献为$2^{ro{w}_0}-1$,即幂集减去空集.对每行和每列的答案求和,因单个元素在行和列都被计算了一次,答案减去$nm$即可.

代码

void solve() {
  int n, m; cin >> n >> m;
  vi row0(n), row1(n);  // 每行0、1的个数
  vi col0(m), col1(m);  // 每列0、1的个数
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      int a; cin >> a;
      if (a == 0) row0[i]++, col0[j]++;
      else row1[i]++, col1[j]++;
    }
  }

  ll ans = -n * m;
  for (int i = 0; i < n; i++) ans += ((ll)1 << row0[i]) - 1 + ((ll)1 << row1[i]) - 1;
  for (int i = 0; i < m; i++) ans += ((ll)1 << col0[i]) - 1 + ((ll)1 << col1[i]) - 1;
  cout << ans;
}

int main() {
  solve();
}

---

---

888D. Almost Identity Permutations

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

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

给定两整数$n,k$.称一个$1\sim n$排列$p$是好的,如果存在至少$(n-k)$个下标$i(1\le i\le n)s.t.p_i=i$.

给定两整数$n,k(1\le k\le 4\le n\le 1000)$,求好的排列的个数.

思路

考虑如何求恰有$m(1\le m\le k)$个下标$i(1\le i\le n)s.t.p_i\ne i$的排列的个数.先从$n$个数中选出$m$个数作为满足$p_i\ne i$的数的下标,有$C_n^m$种情况.将这$m$个数错排,共$f(m)$种情况,其中$f(n)$表示$1\sim n$的排列的错排数,可递推求得.故$ans=\sum _{m=1}^k{C}_n^{m}f(m)$.当$m=1$时,错排数$f(m)=0$,但事实上$k=1$时存在一种满足的方案$1,\cdots ,n$,故$ans=1+\sum _{m=2}^k{C}_n^{m}f(m)$.

代码

const int MAXN = 1005;
ll C[MAXN][MAXN];  // C[i][j]表示组合数C(i,j)
int f[MAXN];  // 错排数

void init() {  // 预处理C[][]和f[]
  for (int i = 0; i < MAXN; i++) {
    for (int j = 0; j <= i; j++) {
      if (!j) C[i][j] = 1;  // 初始条件
      else C[i][j] = C[i - 1][j] + C[i - 1][j - 1];
    }
  }

  f[1] = 0, f[2] = 1;
  for (int i = 3; i < 5; i++) f[i] = (i - 1) * (f[i - 1] + f[i - 2]);
}

void solve() {
  init();

  int n, k; cin >> n >> k;

  ll ans = 1;  // 特判i=1;
  for (int i = 2; i <= k; i++) ans += C[n][i] * f[i];
  cout << ans;
}

int main() {
  solve();
}

---

---

978E. Bus Video System

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

题意

某公交系统会记录每一站车上人数的变化.若上一站车上有$x$人,该站车上有$y$人,则记录$(y-x)$.有$n$个公交站,在每个公交站记录的数据分别是$a_1,\cdots ,a_n$.设车的最大容量为$w$人(无需考虑司机),问初始时车上的人数有几种可能,若无合法方案,输出$0$.

第一行输入两个整数$n,w(1\le n\le 1000,1\le w\le 1\mathrm{e}9)$.第二行输入$n$个整数$a_1,\cdots ,a_n(-1\mathrm{e}6\le a_i\le 1\mathrm{e}6)$.

思路

设初始时车上的人数为$x$.设$a[]$的前缀和为$pre[],maxd=\underset{1\le i\le n}{\max }pr{e}_i,mind=\underset{1\le i\le n}{\min }pr{e}_i$,则全程车上的最大人数为$x+maxd$,最小人数为$x+mind$.

由车的最大容量和人数非负知:$x+maxd\le w,x+mind\ge 0$,该不等式组的整数解个数 a n s = max ⁡ { w − m a x d + m i n d + 1 } ans=\max\{w-maxd+mind+1\} ans=max{w−maxd+mind+1}.

注意$maxd$和$mind$不能初始化为$-INF$和$INF$,因为车上的人数最小为$0$,都初始化为$0$即可.

代码

void solve() {
  int n, w; cin >> n >> w;
  int pre = 0;  // a[]的前缀和
  int maxd = 0, mind = 0;  // 当前pre[]的最大值、最小值,注意初始化
  while (n--) {
    int a; cin >> a;
    pre += a;
    maxd = max(maxd, pre), mind = min(mind, pre);
  }
  cout << max(0, w - maxd + mind + 1);
}

int main() {
  solve();
}

---

---

1007A. Reorder the Array

原题指路:https://codeforces.com/problemset/problem/1007/A

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

给定一个长度为$n(1\le n\le 1\mathrm{e}5)$的整数序列$a_1,\cdots ,a_n(1\le a_i\le 1\mathrm{e}9)$,将其重新排列为序列$b[]$.求一个序列$b[]s.t.\sum _{i=1}^n[b_i>a_i]$最大,输出最大值.

思路

将$a[]$升序排列,对每个$a_i$,考察其后有多少个$a_j(i>j)s.t.a_j>a_i$,答案即为所有$a_i$的结果之和,用双指针即可做到$O(n)$.

代码

void solve() {
  int n; cin >> n;
  vi a(n);
  for (auto& ai : a) cin >> ai;

  sort(all(a));
  int ans = 0;
  for (int i = 0, j = 1; i < n - 1 && j < n; i++, j++) {
    if (a[j] > a[i]) ans++;
    else i--;
  }
  cout << ans;
}

int main() {
  solve();
}

---

---

1081C. Colorful Bricks

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

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

用$m$种颜色给排成一行的$n$块砖染色,要求恰有$k$块砖与其左边的砖的颜色不同,求染色方案数,答案对$998244353$取模.

第一行输入三个整数$n,m,k(1\le n,m\le 2000,0\le k\le n-1)$.

思路I

$dp[i][j]$表示前$i$块砖中有$j$块砖与其左边的砖的颜色不同的方案数,则$ans=dp[n][k]$.

根据第$i$块砖与第$(i-1)$块砖颜色是否相同分类,状态转移方程$dp[i][j]=dp[i-1][j]+(m-1)\cdot dp[i-1][j-1]$.初始条件$dp[i][0]=m$,即$1\sim i$号砖都染同种颜色的方案数.

代码I

const int MAXN = 2005;
const int MOD = 998244353;
int n, m, k;  // 砖数、颜色数、不同数
int dp[MAXN][MAXN];  // dp[i][j]表示前i块砖中有j块砖与其左边的砖的颜色不同的方案数

void solve() {
  cin >> n >> m >> k;

  for (int i = 1; i <= n; i++) {
    dp[i][0] = m;
    for (int j = 1; j <= k; j++)
      dp[i][j] = ((ll)dp[i - 1][j] + (ll)(m - 1) * dp[i - 1][j - 1]) % MOD;
  }
  cout << dp[n][k];
}

int main() {
  solve();
}

---

思路II

将一段连续的相同颜色的砖视为一个部分,将序列分为$(k+1)$个部分,使得相邻颜色不同只出现在相邻两部分的交界处.第一个部分颜色有$m$种选择,后面每个部分与前面的部分颜色不同,故有$m(m-1{)}^k$种情况.

在长度为$n$的序列的$(n-1)$个空中选$k$个作为相邻两部分的分界点,有$C_{n-1}^k$种情况,故$ans=C_{n-1}^{k}m(m-1{)}^k$.

代码II

const int MAXN = 2005;
const int MOD = 998244353;
int n, m, k;  // 砖数、颜色数、不同数
int C[MAXN][MAXN];  // C[i][j]表示组合数C(i,j)

void init() {
  for (int i = 0; i < MAXN; i++) {
    for (int j = 0; j <= i; j++) {
      if (!j) C[i][j] = 1;  // 初始条件
      else C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % MOD;
    }
  }
}

void solve() {
  init();

  cin >> n >> m >> k;
  cout << (ll)C[n - 1][k] * m % MOD * qpow(m - 1, k, MOD) % MOD;
}

int main() {
  solve();
}

---

---

1084C. The Fair Nut and String

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

题意

给定一个长度不超过$1\mathrm{e}5$的只包含小写英文字母的字符串$s$,求满足如下条件的严格递增的整数序列$p_1,\cdots ,p_k$的个数,答案对$1\mathrm{e}9+7$取模:①$s_{p_i}=a(1\le i\le k)$;②对$\forall i\in [1,k),\exists js.t.p_i<j<p_{i+1}$且$s_j=b$.

思路

显然$s$中非’a’、'b’的字符不会对答案产生贡献.

显然满足条件的下标$p_i$和$j$对应的子列有形式$abab\cdots $.对每个’a’,找到其左边第一个’b’,则以该’b’结尾的所有序列后接上当前的’a’仍是合法序列.

用一个变量$last$记录上一次遇到’b’时求得的答案,初始时$last=0$.①遇到’a’时,$ans+=last+1$;②遇到’b’时,$last=ans$.

代码

const int MOD = 1e9 + 7;

void solve() {
  string s; cin >> s;

  int ans = 0;
  int last = 0;  // 上一次遇到'b'时求得的答案
  for (auto ch : s) {
    if (ch == 'a') ans = (ans + last + 1) % MOD;
    else if (ch == 'b') last = ans;
  }
  cout << ans;
}

int main() {
  solve();
}

---

---