[Codeforces] combinatorics (R1600) Part.6

[Codeforces] combinatorics (R1600) Part.6

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

1326C. Permutation Partitions

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

题意

给定一个$1\sim n$的排列$p_1,\cdots ,p_n$,将其划分为不相交的$k$段区间,使得每一段的最大值之和最大,并求取得最大值的划分方案数,后者答案对$998244353$取模.设第$i(1\le i\le k)$段区间为$[l_i,r_i]$,则最大化$\sum _{i=1}^k\underset{l_i\le j\le r_i}{\max }{p}_j$.

第一行输入两个整数$n,k(1\le k\le n\le 2\mathrm{e}5)$.第二行输入一个$1\sim n$的排列$p_1,\cdots ,p_n$.

思路

为使得$\sum _{i=1}^k\underset{l_i\le j\le r_i}{\max }{p}_j$最大,显然应取$1\sim n$中的前$k$大的数,即将它们分在不同的$k$个区间.记录前$k$大的数的下标$pos[]$,则相邻两个前$k$大的数所在的区间的分割线可取中间任一位置,将$pos[]$升序排列后相邻两位置的下标差之积即为方案数.

代码

const int MOD = 998244353;

void solve() {
  int n, k; cin >> n >> k;
  vii a(n);
  for (int i = 0; i < n; i++) {
    int x; cin >> x;
    a[i] = { x,i };
  }

  sort(rall(a));

  vi pos(k);  // 前k大的数的下标
  ll ans1 = 0;
  for (int i = 0; i < k; i++) {
    ans1 = ans1 + a[i].first;
    pos[i] = a[i].second;
  }

  sort(all(pos));

  int ans2 = 1;
  for (int i = 1; i < k; i++) ans2 = (ll)ans2 * (pos[i] - pos[i - 1]) % MOD;
  cout << ans1 << ' ' << ans2;
}

int main() {
  solve();
}

---

---

1328B. K-th Beautiful String

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

题意

将所有包含$(n-2)$个’a’和$2$个’b’的字符串按字典序升序排列,求排名第$k$的字符串.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据输入两个整数$n,k\left(3\le n\le 1\mathrm{e}5,1\le k\le \min \left\{2\mathrm{e}9,\frac{n(n-1)}{2}\right\}\right)$.数据保证所有测试数据的$n$之和不超过$1\mathrm{e}5$.

思路

显然只需确定两个’b’所在的位置.先确定第一个’b’的位置,枚举下标$i\in [n-2,0]$,若$k\le n-i-1$,则当前位置是第一个’b’的位置,进而第二个’b’的下标为$(n-k)$;否则令$k-=n-i-1$,即跳过所有当前位置为’b’的字符串.

代码

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

  string ans(n, 'a');
  for (int i = n - 2; i >= 0; k -= (n - i - 1), i--) {  // 枚举第一个'b'的位置
    if (k <= (n - i - 1)) {
      ans[i] = ans[n - k] = 'b';
      cout << ans << endl;
      return;
    }
  }
}

int main() {
  CaseT  // 单测时注释掉该行
    solve();
}

---

---

1391C. Cyclic Permutations

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

题意

给定一个$1\sim n$的排列$p$,现按如下方式建立一张包含$n$个节点的无向图:①对$\forall i\in [1,n]$,找到最大的$js.t.1\le j<i$且$p_j>p_i$,添加一条从节点$i$到节点$j$的无向边;②对$\forall i\in [1,n]$,找到最小的$js.t.i<j\le n$且$p_j>p_i$,添加一条从节点$i$到节点$j$的无向边.称一个排列$p$是循环排列,如果它对应的图中至少存在一个简单环.给定整数$n(3\le n\le 1\mathrm{e}6)$,求$1\sim n$的排列中循环排列的个数,答案对$1\mathrm{e}9+7$取模.

思路

排列$p$对应的图至少存在一个简单环当且仅当$\exists i<j<ks.t.a_i>a_j,a_k>a_j$.显然若$p$对应的图不存在环,则$p$是一个单峰排列,则$ans=n!-cnt$,其中$cnt$为单峰排列的个数.

设元素$n$在下标$i$的位置,将剩下的$(n-1)$个元素分为两组,第一组有$(i-1)$个元素,第二组有$(n-i)$个元素,将第一组升序排列后放在下标$i$的左边,将第二组降序排列后放在下标$i$的右边,有$C_{n-1}^{i-1}$种方案.故$cnt=\sum _{i=1}^n{C}_{n-1}^{i-1}=\sum _{i=0}^{n-1}{C}_{n-1}^i=2^{n-1},ans=n!-2^{n-1}$.

$cnt$的另一求法:考虑构造一个单峰排列.将$n,(n-1),\cdots ,1$依次插入双端队列的队首或队尾,除元素$n$外,其余的$(n-1)$个元素可任意选择插入队首或队尾,有$cnt=2^{n-1}$种方案.

