洛谷 P1078 文化之旅

P1078 文化之旅

题目描述

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

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

输入输出格式

输入格式：

第一行为五个整数 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：

输出样例#1：

-1

输入样例#2：

输出样例#2：

说明

输入输出样例说明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

NOIP 2012 普及组第四题

好吧，PJ第四题，感觉好难，首先想到Floyd算法，后来发现文化矛盾啥的很烦人，索性就懒得写了。

用STL队列bfs结果WA了（表示不大喜欢bfs……）。

换成DFS瞬间AC，还是DFS懂我（哈哈）。

注意剪枝。

有个问题想问一下，为啥vi[i]不要回溯的，一回溯就TLE……哪位大神能解释下???

AC代码如下：

#include<cstdio>
using namespace std;
const int N=100+5;
const int K=100+5;
const int M=N*N;
const int INF=N*1000+5;
int ans=INF;
int a[N][N],c[N],n,k,m,s,t,o,u,v;
bool ok[K][K],vi[K];
inline int min(int x,int y)
{
    return x<y?x:y;
    }
void dfs(int now,int cost)
{
	if(now==t) {
	ans=min(ans,cost);return;
	}
	if(cost>=ans) return;
	for(int i=1;i<=n;i++)
	if(a[now][i])
	{
		bool flag=0;
		for(int j=1;j<=k;j++)
		if(vi[j]&&ok[c[i]][j]) {flag=1;break;}
		if(flag) continue;
		vi[c[i]]=1; //就是这里，之后不用回溯？？？？回溯就TLE！！！
		dfs(i,cost+a[now][i]);
	}
	return;
}
int main()
{
	scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
	for(int i=1;i<=n;i++)
	scanf("%d",&c[i]);
	for(int i=1;i<=k;i++)
	for(int j=1;j<=k;j++)
	{
		scanf("%d",&o);
		if(o) ok[i][j]=1;
		if(i==j) ok[i][j]=1;
	}
	for(int i=1;i<=m;i++)
	{
	scanf("%d%d%d",&u,&v,&o);
	if(o<a[u][v]||a[u][v]==0) a[u][v]=a[v][u]=o;
	}
	vi[c[s]]=1;
	dfs(s,0);
	if(ans==INF) printf("-1");
	else printf("%d",ans);
	return 0;
 }