AtCoder Beginner Contest 375

省流版
  • A. 枚举所有子串判断即可
  • B. 模拟计算记录累加即可
  • C. 根据旋转的周期性对每个点旋转次数取模后暴力旋转即可
  • D. 枚举\(k\),考虑 \(i,j\)的对数,写成数学表达式后维护个数和位置和即可
  • E. 背包问题,以前两个数组和为状态,考虑每个数移动到何处转移即可
  • F. 逆向,删边变加边,维护加边后的距离变化即可
  • G. 最短路径图的割边判断,根据最短路条数判断是否必经即可

A - Seats (abc375 A)

题目大意

给定一个字符串,问有多少个形如#.#的子串。

解题思路

枚举所有子串依次比较即可。

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    string s;
    cin >> n >> s;
    int ans = 0;
    for (int i = 0; i < n; ++i) {
        ans += (s.substr(i, 3) == "#.#");
    }
    cout << ans << '\n';

    return 0;
}



B - Traveling Takahashi Problem (abc375 B)

题目大意

二维平面,给定\(n\)个点,从\((0,0)\)出发,依次经历这些点,最后回到 \((0,0)\),求其欧几里德距离和。

解题思路

依次模拟计算距离和,累加即可。

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    vector<array<int, 2>> p(n + 2);
    p[0] = {0, 0};
    for (int i = 1; i < n + 1; i++) {
        cin >> p[i][0] >> p[i][1];
    }
    p[n + 1] = {0, 0};
    double ans = 0;
    auto dist = [](array<int, 2> a, array<int, 2> b) {
        return sqrt(1ll * (a[0] - b[0]) * (a[0] - b[0]) +
                    1ll * (a[1] - b[1]) * (a[1] - b[1]));
    };
    for (int i = 1; i < p.size(); ++i) {
        ans += dist(p[i], p[i - 1]);
    }
    cout << fixed << setprecision(10) << ans << '\n';

    return 0;
}



C - Spiral Rotation (abc375 C)

题目大意

给定一个\(n \times n\)的正方形,对于 \(i \in [1, n]\) ,依次进行如下操作:

  • \((i,i)\)为一角的 \((n - i + 1) \times (n - i - 1)\) 的正方形顺时针旋转\(90\)度。

输出最终的矩形。

解题思路

注意到旋转的区间是往中间收缩的,因为旋转操作具有结合性和周期性(旋转\(4\)次等于没旋转),所以依次遍历该正方形的每个元素\((i,j)\),可以求出它进行了多少次 顺时针旋转 \(90\)度,然后对 \(4\)取余后,再进行对应次数的顺时针旋转即可。

其实暴力的复杂度是\((2n^2 \log n)\),虽然\(n \leq 3000\),但貌似还是不能过。

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    vector<string> s(n);
    for (auto& i : s)
        cin >> i;
    auto tr = [](int x, int y, int n) { return make_pair(y, n - 1 - x); };
    auto tr_cnt = [&](int x, int y, int n, int c) {
        for (int i = 0; i < c; ++i) {
            tie(x, y) = tr(x, y, n);
        }
        return make_pair(x, y);
    };
    vector<string> t = s;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            int cnt = min({i, j, n - 1 - i, n - 1 - j}) + 1;
            auto [x, y] = tr_cnt(i, j, n, cnt % 4);
            t[x][y] = s[i][j];
        }
    }
    for (auto i : t)
        cout << i << '\n';

    return 0;
}



D - ABA (abc375 D)

题目大意

给定一个字符串\(s\),问有多少\(i < j < k\),满足 \(s_is_js_k\)是回文串。

解题思路

枚举\(k\),考虑满足条件的 \(i\),那必然有\(s_i == s_k\),然后固定这个\(i\),考虑 \(j\)的方案数,即为 \(k - i - 1\)

也就是说,对于当期 \(k\),其对答案的贡献为 \(\sum_{i < k \& s_i==s_k} k - i - 1 = (k-1) cnt_{a_k} - \sum_{i < k \& s_i == s_k} i\)

