P3489 [POI2009] WIE-Hexer

*原题链接*

最短路+状态压缩

不愧是POI的题,看题面知道要求加了一些限制的最短路,看数据范围很容易想到状态压缩。

求解最短路就用堆优化dijkstra好了。

至于状态压缩,我们对原数组再开一维,表示此时"剑的集合",相应的数组也要多开一维。由于此时的最短路有状态的限制,所以我们要用三元组来维护,如果不想写结构体也可以pair<pair<int,int>,int>。输入时存储边上的"怪物集合",以及一个村庄的"铁匠集合",在来到新村庄时或上铁匠集合,在最短路转移时判断能否走,这道题就做完了。

需要一些位运算技巧,详细见代码。

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define PII pair<pair<int,int>,int>
const int N=210,M=(1<<14),INF=0x3f3f3f3f;

int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
	return x*f;
}

int n,m,p,k,f[N][M],village[N],head[N],tot,vis[N][M];
struct node{
	int to,nxt,w,state;
}edge[N*N];
void add(int x,int y,int w,int s){
	edge[++tot].to=y;
	edge[tot].w=w;
	edge[tot].state=s;//怪物集合
	edge[tot].nxt=head[x];
	head[x]=tot;
}

int dijkstra(int s){
	priority_queue<PII,vector<PII>,greater<PII> > q;
	q.push(make_pair(make_pair(0,s),0)),f[s][0]=0;
	while(!q.empty()){
		int t=q.top().first.second,s=q.top().second,cost=q.top().first.first;q.pop();
		if(t==n) return cost;
		if(vis[t][s]) continue;
		vis[t][s]=1,s|=village[t];//别忘了加上这个村庄对应的集合
		for(int i=head[t];i;i=edge[i].nxt){
			int y=edge[i].to;
			if((edge[i].state|s)!=s) continue;//当前state不是s的子集,无法通过这条路
			if(f[y][s|village[y]]>cost) f[y][s|village[y]]=cost+edge[i].w,q.push(make_pair(make_pair(f[y][s|village[y]],y),s));
		}
	}
	return -1;
}

int main(){
	
	n=read(),m=read(),p=read(),k=read();
	for(int i=1;i<=k;i++){
		int x=read(),y=read();
		for(int j=1;j<=y;j++){int s=read();village[x]|=(1<<(s-1));}
	}
	for(int i=1;i<=m;i++){
		int x=read(),y=read(),w=read(),l=read(),state=0;
		for(int j=1;j<=l;j++){int s=read();state|=(1<<(s-1));}
		add(x,y,w,state),add(y,x,w,state);
	}
	memset(f,0x3f,sizeof(f));
	cout<<dijkstra(1)<<endl;
	
	return 0;
}
相关推荐
不去幼儿园16 分钟前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习
Mr_Xuhhh18 分钟前
重生之我在学环境变量
linux·运维·服务器·前端·chrome·算法
Ajiang28247353041 小时前
对于C++中stack和queue的认识以及priority_queue的模拟实现
开发语言·c++
盼海1 小时前
排序算法(五)--归并排序
数据结构·算法·排序算法
网易独家音乐人Mike Zhou5 小时前
【卡尔曼滤波】数据预测Prediction观测器的理论推导及应用 C语言、Python实现(Kalman Filter)
c语言·python·单片机·物联网·算法·嵌入式·iot
‘’林花谢了春红‘’6 小时前
C++ list (链表)容器
c++·链表·list
机器视觉知识推荐、就业指导8 小时前
C++设计模式:建造者模式(Builder) 房屋建造案例
c++
Swift社区8 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展
Kent_J_Truman9 小时前
greater<>() 、less<>()及运算符 < 重载在排序和堆中的使用
算法