92分WA#10
查看原帖
92分WA#10
361965
K2Cr2O7楼主2020/11/5 13:48

代码。

#include<cstdio>
#include<vector>
using namespace std;
const int maxn=104;
bool vis[maxn];
int n,k,m,s,t,u,v,d,res=0x7fffffff,c[maxn],a[maxn][maxn];
struct edge{ int to,w;};
vector < edge > g[maxn];

inline void read(int &x){
	char c=getchar();bool f=0;x=0;
	for(;c<'0'||c>'9';c=getchar())
		if(c=='-')
			f=1;
	for(;c>='0'&&c<='9';c=getchar())
		x=x*10+(c^48);
	x=f?-x:x;
}

void dfs(int node,int len){
	if(len>=res)
		return ;
	if(node==t){  //到了就更新 
        res=len;
        return ;
    }
	for(int i=g[node].size()-1;i>=0;i--){
		edge e=g[node][i];int to=e.to;
		if(vis[c[to]])
			continue;
		vis[c[to]]=1;
		dfs(to,len+e.w);
		vis[c[to]]=0;
	}
}

signed main(){
	read(n),read(k),read(m),read(s),read(t);
    for(int i=1;i<=n;i++)
		read(c[i]);
	for(int i=1;i<=k;i++)
		for(int j=1;j<=k;j++)
			read(a[i][j]);  //文化i排斥文化j
	while(m--){
		read(u),read(v),read(d);
        if(!a[v][u])  //如果通往的国家不排斥现在的文化 
		    g[u].push_back((edge){v,d});
        if(!a[u][v])  //同理 
		    g[v].push_back((edge){u,d});
	}
	vis[c[s]]=1; 
	dfs(s,0);
	printf("%d",res==0x7fffffff?-1:res);
	return 0;
}
2020/11/5 13:48
加载中...