因此我们只需动态维护两个数组:

  • \(cnt_i\)表示字符为\(i\)的位置数
  • \(sum_i\)表示字符为\(i\)的位置的和

就可以\(O(1)\)求得固定\(k\)时, \((i,j,k)\)的方案数。

对所有的\(k\)累加即为答案。时间复杂度为 \(O(n)\)

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    string s;
    cin >> s;
    array<int, 26> cnt{};
    array<LL, 26> sum{};
    LL ans = 0;
    for (int i = 0; i < s.size(); ++i) {
        int c = s[i] - 'A';
        ans += 1ll * cnt[c] * (i - 1) - sum[c];
        cnt[c]++;
        sum[c] += i;
    }
    cout << ans << '\n';

    return 0;
}



E - 3 Team Division (abc375 E)

题目大意

三个数组,操作为,花费\(1\)的代价,将一个数从一个数组移动到另一个数组。

问最小的代价,使得三个数组的和相等,或告知不可行。

数组和\(\sum b_i \leq 1500\)

解题思路

考虑两个数组的简化版。

考虑朴素搜索,即依次考虑每个数,应该移动还是不移动。其复杂度显然是\(O(2^n)\)

因为状态是包含了每个数的数组位置,它过于冗余了,考虑到我们要求的答案,我们可以仅保留一个简化的状态即可:

  • 一个数组的和

知道了一个数组的和,对于当前数,移动还是不移动,其带来的影响都可以进行转移,最后也能直接得到答案。

因为总数不变,所以知道了一个数组的和,另一个数组的和其实也知道了,即设 \(dp[i][j]\)表示考虑了前 \(i\)个数,第一个数组和为 \(j\)的最小移动代价。对于一个数,考虑移动或不移动,对 \(j\)的影响转移即可。这样的复杂度即为 \(O(n \sum b_i)\)

而这里有三个数组,按照同样的思路,知道了前两个数组的和,第三个数组的和也确定了,即设\(dp[i][j][k]\)表示考虑了前 \(i\)个数,第一个数组和为 \(j\),第二个数组和为\(k\)的最小移动代价,转移仍然是对于一个数,考虑移动到哪个数组,或者不移动即可。这样的时间复杂度为\(O(n (\sum b_i) ^ 2)\)

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

const int inf = 1e9 + 7;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    array<vector<int>, 3> p{};
    int sum = 0;
    for (int i = 0; i < n; i++) {
        int a, b;
        cin >> a >> b;
        --a;
        p[a].push_back(b);
        sum += b;
    }
    if (sum % 3 != 0)
        cout << -1 << '\n';
    else {
        vector<vector<int>> dp(sum + 1, vector<int>(sum + 1, inf));
        dp[accumulate(p[0].begin(), p[0].end(), 0)]
          [accumulate(p[1].begin(), p[1].end(), 0)] = 0;
        for (auto& i : p[0]) {
            auto dp2 = dp;
            for (int j = 0; j <= sum; j++) {
                for (int k = 0; k <= sum; k++) {
                    if (j >= i)
                        dp2[j - i][k] = min(dp2[j - i][k], dp[j][k] + 1);
                    if (j >= i && k + i <= sum)
                        dp2[j - i][k + i] =
                            min(dp2[j - i][k + i], dp[j][k] + 1);
                }
            }
            dp.swap(dp2);
        }
        for (auto& i : p[1]) {
            auto dp2 = dp;
            for (int j = 0; j <= sum; j++) {
                for (int k = 0; k <= sum; k++) {
                    if (k >= i)
                        dp2[j][k - i] = min(dp2[j][k - i], dp[j][k] + 1);
                    if (k >= i && j + i <= sum)
                        dp2[j + i][k - i] =
                            min(dp2[j + i][k - i], dp[j][k] + 1);
                }
            }
            dp.swap(dp2);
        }
        for (auto& i : p[2]) {
            auto dp2 = dp;
            for (int j = 0; j <= sum; j++) {
                for (int k = 0; k <= sum; k++) {
                    if (k + i <= sum)
                        dp2[j][k + i] = min(dp2[j][k + i], dp[j][k] + 1);
                    if (j + i <= sum)
                        dp2[j + i][k] = min(dp2[j + i][k], dp[j][k] + 1);
                }
            }
            dp.swap(dp2);
        }
        int target = sum / 3;
        if (dp[target][target] == inf) {
            dp[target][target] = -1;
        }
        cout << dp[target][target] << '\n';
    }

    return 0;
}



