4275.Dijkstra序列

原题链接

思路

Dijkstra板子改改(内部判断一下能否实现即可)

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
int g[N][N];
int dist[N];
bool st[N];
int fl[N];
int n,m,k;

bool dj()
{
memset(dist, 0x3f, sizeof dist);
//这里st忘了复原了,所以我当时出错。
memset(st, 0, sizeof st);
dist[fl[1]] = 0;
for(int t = 1; t <= n; t ++)
{
// 我的写法
int i = -1;
for(int j = 1; j <= n; j ++)
if(!st[j] && (i == -1 || dist[i] >= dist[j]))
{ // 需要判断是否是数列给的数,所以等于时需要加上
if(dist[i] == dist[j])
{
if(i != fl[t])i = j;
// i == fl[t]时i不变即可
continue;
}
i = j;
}
if(i != fl[t])return false;
// y总写法

//int i = fl[i];
// for(int j = 1; j <= n; j ++)
// if(!st[j] && dist[i] >= dist[j])
// {
// return false;
// }

st[i] = true;
for(int j = 1; j <= n; j ++)
{
dist[j] = min(dist[j], dist[i] + g[i][j]);
}
}
return true;
}
int main()
{
memset(g, 0x3f, sizeof g);
cin >> n >> m;
while(m --)
{
int a, b, c;
cin >> a >> b >> c;
g[a][b] = min(g[a][b], c);
g[b][a] = min(g[b][a], c);
}
cin >> k;
while(k --)
{
for(int i = 1; i <= n; i ++)cin >> fl[i];
if(dj())cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}