【JZOJ 3430】 DY引擎

BOSS送给小唐一辆车。小唐开着这辆车从PKU出发去ZJU上课了。
众所周知，天朝公路的收费站超多的。经过观察地图，小唐发现从PKU出发到ZJU的所有路径只会有N（2<=N<=300）个不同的中转点，其中有M（max(0, N-100) <=M<=N）个点是天朝的收费站。N个中转点标号为1…N，其中1代表PKU，N代表ZJU。中转点之间总共有E（E<=50,000）条双向边连接。
每个点还有一个附加属性，用0/1标记，0代表普通中转点，1代表收费站。当然，天朝的地图上面是不会直接告诉你第i个点是普通中转点还是收费站的。地图上有P（1<=P<=3,000）个提示，用[u, v, t]表示：[u, v]区间的所有中转点中，至少有t个收费站。数据保证由所有标记得到的每个点的属性是唯一的。
车既然是BOSS送的，自然非比寻常了。车子使用了世界上最先进的DaxiaYayamao引擎，简称DY引擎。DY引擎可以让车子从U瞬间转移到V，只要U和V的距离不超过L（1<=L<=1,000,000），并且U和V之间不能有收费站（小唐良民一枚，所以要是经过收费站就会停下来交完钱再走）。
DY引擎果然是好东西，但是可惜引擎最多只能用K（0<=K<=30）次。
对于100%的数据保证：
2<=N<=300，max(0, N-100) <=M<=N，E<=50,000，1<=P<=3,0001<=L<=1,000,000，0<=K<=30

Analysis

差分约束系统的第一题。
若有疑问可以看我写的差分约束系统。

Code

#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=310,M=50010,INF=2139062143;
int n,m,e,p,l,lim,s[N],dis[35][N],a[N][N],b[N][N],c[N][N],f[N][N],ma[N][N],mb[N][N];
bool bz[N],tp[N],lyd[N][N];
queue<int> q;
void spfa0()
{
 memset(s,128,sizeof(s));
 s[0]=0;
 q.push(0);
 bz[0]=1;
 while(!q.empty())
 {
 int u=q.front();q.pop();
 bz[u]=0;
 fo(i,1,b[u][0])
 {
 int v=b[u][i];
 if(s[u]+mb[u][v]>s[v])
 {
 s[v]=s[u]+mb[u][v];
 if(!bz[v])
 {
 bz[v]=1;
 q.push(v);
 }
 }
 }
 }
}
void spfa(int k)
{
 memset(bz,0,sizeof(bz));
 bz[1]=1;
 dis[k][1]=0;
 q.push(1);
 while(!q.empty())
 {
 int u=q.front();q.pop();
 bz[u]=0;
 fo(i,1,a[u][0])
 {
 int v=a[u][i];
 if(dis[k][u]+ma[u][v]<dis[k][v])
 {
 dis[k][v]=dis[k][u]+ma[u][v];
 if(!bz[v])
 {
 bz[v]=1;
 q.push(v);
 }
 }
 }
 if(k)
 fo(i,1,c[u][0])
 {
 int v=c[u][i];
 if(dis[k-1][u]<dis[k][v])
 {
 dis[k][v]=dis[k-1][u];
 if(!bz[v])
 {
 bz[v]=1;
 q.push(v);
 }
 }
 }
 }
}
int main()
{
 scanf("%d %d %d %d %d %d",&n,&m,&e,&p,&l,&lim);
 int u,v,w;
 memset(ma,127,sizeof(ma));
 fo(i,1,e)
 {
 scanf("%d %d %d",&u,&v,&w);
 if(w<ma[u][v]) ma[u][v]=ma[v][u]=w;
 a[u][++a[u][0]]=v;
 a[v][++a[v][0]]=u;
 }
 memcpy(f,ma,sizeof(ma));
 fo(i,1,p)
 {
 scanf("%d %d %d",&u,&v,&w);u--;
 mb[u][v]=w;
 b[u][++b[u][0]]=v;
 }
 fo(i,1,n)
 {
 b[i][++b[i][0]]=i-1;
 mb[i][i-1]=-1;
 b[i-1][++b[i-1][0]]=i;
 mb[i-1][i]=0;
 }
 spfa0();
 fo(i,1,n)
 if(s[i]-s[i-1]) tp[i]=1;
 fo(k,1,n)
 if(!tp[k])
 fo(i,1,n)
 if(i!=k)
 fo(j,1,n)
 if(k!=j && i!=j)
 {
 if(f[i][k]!=INF && f[k][j]!=INF && f[i][k]+f[k][j]<f[i][j]) f[i][j]=f[i][k]+f[k][j];
 if(f[i][j]<=l && !lyd[i][j])
 {
 c[i][++c[i][0]]=j;
 c[j][++c[j][0]]=i;
 lyd[i][j]=lyd[j][i]=1;
 }
 }
 memset(dis,127,sizeof(dis));
 fo(i,0,lim) spfa(i);
 printf("%d",dis[lim][n]);
 return 0;
}