Hello 2023 A-D

比赛链接

A

题意

给一个字符串每个物品对应的灯的照明方向，L/R 能照亮它左侧/右侧的所有物品（不包括自己对应的物品），现在能交换相邻两个灯一次（不改变照明方向），问能否找亮所有物品。

题解

知识点：贪心。

显然，如果存在 LR 或 RL 就可以照亮全部，否则全是 L 或 R 就不可行。

时间复杂度 \(O(n)\)

空间复杂度 \(O(n)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

bool solve() {
    int n;
    cin >> n;
    string s;
    cin >> s;
    s = "?" + s;
    for (int i = 1;i < n;i++) {
        if (s[i] != s[i + 1]) {
            if (s[i] == 'L') cout << i << '\n';
            else cout << 0 << '\n';
            return true;
        }
    }
    return false;
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t = 1;
    cin >> t;
    while (t--) {
        if (!solve()) cout << -1 << '\n';
    }
    return 0;
}

B

题意

构造一组数，使得任意相邻两项之和等于全部和。

题解

知识点：构造。

\(n\) 为偶数时，构造 \(1,-1,1,-1,\cdots\) 即可。

\(n\) 为奇数时，显然奇数项和偶数项要各自相等，随后由 \(a_1+\cdots+a_n = a_{n-1}+a_{n}\) 可以得到 \((n-1)a_1+(n-3)a_2 = 0\) ，取 \(a_1 = n-3,a_2 = 1-n\) 即可，只有 \(n=3\) 时无解（因为 \(a_1 = 0\)）。

时间复杂度 \(O(n)\)

空间复杂度 \(O(n)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

bool solve() {
    int n;
    cin >> n;
    if (n & 1) {
        if (n == 3) return false;
        cout << "YES" << '\n';
        for (int i = 1;i <= n;i++) {
            if (i & 1) cout << n - 3 << ' ';
            else cout << 1 - n << ' ';
        }
        cout << '\n';
    }
    else {
        cout << "YES" << '\n';
        for (int i = 1;i <= n;i++) {
            if (i & 1) cout << 1 << ' ';
            else cout << -1 << ' ';
        }
        cout << '\n';
    }
    return true;
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t = 1;
    cin >> t;
    while (t--) {
        if (!solve()) cout << "NO" << '\n';
    }
    return 0;
}

C

题意

给一组数，可以修改元素变成其相反数。问最少修改几次，可以使得第 \(m\) 个前缀和 \(a_1+\cdots+a_m\) 是所有前缀和里最小的。

题解

知识点：枚举，前缀和，数学，贪心。

定义 \(a[l,r] = a_l+\cdots+a_r\) 。

当 \(k\in [1,m-1]\) 时

\[\begin{aligned} a[1,k] &\geq a[1,m]\\ a[1,k] &\geq a[1,k] + a[k+1,m]\\ 0 &\geq a[k+1,m] \end{aligned} \]

当 \(k\in [m+1,n]\) 时

\[\begin{aligned} a[1,k] &\geq a[1,m]\\ a[1,m] + a[m+1,k] &\geq a[1,m]\\ a[m+1,k] &\geq 0 \end{aligned} \]

所以只要保证任意 \(i\in[2,m]\) ，满足 \(a[i,m]\leq 0\) ；任意 \(i\in[m+1,n]\) ，满足 \(a[m+1,i] \geq 0\) 即可。

每次操作时，贪心地取最优的即可。

时间复杂度 \(O(n\log n)\)

空间复杂度 \(O(n)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

int a[200007];
bool solve() {
    int n, m;
    cin >> n >> m;
    for (int i = 1;i <= n;i++) cin >> a[i];
    int cnt = 0;
    multiset<int> ms;
    ll sum = 0;
    for (int i = m;i >= 2;i--) {
        sum += a[i];
        ms.insert(a[i]);
        if (sum > 0) {
            sum -= 2 * (*prev(ms.end()));
            ms.erase(prev(ms.end()));
            cnt++;
        }
    }
    ms.clear();
    sum = 0;
    for (int i = m + 1;i <= n;i++) {
        sum += a[i];
        ms.insert(a[i]);
        if (sum < 0) {
            sum -= 2 * (*ms.begin());
            ms.erase(ms.begin());
            cnt++;
        }
    }
    cout << cnt << '\n';
    return true;
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t = 1;
    cin >> t;
    while (t--) {
        if (!solve()) cout << -1 << '\n';
    }
    return 0;
}

D

题意

给定一组头发长度 \(a_i\) ，以及理想头发长度 \(b_i\) 。

理发师有刀片 \(x_i\) ，每个刀片只能用一次，每次可以修减一段连续区间的头发，满足 \(a'_i = \min(a_i,x),i\in[L,R]\)。

问理发师能不能通过这些刀片将 \(a\) 修剪至 \(b\) 。

题解

知识点：单调栈。

显然 \(a_i<b_i\) 无解。

利用最大值单调栈维护刀片的值。以下按顺序满足：

1. \(b_i\) 大于栈顶刀片，则栈顶刀片因为太小不能再用了，刀片需要出栈直至 \(b_i\) 小于等于栈顶刀片或栈空。
2. \(b_i = a_i\) ，说明 \(b_i\) 不需要修剪，什么都不用干。
3. \(b_i \neq a_i\) ，说明 \(b_i\) 需要修剪，此时如果 \(b_i\) 小于栈顶刀片或栈空，则需要使用新的刀片，满足 \(x = b[i]\) ，如果不存在这个刀片则无解。

全部满足后，即 YES 。

时间复杂度 \(O(n+m\log m)\)

空间复杂度 \(O(n+m)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

int a[200007];
int b[200007];
bool solve() {
    int n;
    cin >> n;
    for (int i = 1;i <= n;i++) cin >> a[i];
    for (int i = 1;i <= n;i++) cin >> b[i];
    int m;
    cin >> m;
    map<int, int> mp;
    for (int i = 1;i <= m;i++) {
        int x;
        cin >> x;
        mp[x]++;
    }
    stack<int> st;
    for (int i = 1;i <= n;i++) {
        if (a[i] < b[i]) return false;
        while (!st.empty() && b[i] > st.top()) st.pop();
        if (a[i] != b[i]) {
            if (st.empty() || b[i] < st.top()) {
                if (mp[b[i]]) {
                    mp[b[i]]--;
                    st.push(b[i]);
                }
                else return false;
            }
        }
    }
    cout << "YES" << '\n';
    return true;
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t = 1;
    cin >> t;
    while (t--) {
        if (!solve()) cout << "NO" << '\n';
    }
    return 0;
}