代码

const int MOD = 1e9 + 7;

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

  int ans = 1;
  for (int i = 2; i <= n; i++) ans = (ll)ans * i % MOD;
  cout << ((ans - qpow(2, n - 1, MOD)) % MOD + MOD) % MOD;
}

int main() {
  solve();
}

---

---

1433E. Two Round Dances

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

题意

$n(2\le n\le 20)$($n$为偶数)个人进行两轮跳舞,每轮跳舞共$\frac{n}{2}$人,每个人只能参加其中一轮,每轮跳舞的人有序.求方案数.

思路

先从$n$个人中选出$\frac{n}{2}$个人在第一轮跳舞,共$C_n^{\frac{n}{2}}$种方案.在选出的第一组中固定第一个跳舞的人,剩下的$\left(\frac{n}{2}-1\right)$个人任意排列,有$\left(\frac{n}{2}-1\right)!$种方案,同理第二组的方案也为该数.两组人数相等,故结果要除以$2$,即$ans=\frac{C_n^{\frac{n}{2}}\cdot \left(\frac{n}{2}-1\right)!\cdot \left(\frac{n}{2}-1\right)!}{2}=\frac{n!}{\frac{n^2}{2}}$.

$20!=2432902008176640000$,故最终结果不会爆ll.

代码

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

  ll ans = 1;
  for (int i = 2; i <= n; i++) ans *= i;
  cout << ans / (n * n / 2);
}

int main() {
  solve();
}

---

---

1436C. Binary Search

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

题意

上述代码实现了在下标从$0$开始的有序序列$a[]$中查找$x$.问$1\sim n$的所有排列中有多少个排列使得$x$出现在下标$pos$(下标从$0$开始)的位置,且上述代码能成功找到$x$,答案对$1\mathrm{e}9+7$取模.

第一行输入三个整数$n,x,pos(1\le x\le n\le 1000,0\le pos\le n-1)$.

思路

在二分过程中,若$mid<pos$,则$a_{mid}\le x$;若$mid>pos$,则$a_{mid}>x$.模拟二分过程,统计需要$<x$的数的个数$cn{t}_1$和需要$>x$的数的个数$cn{t}_2$,从$1\sim n$中的$su{m}_1$个$<x$的数中选$cn{t}_1$个任意排列,$su{m}_2$个$>x$的数中选$cn{t}_2$个任意排列,其余$(n-cn{t}_1-cn{t}_2-1)$个元素任意排列,其中$-1$是去掉了$x$,故$ans=A_{su{m}_1}^{cn{t}_1}\cdot A_{su{m}_2}^{cn{t}_2}\cdot (n-cn{t}_1-cn{t}_2-1)!$.

代码

const int MAXN = 1005;
const int MOD = 1e9 + 7;
int n, x, pos;
int fac[MAXN], ifac[MAXN];

void init() {  // 预处理阶乘及其逆元
  fac[0] = 1;
  for (int i = 1; i < MAXN; i++) fac[i] = (ll)fac[i - 1] * i % MOD;

  ifac[MAXN - 1] = qpow(fac[MAXN - 1], MOD - 2, MOD);
  for (int i = MAXN - 1; i; i--) ifac[i - 1] = (ll)ifac[i] * i % MOD;
}

int A(int n, int m) {  // 排列数A(n,m)
  return (ll)fac[n] * ifac[n - m] % MOD;
}

void solve() {
  init();

  cin >> n >> x >> pos;

  int cnt1 = 0, cnt2 = 0;  // 需要<x、>x的数个数
  int l = 0, r = n;
  while (l < r) {  // 模拟二分过程
    int mid = l + r >> 1;
    if (mid <= pos) l = mid + 1, cnt1 += (mid < pos);
    else r = mid, cnt2++;
  }

  cout << (ll)A(x - 1, cnt1) * A(n - x, cnt2) % MOD * fac[n - cnt1 - cnt2 - 1] % MOD;
}

int main() {
  solve();
}

---

---

1452D. Radio Towers

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

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

$x$轴上有编号$0\sim (n+1)$的$(n+2)(1\le n\le 2\mathrm{e}5)$个城市,依次在坐标$0,\cdots ,(n+1)$的位置.坐标$1,\cdots ,n$的位置上各自独立地有$\frac{1}{2}$的概率造一个信号塔,每个信号塔有一个能量$p\in [1,n]$.若在坐标$i$处有信号塔,则所有满足$|c-i|<p$的城市$c$都会收到信号.现要造信号塔,使得$0$号、$(n+1)$号城市不能收到信号,且$1,\cdots ,n$号城市都受到且只收到一个信号塔的信号,求满足要求的概率.可以证明答案可以写成分数的形式,输出该分数模$998244353$的结果.

