D. Colored Portals

最新推荐文章于 2024-08-23 23:26:17 发布

.y.a.o.

最新推荐文章于 2024-08-23 23:26:17 发布

阅读量831

点赞数 19

文章标签：算法 c++

本文链接：https://blog.csdn.net/m0_75129175/article/details/141319473

版权

题面

分析

如果 $x$ 和 $y$ 能够直接到达，那么最近的一定是二者绝对值，如果不能直接到达，需要找一个中间量，那么最优方法就是找一个新的 $z$ ，有以下三种情况：

如果 $\lt x$ ，那么最短距离为 $(x - z) + (y - z)$ 。
如果 $\leq z \leq y$ ，那么最短距离为 $(y - x)$ 。
如果 $\gt y$ ，那么最短距离为 $(z - x) + (z - y)$ 。
可以发现从 $x$ 出发那么最优方案就是选一个距离 $x$ 最短的点作为中间量。

代码

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

string c[] = {"BG", "BR", "BY", "GR", "GY", "RY"};

void solve() {
    int n, q;
    cin >> n >> q;
    vector<int> a(n);
    vector<vector<int>> l(n, vector<int>(6));
    vector<vector<int>> r(n, vector<int>(6));
    for (int i = 0; i < n; ++i) {
        string s;
        cin >> s;
        for (int j = 0; j < 6; ++j) {
            if (s == c[j]) {
                a[i] = j;
                break;
            }
        }
    }
    for (int i = 0; i < 6; ++i) {
        int last = 1e9;
        for (int j = 0; j < n; ++j) {
            l[j][i] = last;
            if (a[j] == i) {
                last = j;
            }
        }
        last = 1e9;
        for (int j = n - 1; j >= 0; --j) {
            r[j][i] = last;
            if (a[j] == i) {
                last = j;
            }
        }
    }
    while (q--) {
        int x, y;
        cin >> x >> y;
        --x, --y;
        int ans = 1e9;
        for (int i = 0; i < 6; ++i) {
            string s = c[i];
            string s1 = c[a[x]];
            string s2 = c[a[y]];
            if (s1 == s2) {
                ans = min(ans, abs(x - y));
            }
            if ((s1[0] == s[0] || s1[0] == s[1] || s1[1] == s[0] || s1[1] == s[1]) && (s2[0] == s[0] || s2[0] == s[1] || s2[1] == s[0] || s2[1] == s[1])) {
                ans = min(ans, abs(x - l[x][i]) + abs(y - l[x][i]));
                ans = min(ans, abs(x - r[x][i]) + abs(y - r[x][i]));
            }
        }
        if (ans == 1e9) ans = -1;
        cout << ans << "\n";
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T;
    cin >> T;
    while (T--) {
        solve();
    }
}