#P0761. [NOIP2012普及组] 文化之旅

题目描述

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

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

输入格式

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

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

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

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

输出格式

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

输入数据 1

2 2 1 1 2 
1 2 
0 1 
1 0 
1 2 10

Copy

输出数据 1

-1

Copy

输入数据 2

2 2 1 1 2 
1 2 
0 1 
0 0 
1 2 10

Copy

输出数据 2

10

Copy

提示

输入输出样例说明11

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

输入输出样例说明22

路线为11 ->22

【数据范围】

对于 100%的数据，有2≤N≤1002≤N≤100

1≤K≤1001≤K≤100

1≤M≤N^21≤M≤N2

1≤k_i≤K1≤ki​≤K

1≤u, v≤N1≤u,v≤N

1≤d≤1000,S≠T,1≤S,T≤N1≤d≤1000,S=T,1≤S,T≤N

NOIP 2012 普及组 第四题

代码:

#include <cstdio>
#include <cstring>
int n,k,m,s,t,country[105],first[20005],next[20005],v[20005],w[20005];
int wh[105][105],cnt,head,tail,q[100005],dis[105];
int q1[100005][105];
using namespace std;
inline int read()
{
    int f=1,x=0;
    char ch=getchar();
    if (ch=='-')
    {
        f=-1;
        ch=getchar();
    }
    while ((ch<'0')||(ch>'9')) ch=getchar();
    while ((ch>='0')&&(ch<='9'))
    {
        x=x*10+ch-48;
        ch=getchar();
    }
    return f*x;
}
inline void spfa(int s)
{
    memset(dis,0x3f3f3f3f,sizeof(dis));//dis数组记录的是s点到所有点的距离 
    dis[s]=0;
    head=0;
    tail=1;
    q[1]=s;
    q1[1][0]=1;//q1[i][0]记录的是知道了多少种文化 
    q1[1][1]=country[s];
    while (head<tail)
    {
        head=head%100000+1; 
        for (int i=first[q[head]];i;i=next[i])
        {
            int k=v[i];//k为目标点 
            bool bo=0;
            for (int j=1;j<=q1[head][0];j++)//判断目标点可否到达 
              if (wh[country[k]][q1[head][j]]||(country[k]==q1[head][j]))
              {
              	bo=1;
              	break;
              }
            if (bo) continue;//不可到达，则跳过 
            if (dis[q[head]]+w[i]<dis[k])//如果可以使距离变短的话，就加入 
            {
                dis[k]=dis[q[head]]+w[i];
                tail=tail%100000+1;
                q[tail]=k;
                q1[tail][0]=q1[head][0]+1;
                for (int j=1;j<=q1[head][0];j++)
                  q1[tail][j]=q1[head][j];
                q1[tail][q1[tail][0]]=country[k];
            }
        }
    }//下面判断输出 
    if (dis[t]==0x3f3f3f3f)
    {
        printf("-1\n");
        return;
    } else
    {
        printf("%d\n",dis[t]);
        return;
    }
}
int main()
{
    n=read(),k=read(),m=read(),s=read(),t=read();
    for (int i=1;i<=n;i++) country[i]=read();
    for (int i=1;i<=k;i++)
      for (int j=1;j<=k;j++) wh[i][j]=read();
    for (int i=1;i<=m;i++)
    {
        int x,y,z;
        x=read(),y=read(),z=read();
        next[++cnt]=first[x];
        first[x]=cnt;
        v[cnt]=y;
        w[cnt]=z;
        next[++cnt]=first[y];
        first[y]=cnt;
        v[cnt]=x;
        w[cnt]=z;
    }
    spfa(s);
    return 0;
}