codefoces 1574 B. Combinatorics Homework

本场比赛其他题目题解

A. Regular Bracket Sequences
B. Combinatorics Homework
C. Slay the Dragon
D. The Strongest Build

B. Combinatorics Homework

开启传送门

题目描述

You are given four integer values $a$, $b$, $c$ and $m$.

Check if there exists a string that contains:

• $a$ letters ‘A’;
• $b$ letters ‘B’;
• $c$ letters ‘C’;
• no other letters;
• exactly $m$ pairs of adjacent equal letters (exactly $m$ such positions $i$ that the $i$-th letter is equal to the $(i+1)$-th one).

Input

The first line contains a single integer $t$ ($1\le t\le 10^4$) — the number of testcases.

Each of the next $t$ lines contains the description of the testcase — four integers $a$, $b$, $c$ and $m$ ($1\le a,b,c\le 10^8$; $0\le m\le 10^8$).

Output

For each testcase print “YES” if there exists a string that satisfies all the requirements. Print “NO” if there are no such strings.

You may print every letter in any case you want (so, for example, the strings yEs, yes, Yes and YES will all be recognized as positive answer).

Example

input1

3
2 2 1 0
1 1 1 1
1 2 3 2

output1

YES
NO
YES

Note

In the first testcase strings “ABCAB” or “BCABA” satisfy the requirements. There exist other possible strings.

In the second testcase there’s no way to put adjacent equal letters if there’s no letter that appears at least twice.

In the third testcase string “CABBCC” satisfies the requirements. There exist other possible strings.

题意

多组样例，每次给你四个整数 $a$, $b$, $c$, $m$。让你找到一个字符串 $s$ 满足：

• 有 $a$ 个字符 ‘A’;
• 有 $b$ 个字符 ‘B’;
• 有 $c$ 个字符 ‘C’;
• 没有除上述三个字符外的其他字符;
• 恰好有 $m$ 个相邻的字符一样（也即，恰好有 $m$ 个位置 $i$ 满足 $s[i]==s[i+1]$）

分析

先说结论：找到最小的 $m_{min}$ 和 最大的 $m_{max}$ ，如果 $m_{min}\le m\le m_{max}$，那么输出 “YES”，否则输出 “NO”。

所以转化为两个问题：

1. 如何求极限
2. 如何证明 $m\in [m_{min},m_{max}]$ 就有解。

先来求 $1$。

$m_{max}$ 显然比较好求，显然我只需要构成这样的字符串 $AA\dots AABB\dots BBCC\dots CC$ 即可，所以 $m_{max}=max(0,a-1)+max(0,b-1)+max(0,c-1)$

$m_{min}$ 比较难想一点，我们不妨设 $a\le b\le c$ （谁大谁小无所谓，无非就是轮换一下）。那么我们想象一下，如果 $a>b+c+1$ ，那么根>据鹊巢原理， $m_{min}\le 1$。所以一个比较容易想的是：$m_{min}=max(0,a-(b+c+1))$。

然后证明 $2$。

从上述的推导中，我们证明了 $m_{max}$ 和 $m_{min}$ 是可达的。

也即现在需要证明 $\forall i\in [m_{min},m_{max})$ 如果 $m$ 可以等于 $i$ ，那么一定可以构造使得 $m$ 可以等于 $i+1$。

这显然就是个数学归纳法了。

我们证明了初始条件 $m_{min}$ 可达，现在假设 $i$ 可达，证明 $i+1$ 可达。

我们不妨设 $i$ 对应的字符串为 $s_i$。因为 $i\ne m_{max}$ ，所以 $s_i$ 一定不是 $AA\dots AABB\dots BBCC\dots CC$ 及其类似形式。也即一定
存在 $(i,j)$ 满足 $a[i]==a[j]\&\&a[i]\ne a[i+1\dots j-1]$，那么我们不妨将 $a[i+1\dots j-1]$ 挪到 $a_j$ 后面，显然，这样做会造成 $m++$ 。命题得证。

#include <bits/stdc++.h>
using namespace std;
int main() {
  ios::sync_with_stdio(0),cin.tie(0);
  int t; cin >> t; while (t--) {
    int a, b, c, m;
    cin >> a >> b >> c >> m;
    int mn = max({c - (a + b + 1),
                  b - (c + a + 1),
                  a - (b + c + 1)}),
        mx = (a - 1) + (b - 1) + (c - 1);
    if (mn <= m && m <= mx) {
      cout << "YES\n";
    } else {
      cout << "NO\n";
    }
  }
  return 0;
}