F - Road Blocked (abc375 F)

题目大意

给定一张无向图,维护以下两种操作:

  • 1 x,删去第\(x\)条边
  • 2 u v,问\(u \to v\)的最短路长度

其中操作 \(1\)最多 \(300\)次。

解题思路

加边往往比删边好做,注意到询问没有强制在线,因此可以倒着考虑操作,这样删边操作就变成加边操作了。

因为点数\(\leq 300\),先用 \(floyd\)\(O(n^3)\)求出任意两点的最短路,这样操作 \(2\)就可以 \(O(1)\)回答,然后考虑操作 \(1\)加边后,怎么更新这个两点的最短路。

考虑加的边 \(u \to v\)对任意两点 \(x \to y\) 的最短路的影响,其实就两种情况:

  • \(x \to y\)的最短路不经过 \(u \to v\)
  • \(x \to y\)的最短路经过 \(u \to v\)

在加边之前,\(dis[x][y]\)的值就属于第一种情况,现在加边后, \(dis[x][y]\)就是这两种情况的最小值,因此\(dis[x][y]\)只要和第二种情况取个最小值,这样\(dis[x][y]\)就变成 加上边 \(u \to v\)后的值了。

而第二种情况的距离即为 \(dis[x][u] + u \to v + dis[v][y]\)\(dis[x][v] + v \to u + dis[u][y]\)

加一边,更新一次的复杂度为 \(O(n^2)\),因为操作 \(1\)不超过 \(300\),因此总的时间复杂度为 \(O(n^3)\)

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

const LL inf = 1e18;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, m, q;
    cin >> n >> m >> q;
    vector<vector<LL>> dis(n, vector<LL>(n, inf));
    vector<array<int, 3>> edges(m);
    for (int i = 0; i < m; ++i) {
        int u, v, w;
        cin >> u >> v >> w;
        --u, --v;
        dis[u][v] = dis[v][u] = min(dis[u][v], (LL)w);
        edges[i] = {u, v, w};
    }
    for (int i = 0; i < n; ++i)
        dis[i][i] = 0;
    vector<array<int, 3>> query(q);
    for (auto& [op, x, y] : query) {
        cin >> op >> x;
        if (op == 1) {
            --x;
            auto [u, v, w] = edges[x];
            dis[u][v] = dis[v][u] = inf;
        } else {
            cin >> y;
            --x, --y;
        }
    }
    for (int k = 0; k < n; ++k)
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < n; ++j)
                dis[i][j] = min(dis[i][j], dis[i][k] + dis[k][j]);
    vector<LL> ans;
    for (int i = q - 1; i >= 0; --i) {
        auto [op, x, y] = query[i];
        if (op == 1) {
            auto [u, v, w] = edges[x];
            dis[u][v] = dis[v][u] = min(dis[u][v], (LL)w);
            for (int i = 0; i < n; ++i)
                for (int j = 0; j < n; ++j) {
                    dis[i][j] =
                        min(dis[i][j], dis[i][u] + dis[u][v] + dis[v][j]);
                    dis[i][j] =
                        min(dis[i][j], dis[i][v] + dis[v][u] + dis[u][j]);
                }
        } else {
            ans.push_back((dis[x][y] == inf ? -1 : dis[x][y]));
        }
    }
    reverse(ans.begin(), ans.end());
    for (auto x : ans)
        cout << x << '\n';

    return 0;
}



G - Road Blocked 2 (abc375 G)

