luoguP1078 文化之旅

题目描述

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

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

输入输出格式

输入格式：

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

NOIP 2012 普及组 第四题

分析
思路有两个但是我还是没有想出floyd的暴力写法无辜被卡了两次；放弃了只好写Dijkstra事实上优先队列优化的dijkstra的效率和SPFA差不多再说这题的数据小而且水QwQ之后的两个代码就都是有Dijkstra写的如果有人用SPFA实现了还请发我借鉴借鉴QAQ
思路一
使用倒着求最短路，然后从起点开始爆搜剪枝；
剪枝就是到当前的点的路程sum+符合条件的边(u->v)+d[v](v到终点的距离)
代码

#include<bits/stdc++.h>
#define INF 1e9
#define N 105
#define M 10005
using namespace std;
int n,m,k,s,t;
struct node{ 
    int vet,val,Next; 
}g[M<<1];
int cnt,head[N];
void add(int x,int y,int z){
    cnt++;
    g[cnt]=(node){y,z,head[x]};
    head[x]=cnt;
}//边表结构体存储比较方便简洁
struct edge{
    int d,u;
    bool operator<(const edge&rhs)const{
        return d>rhs.d;
    }
};//优先队列
int c[N],ct[N][N],d[N],done[N];
void Dijkstra(int s){
    priority_queue<edge>q;
    for(int i=1;i<=n;i++)d[i]=INF;
	d[s]=0;
    q.push(edge{0,s});
    while(!q.empty()){
        edge x=q.top();q.pop();
        int u=x.u;
        if(done[u])continue;
        done[u]=1;
        for(int i=head[u];i;i=g[i].Next){
            int v=g[i].vet;
            if(d[v]>d[u]+g[i].val){
                d[v]=d[u]+g[i].val;
                q.push((edge){d[v],v});
            }
        }
    }
}//标准的Dijkstra
int been[M],vis[N],tot,ans=INF;
bool check(int x){
    for(int i=1;i<=tot;i++)
        if(ct[been[i]][c[x]])return false;
    return true;
}//判断冲突的函数
void dfs(int x,int sum){
    if(x==t)ans=min(ans,sum);
    if(sum>=ans)return;
    int f=1,temp=tot;vis[c[x]]=1;
    for(int i=head[x];i;i=g[i].Next){
        int v=g[i].vet;
        if(!ct[c[v]][c[x]]&&!vis[c[v]]&&check(v)&&d[v]+sum+g[i].val<ans){//核心操作剪枝
            f=0;
            been[++tot]=c[v];
            dfs(v,sum+g[i].val);
        }
    }
    tot=temp;//还原现场
    if(f)vis[c[x]]=0;
}
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",&ct[i][j]);	
    for(int i=1;i<=m;i++){
    	int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
    	add(x,y,z);add(y,x,z);
    }//边表连边
    Dijkstra(t);
    dfs(s,0);
    if(ans!=INF)printf("%d",ans);
    else printf("-1");
    return 0;
} 

思路二：
爆用STL好吧我瞎说的；看代码注释吧其实就是爆搜

#include<bits/stdc++.h>
#define N 105
#define INF 1e9
using namespace std;
struct edge{
    int from,to,dist;
};//向量存储
struct node{ 
    int d,u;
    set <int> r;//集合，其实想用bool结果不会写；
    bool operator<(const node&rhs)const{
        return d>rhs.d;
    }//重载运算符
};
int n,m,k,s,t,ans,tot=-1;
int d[N],a[N][N],c[N];
vector <edge> ed;
vector <int> g[N];
set <int> r1;//存储排斥起点的点
priority_queue <node> q;
void ins(node q,int x){
    q.r.insert(x);
    for(int i=1;i<=k;i++)
        if(a[i][x])q.r.insert(i);
}//存入集合
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",&a[i][j]);
    for(int i=1;i<=m;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        ed.push_back((edge){x,y,z});
        ed.push_back((edge){y,x,z});
        g[x].push_back(++tot);
        g[y].push_back(++tot);
    }//存边
    r1.insert(c[s]);
    for(int i=1;i<=k;i++)
        if(a[i][s])r1.insert(i);//处理起点
    for(int i=1;i<=n;i++)d[i]=INF;
    d[s]=0;q.push((node){0,s,r1});
    while(!q.empty()){
        node x=q.top();q.pop();
        int u=x.u;
        if(x.d!=d[u])continue;
        for(int i=0;i<g[u].size();i++){
            edge e=ed[g[u][i]];
            node tmp=x;
            if(!tmp.r.count(c[e.to])){//集合的一个函数（判断是否有这个元素）
                if(d[e.to]>d[u]+e.dist){
                    d[e.to]=d[u]+e.dist;
                    ins(tmp,c[e.to]);//该题核心操作自行理解ovo
                    q.push((node){d[e.to],e.to,tmp.r});
                }
            }
        }
    }
    if(d[t]==INF)printf("-1");
    else printf("%d",d[t]);
    return 0;
}

卡点
首先错误的想法Floyd卡了很久，其次尝试新的的写法也是会有错误；
心得
写的题目多了自然有好多的写法倒是后就要找到最适合自己的一种了