NOIP2012普及组 (四年后的)解题报告 -SilverN

 1 /*NOIP普及组2012 t4 culture*/
 2 //根据多个解题报告所说，floyd算法能过只是因为数据范围小
 3 //我们需要dij或者dfs 
 4 #include<iostream>
 5 #include<algorithm>
 6 #include<cmath>
 7 using namespace std;
 8 const int inf=10000000;
 9 int c[5000];//文明[i]的文化 
10 int fl[300][300];//文化拒绝 
11 int mp[300][300];//图 
12 //int DFS(){
13     
14 //}
15 int main(){
16     int i,j,n,k,m,s,t;
17 //    freopen("culture.in","r",stdin);
18 //    freopen("culture.out","w",stdout);
19     //in
20     scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
21     for(i=1;i<=n;i++){
22         scanf("%d",&c[i]);
23     }
24     for(i=1;i<=n;i++)
25      for(j=1;j<=n;j++){
26          if(i!=j)mp[i][j]=inf;//预处理 
27      }
28     for(i=1;i<=k;i++)
29      for(j=1;j<=k;j++)
30        scanf("%d",&fl[i][j]);//读入文化兼容情况 
31 
32     for(i=1;i<=m;i++){
33         int u,v,d;
34         scanf("%d%d%d",&u,&v,&d);
35         mp[u][v]=min(mp[u][v],d);//两座城市间若有多条道路，记录最短的即可 
36         mp[v][u]=mp[u][v];//双向图 
37     }
38     for(i=1;i<=n;i++)
39      for(j=1;j<=n;j++){
40          if(fl[c[i]][c[j]]==1)mp[j][i]=inf;//如果文化不兼容，相当于道路不通
41          //刚开始mp[j][i]误打成mp[i][j]，费了半天时间查错 
42          //这部分原本想边读边处理，但是发现会影响到双向距离处理，只好单拿出来了 
43      }
44     //
45     //floyd 不能走的路之前都处理成inf了，直接floyd就行 
46     for(k=1;k<=n;k++)
47       for(i=1;i<=n;i++)
48         for(j=1;j<=n;j++){
49             if(mp[i][k]+mp[k][j]<mp[i][j]){
50                 mp[i][j]=mp[i][k]+mp[k][j]; 
51             }
52         }
53     //
54     if(mp[s][t]==inf)printf("-1");
55     else printf("%d",mp[s][t]);
56 //    fclose(stdin);
57 //    fclose(stdout);
58     return 0;
59     
60 }