思路

每个信号塔可视为以其为中心的一个长度为奇数的区间,问题转化为用若干个长度为奇数的区间覆盖$[1,n]$的概率.

$dp[i]$表示用若干个长度为奇数的区间覆盖区间$[1,i]$的方案数.按坐标$i$处是否造信号塔分类:①若造,则该塔只能取$p=1$,方案数即覆盖区间$[1,i-1]$的方案数,即$dp[i-1]$;②若不造,为使得坐标$i$的位置被覆盖,需将当前方案的最后一个区间的长度$+=2$,即$dp[i-2]$.状态转移方程:$dp[n]=dp[n-1]+dp[n-2]$,即Fibonacci数列.

显然总情况数为$2^n$,故$ans=\frac{fi{b}_n}{2^n}$,其中$fi{b}_n$表示Fibonacci数列的第$n$项.

代码

const int MOD = 998244353;

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

  vi fib(n + 1);
  fib[1] = 1;
  for (int i = 2; i <= n; i++) fib[i] = ((ll)fib[i - 1] + fib[i - 2]) % MOD;
  cout << qpow(qpow(2, n, MOD), MOD - 2, MOD) * fib[n] % MOD;
}

int main() {
  solve();
}

---

---

1462E1. Close Tuples (easy version)

原题指路:https://codeforces.com/problemset/problem/1462/E1

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

给定一个长度为$n$的序列$a_1,\cdots ,a_n$,求其中使得$i<j<z$且 max ⁡ { a i , a j , a z } − min ⁡ { a i , a j , a z } ≤ 2 \max\{a_i,a_j,a_z\}-\min\{a_i,a_j,a_z\}\leq 2 max{ai​,aj​,az​}−min{ai​,aj​,az​}≤2的$(i,j,z)$三元组的个数.

有$t(1\le t\le 2\mathrm{e}5)$组测试数据.每组测试数据第一行输入一个整数$n(1\le n\le 2\mathrm{e}5)$.第二行输入$n$个整数$a_1,\cdots ,a_n(1\le a_i\le n)$.数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$.

思路

满足条件的有序三元组$(a_i,a_j,a_z)$有如下几类:①$(x,x+2,x+2)$;②$(x,x+1,x+2)$;③$(x,x+1,x+1)$;④$(x,x,x+2)$;⑤$(x,x,x+1)$;⑥$(x,x,x)$.统计每个数出现的次数,枚举三元组的最小元素$x$,组合数即可.

代码

ll C(int n, int m) {  // 组合数C(n,m)
  if (n <= m) return n == m;
  else {
    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 cnt(n + 5);  // 防止越界
  for (int i = 0; i < n; i++) {
    int a; cin >> a;
    cnt[a]++;
  }

  ll ans = 0;
  for (int i = 1; i <= n; i++) {
    ans += (ll)cnt[i] * C(cnt[i + 2], 2);
    ans += (ll)cnt[i] * cnt[i + 1] * cnt[i + 2];
    ans += (ll)cnt[i] * C(cnt[i + 1], 2);
    ans += C(cnt[i], 2) * cnt[i + 2];
    ans += C(cnt[i], 2) * cnt[i + 1];
    ans += C(cnt[i], 3);
  }
  cout << ans << endl;
}

int main() {
  CaseT  // 单测时注释掉该行
    solve();
}

---

---

1475C. Ball in Berland

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

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

有编号$1\sim a$的$a$个男生和编号$1\sim b$的$b$个女生,他们组成了$k$对舞伴$(x,y)(1\le x\le a,1\le y\le b)$.从中选出两对舞伴,使得选中的$4$个人不同,求方案数.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据第一行输入三个整数$a,b,k(1\le a,b,k\le 2\mathrm{e}5)$.第二行输入$k$个整数$a_1,\cdots ,a_k$,分别表示每对舞伴中男生的编号.第三行输入$k$个整数$b_1,\cdots ,b_k$,分别表示每对舞伴中女生的编号.数据保证每组测试数据给定的$k$对舞伴相异,且所有测试数据的$a,b,k$之和不超过$2\mathrm{e}5$.

思路

$cnta[a_i]$表示$a_i$号男生在舞伴中出现的次数,$cntb[b_i]$表示$b_i$号女生在舞伴中出现的次数.对第$i$对舞伴,其对答案的贡献为总舞伴数$-a[i]$的重复次数$-b[i]$的重复次数$+1$,其中$+1$是因为$(a[i],b[i])$多减了一次.因每队舞伴都被多算了一次,故最后答案需除以$2$.

代码

void solve() {
  int n, m, k; cin >> n >> m >> k;
  vi a(k), b(k);
  vi cnta(n + 1);  // cnta[a[i]]表示a[i]号男生在舞伴中出现的次数
  vi cntb(m + 1);  // cntb[b[i]]表示b[i]号男生在舞伴中出现的次数
  for (int i = 0; i < k; i++) {
    cin >> a[i];
    cnta[a[i]]++;
  }
  for (int i = 0; i < k; i++) {
    cin >> b[i];
    cntb[b[i]]++;
  }

  ll ans = 0;
  for (int i = 0; i < k; i++) ans += k - cnta[a[i]] - cntb[b[i]] + 1;
  cout << ans / 2 << endl;
}

int main() {
  CaseT  // 单测时注释掉该行
    solve();
}

---

---

1475E. Advertising Agency

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

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

编号$1\sim n$的$n$个博主分别有$a_1,\cdots ,a_n$个粉丝,假定任一粉丝只关注一个博主.现要请$k$个博主,使得所有粉丝数之和最大,求方案数,答案对$1\mathrm{e}9+7$取模.

有$t(1\le t\le 1000)$组测试数据.每组测试数据第一行输入两个整数$n,k(1\le k\le n\le 1000)$.第二行输入$n$个整数$a_1,\cdots ,a_n(1\le a_i\le n)$.数据保证所有测试数据的$n$之和不超过$1000$.

思路

将博主按粉丝数降序排列,贪心地选前$k$个博主.若遇到粉丝数相同的博主,组合数即可.

代码

const int MAXN = 1005;
const int MOD = 1e9 + 7;
int fac[MAXN], ifac[MAXN];

void init() {  // 预处理阶乘及其逆元
  fac[0] = 1;
  for (int i = 1; i < MAXN; i++) fac[i] = (ll)fac[i - 1] * i % MOD;

  ifac[MAXN - 1] = qpow(fac[MAXN - 1], MOD - 2, MOD);
  for (int i = MAXN - 1; i; i--) ifac[i - 1] = (ll)ifac[i] * i % MOD;
}

int C(int n, int m) {  // 组合数C(n,m)
  return (ll)fac[n] * ifac[m] % MOD * ifac[n - m] % MOD;
}

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

  for (int i = n; i >= 0; i--) {
    if (cnt[i] >= k) {
      cout << C(cnt[i], k) << endl;
      return;
    }
    else k -= cnt[i];
  }
}

