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

时间:2021-06-06 06:37:54

文化之旅

思路:

  暴搜,倒搜;

代码:

#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=;
while(Cget>''||Cget<'')Cget=getchar();
while(Cget>=''&&Cget<='')
{
now=now*+Cget-'';
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=;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=;i<=n;i++) in(ci[i]);
int pos;
for(int i=;i<=k;i++)
{
for(int j=;j<=k;j++)in(pos),map_[i][j]=pos;
}
int u,v,w;
for(int i=;i<=m;i++)
{
in(u),in(v),in(w);
edge_add(u,v,w);
}
dfs(t,);
printf("%d\n",ans==INF?-:ans);
return ;
}