【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;
}