int main() {
  init();
  CaseT  // 单测时注释掉该行
    solve();
}

---

---

1513B. AND Sequences

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

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

称一个长度为$n(n\ge 2)$的序列$a_1,\cdots ,a_n$是好的,如果对$\forall i\in [1,n-1]$,都有$a_1\&\cdots \&a_i=a_{i+1}\&\cdots \&a_n$.给定一个序列$a_1,\cdots ,a_n$,求有多少个$1\sim n$的排列$p_1,\cdots ,p_{n}s.t.a_{p_1},\cdots ,a_{p_n}$是好的,答案对$1\mathrm{e}9+7$取模.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据第一行输入一个整数$n(2\le n\le 2\mathrm{e}5)$.第二行输入$n$个整数$a_1,\cdots ,a_n(0\le a_i\le 1\mathrm{e}9)$.数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$.

思路

若序列$a_1,\cdots ,a_n$是好的,则$a_1=a_2\&\cdots \&a_n$,两边与$a_1$得$a_1=a_1\&\cdots \&a_n$,同理$a_n=a_1\&\cdots \&a_n$,即序列首尾的数只能是序列所有元素的与.

设$a_1=a_n=a_1\&\cdots \&a_n=x$,则 x ≤ min ⁡ { a 1 , ⋯   , a n } x\leq \min\{a_1,\cdots,a_n\} x≤min{a1​,⋯,an​}.显然确定$a_1$和$a_n$后,任一前缀与、后缀与都分别由$a_1$、$a_n$决定,故中间的$(n-2)$个元素可任意排列.设序列中值为$x$的元素的个数为$cnt$,则$ans=A_x^2\cdot (n-2)!$.

代码

const int MAXN = 2e5 + 5;
const int MOD = 1e9 + 7;
int fac[MAXN], ifac[MAXN];

void init() {  // 预处理阶乘及其逆元
  fac[0] = 1;
  for (int i = 1; i < MAXN; i++) fac[i] = (ll)fac[i - 1] * i % MOD;

  ifac[MAXN - 1] = qpow(fac[MAXN - 1], MOD - 2, MOD);
  for (int i = MAXN - 1; i; i--) ifac[i - 1] = (ll)ifac[i] * i % MOD;
}

int A(int n, int m) {  // 排列数A(n,m)
  if (n < m) return 0;
  else return (ll)fac[n] * ifac[n - m] % MOD;
}

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

  int cnt = count(all(a), tmp);
  cout << (ll)A(cnt, 2) * fac[n - 2] % MOD << endl;
}

int main() {
  init();
  CaseT  // 单测时注释掉该行
    solve();
}

---

---