P1078 [NOIP2012 普及组] 文化之旅

题目描述

有一位使者要游历各国，他每到一个国家，都能学到一种文化，但他不愿意学习任何一种文化超过一次（即如果他学习了某种文化，则他就不能到达其他有这种文化的国家）。不同的国家可能有相同的文化。不同文化的国家对其他文化的看法不同，有些文化会排斥外来文化（即如果他学习了某种文化，则他不能到达排斥这种文化的其他国家）。

现给定各个国家间的地理关系，各个国家的文化，每种文化对其他文化的看法，以及这位使者游历的起点和终点（在起点和终点也会学习当地的文化），国家间的道路距离，试求从起点到终点最少需走多少路。

下面是这道题的链接

https://www.luogu.com.cn/problem/P1078

说下这道题的总结：

这道题目其实（应该）没有想象中的难，一个DFS加简单的剪枝应该就能过。 首先说一下剪枝。对于国家i、j间的多条路径，可以在输入时就直接取其中距离最短的路径，把其它路径直接扔掉。显然这种方法对于求最优解无影响，且得到的总距离也较小。接下来假设一条路径可经过国家i到达终点t（且该路径距离最短），很显然s到i的路径也是最短的，i到t的路径也是最短的，故建立一数组sum[i]来表示到达国家i的最短路径，当dfs到国家i时若此时的总距离>=sum[i]，则直接结束递归，这样可以大幅减小时间复杂度（相应的，也不用单独建立数组确认国家i是否已走过）。 然后是判断部分。这一部分很烦，需要慢慢来。首先建立一个数组存储旅行者已经学习过的所有文化，然后再准备前往国家i前进行检测就行了。注意国家i的文化不能与已学习的任何一种文化相同或排斥。 显然任和一条路径的长度均>0，故只要查看路径长度就能判断两个国家间是否有路径。 

下面是我的代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 1020;
typedef pair<int, int>PII;
int n, k, m;
int s, t;
struct Node { int w, v; };
vector<Node>e[N];
int dist[N];
int vis[N];
int c[N];
int mp[N][N];
int pre[N];
priority_queue<PII>q;
const int MAXN = 707406378;//最大值 
int solve(int u, int v) {
    int t = u;
    while (t) {
        if (mp[c[v]][c[t]]||c[t] == c[v]) {
            return false;
        }
        t = pre[t];
    }
    return true;
}
void dij(int s) {
    memset(dist, 127 / 3, sizeof(dist));
    dist[s] = 0;
    q.push({ 0,s });
    while (q.size()) {
        auto t = q.top();
        q.pop();
        int u = t.second;
        if (vis[u])continue;
        vis[u] = 1;
        for (auto ed : e[u]) {
            int w = ed.w;
            int v = ed.v;
            if ((dist[v] > dist[u] + w) && solve(u, v)) {
                dist[v] = dist[u] + w;
                pre[v] = u;
                q.push({ -dist[v],v });
            }
        }
    }
}
int main() {
    cin >> n >> k >> m >> s >> t;
    for (int i = 1; i <= n; i++)cin >> c[i];
    for (int i = 1; i <= k; i++) {
        for (int j = 1; j <= k; j++)cin >> mp[i][j];
    }
    for (int i = 0; i < m; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        e[a].push_back({ c,b });
        e[b].push_back({ c,a });
    }
    dij(s);
    if (dist[t] ==MAXN)cout << "-1";
    else cout << dist[t];
    return 0;
}