QAQ第一反应是dij,把走过的时间和耗费的体力放在结构体里。。但是死活TLE并且MLE
好吧,去翻正解,是spfa
哇哇,我IE什么我的脑回路就是和正解对不上。。
其实可以不用记录走过的时间,记录奇数偶数,用0和1表示偶数和奇数
很关键的一点,0和1的转化可以用异或来解决,QAQsummer说加一取模也可以,但是慢
每次还要分直接走过还是先停一秒再走,是两种状态,所以dis和vis都开二维,第一位表示点,第二维表示状态(走还是停)
还是看代码吧:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn1=5000+10,maxn2=30000+10;
int cnt,n,m,ans;
int w[maxn1],s[maxn1],color[maxn1],fist[maxn1],nxt[maxn2],dis[maxn1][2];
bool vis[maxn1][2];
struct hh
{
int f,t,v;
}e[maxn2];
struct lxt
{
int num,x;
};
queue<lxt>q;
void build(int f,int t,int v)
{
e[++cnt]=(hh){f,t,v};
nxt[cnt]=fist[f];
fist[f]=cnt;
}
int muti(int now,int i,int j)
{
int coli=color[i],colj=color[j];
int delta=max(w[i],w[j])-min(w[i],w[j]);
if(now) swap(coli,colj);
if(coli==colj) return 0;
else if(coli) return delta;
else return -delta;
}
void spfa()
{
vis[1][0]=true;
dis[1][0]=0;
q.push((lxt){1,0});
while(!q.empty())
{
lxt u=q.front();
q.pop();
vis[u.num][u.x]=false;
for(int i=fist[u.num];i!=-1;i=nxt[i])
{
int v=e[i].t;
if(dis[v][u.x^1]>dis[u.num][u.x]+max(e[i].v+muti(u.x,u.num,v),0))
{
dis[v][u.x^1]=dis[u.num][u.x]+max(e[i].v+muti(u.x,u.num,v),0);
if(!vis[v][u.x^1])
{
vis[v][u.x^1]=true;
q.push((lxt){v,u.x^1});
}
}
}
if(dis[u.num][u.x^1]>dis[u.num][u.x]+s[u.num]*(color[u.num]^u.x))
{
dis[u.num][u.x^1]=dis[u.num][u.x]+s[u.num]*(color[u.num]^u.x);
if(!vis[u.num][u.x^1])
{
vis[u.num][u.x^1]=true;
q.push((lxt){u.num,u.x^1});
}
}
}
}
int main()
{
memset(dis,0x3f,sizeof(dis));
memset(fist,-1,sizeof(fist));
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) scanf("%d",&color[i]);
for(int i=1;i<=n;++i) scanf("%d",&w[i]);
for(int i=1;i<=n;++i) scanf("%d",&s[i]);
for(int i=1;i<=m;++i)
{
int u,v,k;
scanf("%d%d%d",&u,&v,&k);
build(u,v,k);
}
spfa();
ans=min(dis[n][0],dis[n][1]);
printf("%d",ans);
return 0;
}