luogu1078 文化之旅

​​http://www.elijahqi.win/archives/930​​​

題目描述

有一位使者要遊曆各國，他每到一個國家，都能學到一種文化，但他不願意學習任何一種文化超過一次（即如果他學習了某種文化，則他就不能到達其他有這種文化的國家）。不同的國家可能有相同的文化。不同文化的國家對其他文化的看法不同，有些文化會排斥外來

文化（即如果他學習了某種文化，則他不能到達排斥這種文化的其他國家）。

現給定各個國家間的地理關系，各個國家的文化，每種文化對其他文化的看法，以及這位使者遊曆的起點和終點（在起點和終點也會學習當地的文化），國家間的道路距離，試求從起點到終點最少需走多少路。

輸入輸出格式

輸入格式：

第一行為五個整數 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，兩個國家之間可能有多條道路）。

輸出格式：

輸出隻有一行，一個整數，表示使者從起點國家到達終點國家最少需要走的距離數（如果無解則輸出-1）。

輸入輸出樣例

輸入樣例#1：

2 2 1 1 2

1 2

0 1

1 0

1 2 10

輸出樣例#1：

-1

輸入樣例#2：

2 2 1 1 2

1 2

0 1

0 0

1 2 10

輸出樣例#2：

10

說明

輸入輸出樣例說明1

由于到國家 2 必須要經過國家 1，而國家 2 的文明卻排斥國家 1 的文明，是以不可能到達國家 2。

輸入輸出樣例說明2

路線為 1 -> 2

【資料範圍】

對于 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<cstdio>
#define N 110
inline int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int ans,h[N],h1[N],n,k,m,s,t,c[N],fa[N],num;bool flag[N];
struct node{
    int y,z,next;
}data[4*N*N];
inline int min(int x,int y){
    return x<y?x:y;
}
bool dfs1(int x,int ans1){
    if (x==s) {
        ans=min(ans,ans1);return true;
    }
    flag[c[x]]=true;
    for (int i=h1[c[x]];i;i=data[i].next){
        int y=data[i].y;flag[y]=true;
    }bool flag1=false;
    if (flag[c[s]]) return false;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y,z=data[i].z;
        if (flag[c[y]]) continue;
        if (fa[x]==y) continue;if (ans1+z>ans) continue;
        fa[y]=x;bool tm=dfs1(y,ans1+z);
        if (tm) flag1=true;
    }
    flag[c[x]]=false;
    for (int i=h1[c[x]];i;i=data[i].next){
        int y=data[i].y;flag[y]=false;
    }
    return flag1;
}
int main(){
    //freopen("1078.in","r",stdin);
    n=read();k=read();m=read();s=read();t=read();
    for (int i=1;i<=n;++i) c[i]=read();
    for (int i=1;i<=k;++i)
        for (int j=1;j<=k;++j) if (read()) data[++num].y=i,data[num].next=h1[j],h1[j]=num;
    for (int i=1;i<=m;++i) {
        int x=read(),y=read(),z=read();
        data[++num].y=y;data[num].next=h[x];data[num].z=z;h[x]=num;
        data[++num].y=x;data[num].next=h[y];data[num].z=z;h[y]=num;
    }ans=0x7fffffff;
    bool tm=dfs1(t,0);
    if (!tm) printf("-1");else printf("%d",ans);
    return 0;
}