1316 文化之旅 2012年NOIP全国联赛普及组

 

题目描述 Description

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

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

输入描述 Input Description

第一行为五个整数N，K，M，S，T，每两个整数之间用一个空格隔开，依次代表国家个数（国家编号为1到N），文化种数（文化编号为1到K），道路的条数，以及起点和终点的编号（保证S不等于T）；

第二行为N个整数，每两个整数之间用一个空格隔开，其中第i个数Ci，表示国家i的文化为Ci。

接下来的K行，每行K个整数，每两个整数之间用一个空格隔开，记第i行的第j个数为aij，aij= 1表示文化i排斥外来文化j（i等于j时表示排斥相同文化的外来人），aij= 0表示不排斥（注意i排斥j并不保证j一定也排斥i）。

接下来的M行，每行三个整数u，v，d，每两个整数之间用一个空格隔开，表示国家u与国家v有一条距离为d的可双向通行的道路（保证u不等于v，两个国家之间可能有多条道路）。

输出描述 Output Description

输出只有一行，一个整数，表示使者从起点国家到达终点国家最少需要走的距离数（如果无解则输出-1）。

样例输入 Sample Input

输入样例1

2 2 1 1 2

1 2

0 1

1 0

1 2 10

输入样例2

2 2 1 1 2

1 2

0 1

0 0

1 2 10

样例输出 Sample Output

输出样例1

-1

输出样例2

10

数据范围及提示 Data Size & Hint

【输入输出样例1说明】

由于到国家2必须要经过国家1，而国家2的文明却排斥国家1的文明，所以不可能到达国家2。

【输入输出样例2说明】

路线为1 -> 2。

【数据范围】

对于20%的数据，有2≤N≤8，K≤5；

对于30%的数据，有2≤N≤10，K≤5；

对于50%的数据，有2≤N≤20，K≤8；

对于70%的数据，有2≤N≤100，K≤10；

对于100%的数据，有2≤N≤100，1≤K≤100，1≤M≤N2，1≤ki≤K，1≤u,v≤N，1≤d≤1000，S≠T，1 ≤S, T≤N。

 #include<iostream>
 #include<cstdio>
 #include<queue>
 #include<cstring>
 using namespace std;
 const int maxn=0x7fffffff;
 struct node{
     int u,v,next,w;
 }edge[];

 int head[],num=;
 int N,K,M,S,T;
  int flag[];

 void add_edge(int x,int y,int z)
 {
     edge[num].u=x;
     edge[num].v=y;
     edge[num].w=z;
     edge[num].next=head[x];
     head[x]=num++;
 }

 bool hate[][];int country[];
 int dis[],cul[];

 bool vis[];

 void spfa(int s)
 {
     queue<int>q;
     dis[s]=;
     vis[s]=;
     q.push(s);
     flag[s]--;
     while(q.size()!=)
     {
         int p=q.front();
         q.pop();
         vis[p]=;
         for(int i=head[p];i!=-;i=edge[i].next)
         {
             int to=edge[i].v;
             if(dis[to]>dis[p]+edge[i].w&&flag[edge[i].v])
             {
                 flag[edge[i].v]--;
                 dis[to]=dis[p]+edge[i].w;
                 if(vis[to]==)
                 {
                     q.push(to);
                     vis[to]=;
                 }
             }
         }

     }
     if(dis[T]>=)
     printf("-1");
     else
     printf("%d",dis[T]);
 }
 int main()
 {
     memset(head,-,sizeof(head));
     memset(dis,0x7f,sizeof(dis));

     scanf("%d%d%d%d%d",&N,&K,&M,&S,&T);
     for(int i=;i<=N+;i++)
     dis[i]=maxn;
     for(int i=;i<=N;i++)
     {
         dis[i]=maxn;
         scanf("%d",&country[i]);
         cul[country[i]]=i;
     }
     int a;
     for(int i=;i<=K;i++)
     for(int j=;j<=K;j++)
     {
         scanf("%d",&a);
         if(i==j&&a==)
         {
             flag[i]=;
         }
         else
         {
             flag[i]=0x7fffffff;
              hate[cul[i]][cul[j]]=a;
         }
     }
     int u,v,d;
     for(int i=;i<=M;i++)
     {
         scanf("%d%d%d",&u,&v,&d);
         if(!hate[v][u])
         add_edge(u,v,d);
         if(!hate[u][v])
         add_edge(v,u,d);
     }
     spfa(S);
     return ;
 }