题目大意

给定一张无向图,对于每一条边,回答以下询问:

  • 删去该边后,\(1 \to n\)的最短路长度是否发生变化。

解题思路

题意就是问一条边是不是最短路的必经之边,换句话说就是一条边是否被所有最短路径经过。

由于\(1 \to n\)可能有多条最短路,我们先求出每条边,是否被一条最短路经过。

如果一条边 \(u \to v\)满足: \(1 \to n = 1 \to u \to v \to n\)或者 \(1 \to n = 1 \to v \to u \to n\),这说明该边被一条最短路径经过。

但怎么说该边被所有的最短路径经过呢?

如果\(1 \to n\)的最短路径有 \(z\)条,而 \(1 \to u\)的最短路径有 \(x\)条, \(v \to n\)的最短路径有 \(y\)条,由乘法原理,\(1 \to n\)且经过 \(u \to v\)的路径就有 \(xy\)条,而如果有\(z = xy\),那这就说明该边是必经边。

因此分别从点\(1\)和点 \(n\)进行 \(dijkstra\),求出 \(dis_1[i],dis_n[i]\)\(cnt_1[i],cnt_n[i]\) 表示\(1 \to i\)的最短路径长度和数量, \(n \to i\)的最短路径长度和数量,对于任意一条边\(u \to v, w\),如果满足二者任意其一:

  • \(dis_1[u] + w + dis_n[v] == dis_1[n]\)\(cnt_1[u] \times cnt_n[v] == cnt_1[n]\)
  • \(dis_1[v] + w + dis_n[u] == dis_1[n]\)\(cnt_1[v] \times cnt_n[u] == cnt_1[n]\)

则该边是最短路必经边。

其实根据一条边是否被最短路经过,建立一张最短路图,这些必经边就是所谓的割边,即删去该边后, \(1 \to n\)不连通。

注意 \(cnt\)可能会非常大,甚至会超过 \(int128\), 而此处我们只需判断乘积是否相等,可以取模。

神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;

const LL inf = 1e18;
const int mo = 1e9 + 7;

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, m;
    cin >> n >> m;
    vector<vector<array<int, 2>>> edge(n);
    vector<array<int, 3>> e(m);
    for (int i = 0; i < m; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        --u, --v;
        edge[u].push_back({v, w});
        edge[v].push_back({u, w});
        e[i] = {u, v, w};
    }
    auto dijkstra = [&](int s) {
        vector<LL> dis(n, inf);
        vector<LL> cnt(n, 0);
        dis[s] = 0;
        cnt[s] = 1;
        priority_queue<array<LL, 2>, vector<array<LL, 2>>,
                       greater<array<LL, 2>>>
            team;
        team.push({0, s});
        while (!team.empty()) {
            auto [d, u] = team.top();
            team.pop();
            if (dis[u] < d)
                continue;
            for (auto [v, w] : edge[u]) {
                if (dis[v] > dis[u] + w) {
                    dis[v] = dis[u] + w;
                    team.push({dis[v], v});
                    cnt[v] = cnt[u];
                } else if (dis[v] == dis[u] + w) {
                    cnt[v] += cnt[u];
                    if (cnt[v] >= mo)
                        cnt[v] -= mo;
                }
            }
        }
        return make_pair(dis, cnt);
    };
    auto [dis1, cnt1] = dijkstra(0);
    auto [dis2, cnt2] = dijkstra(n - 1);
    vector<int> ans(m, 0);
    for (int i = 0; i < m; ++i) {
        auto& [u, v, w] = e[i];
        if (dis1[u] + w + dis2[v] == dis1[n - 1] &&
                cnt1[u] * cnt2[v] % mo == cnt1[n - 1] ||
            dis1[v] + w + dis2[u] == dis1[n - 1] &&
                cnt1[v] * cnt2[u] % mo == cnt1[n - 1]) {
            cout << "Yes" << '\n';
        } else {
            cout << "No" << '\n';
        }
    }

    return 0;
}