ICPC North Central NA Contest 2018 G. Tima goes to Xentopia 最短路变形

一道很好的最短路

d[i][j][k]到i点，经过j条红边，k条蓝边，花费的最少时间是多少。

我们发现边权都是正数，即：不存在负环。

且花费时间满足dijkstra的条件。即：当前最小值是最终最小值。

所以放心的用dij跑就行。复杂度O(n*k1*k2*logm)

一个小优化：题目保证k1*k2<=800  我们可以让较大的在前面开800，较小的在后面开100.这样节约8倍的空间。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 1e5+7;

int head[M],cnt;
void init(){cnt=0,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,val,op;}ee[M*2];
void add(int x,int y,int w,int op){ee[++cnt].nxt=head[x],ee[cnt].to=y,ee[cnt].val=w,ee[cnt].op=op,head[x]=cnt;}

ll d[451][801][101];//从s到i点，经过j条红，k条蓝，最少用时 
bool vs[451][801][101];
int n,m,k1,k2,s,t;
struct node{
	ll dis,id,n1,n2;//s到id的用时，经过n1红，n2蓝 
	bool operator <(const node &r)const
	{
		return dis>r.dis;
	} 
};
void dij()
{
	memset(d,60,sizeof(d));
	ll P=d[0][0][0];
	priority_queue<node>q;
	q.push(node{0,s,0,0});
	d[s][0][0]=0;
	while(!q.empty())
	{
		node tp=q.top();q.pop();
		int u=tp.id,a=tp.n1,b=tp.n2;
		//cout<<u<<"  "<<a<<" "<<b<<endl;
		if(vs[u][a][b]||a>k1||b>k2)continue;
		vs[u][a][b]=1;
		for(int i=head[u];i;i=ee[i].nxt)
		{
			int v=ee[i].to,w=ee[i].val,op=ee[i].op;
			if(d[v][a+(op==1)][b+(op==2)]>d[u][a][b]+w)//更优解 
			{
				d[v][a+(op==1)][b+(op==2)]=d[u][a][b]+w;
				q.push(node{d[v][a+(op==1)][b+(op==2)],v,a+(op==1),b+(op==2)});
			}
		}
	}
	if(d[t][k1][k2]==P)cout<<-1<<endl;
	else cout<<d[t][k1][k2]<<endl;
}
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	cin>>n>>m>>k1>>k2;
  	bool f=false;
  	if(k1<k2)swap(k1,k2),f=true;//让大的在前面，省空间 ,100*100>800
  	for(int i=1;i<=m;i++)
  	{
  		int u,v,w,op;
  		cin>>u>>v>>w>>op;
  		if(f)
  		{
  			if(op==1)op=2;
  			else if(op==2)op=1;
		}
  		add(u,v,w,op),add(v,u,w,op);
	}
	cin>>s>>t;
	dij();
	return 0;
}