AC日记——文化之旅 洛谷 P1078

#include <bits/stdc++.h>
using namespace std;
#define maxn 105
#define maxm 200005
#define INF 0x7fffffff
int n,k,m,s,t,E[maxm],V[maxm],head[maxn];
int cnt,W[maxm],num[maxn],ci[maxn],ans=INF;
bool map_[maxn][maxn],if_[maxn];
inline void in(int &now)
{
char Cget=getchar();now=0;
while(Cget>'9'||Cget<'0')Cget=getchar();
while(Cget>='0'&&Cget<='9')
{
now=now*10+Cget-'0';
Cget=getchar();
}
}
inline void edge_add(int u,int v,int w)
{
E[++cnt]=head[u],V[cnt]=v,W[cnt]=w,head[u]=cnt;
E[++cnt]=head[v],V[cnt]=u,W[cnt]=w,head[v]=cnt;
}
inline bool check(int now)
{
for(int i=1;i<=k;i++) if(map_[now][i]&&num[ci[i]]) return true;
return false;
}
void dfs(int now,int dis)
{
if(dis>=ans) return;
if(now==s)
{
ans=dis;return;
}
if_[now]=true,num[ci[now]]++;
for(int i=head[now];i;i=E[i])
{
if(if_[V[i]]||check(ci[V[i]])) continue;
dfs(V[i],dis+W[i]);
}
if_[now]=false,num[ci[now]]--;
}
int main()
{
in(n),in(k),in(m),in(s),in(t);
for(int i=1;i<=n;i++) in(ci[i]);
int pos;
for(int i=1;i<=k;i++)
{
for(int j=1;j<=k;j++)in(pos),map_[i][j]=pos;
}
int u,v,w;
for(int i=1;i<=m;i++)
{
in(u),in(v),in(w);
edge_add(u,v,w);
}
dfs(t,0);
printf("%d\n",ans==INF?-1:ans);
return 0;
}