noip2017逛公园

https://www.zybuluo.com/ysner/note/1330542

题面

无人不知，无人不晓。

解析

这绝对是(noip2017)得分最难的题目。

(30pts)算法

最短路计数？跑(Dijstra)的同时，顺便(DP)算下方案数就好。
于是设(f[i])表示由(1)到(i)点的方案数。

#define pi pair<int,int>
#define mk make_pair
#define fi first
#define se second
priority_queue<pi,vector<pi>,greater<pi> >Q;
il void Dijstra()
{
  fp(i,1,n) dis[i]=1e9,vis[i]=0;
  dis[1]=0;Q.push(mk(0,1));
  while(!Q.empty())
    {
      re int u=Q.top().se;Q.pop();
      vis[u]=1;
      for(re int i=h[u];i+1;i=e[i].nxt)
      {
        re int v=e[i].to;
        if(dis[v]>dis[u]+e[i].w)
    {
      dis[v]=dis[u]+e[i].w;f[v]=f[u];
      Q.push(mk(dis[v],v));
    }
    else if(dis[v]==dis[u]+e[i].w) (f[v]+=f[u])%=mod;
      }
      while(!Q.empty()&&vis[Q.top().se]) Q.pop();
    }
}

(70pts)算法

考虑到(k)很小，可以把(k)加入状态。
于是设(dp[j][i])表示到达(i)点，路径长度比最短路长(j)的方案数。
看这个状态就知道我们要预处理最短路。
状态都会设，正确转移估计没多少人会了

显然转移式为

[dp[j][u]=sum dp[dis_u+j-e[i].w-dis_v][v] ]

但是题目背景是张图，(DP)时是要考虑转移顺序的。

由于在转移过程中，(j)这一维一定是单调不降的，那么肯定是由(j)小的状态往(j)大的状态转移。
但是这样还不止，你会发现只这么写会过不了某个样例。。。而且这个样例的(k=0)。。。

注意到(j)相等的转移（在最短路上的转移）是有锅的。
其实这一类型的转移，我们都是由(dis)值小的转移到(dis)值大的，这一点看看(30pts)最短路计数的转移就应该明白。
那么这里也一样，给所有点依(dis)排个序，这就是转移的拓扑序。
然后像拓扑排序那样转移就行，由一个节点向与其相连的多个结点转移。

于是就有(70pts)了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define ll long long
#define re register
#define il inline
#define pi pair<int,int>
#define mk make_pair
#define fi first
#define se second
#define fp(i,a,b) for(re int i=a;i<=b;++i)
#define fq(i,a,b) for(re int i=a;i>=b;--i)
using namespace std;
const int N=1e5+100;
int f[55][N],n,m,k,mod,h[N],cnt,dis[N],ans,sta[N];
bool vis[N];
struct Edge{int to,nxt,w;}e[N<<1];
il void add(re int u,re int v,re int w){e[++cnt]=(Edge){v,h[u],w};h[u]=cnt;}
il int gi()
{
  re int x=0,t=1;
  re char ch=getchar();
  while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
  if(ch=='-') t=-1,ch=getchar();
  while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
  return x*t;
}
il void Dijstra()
{
  priority_queue<pi,vector<pi>,greater<pi> >Q;
  fp(i,1,n) dis[i]=1e9,vis[i]=0;
  dis[1]=0;Q.push(mk(0,1));
  while(!Q.empty())
    {
      re int u=Q.top().se;Q.pop();
      vis[u]=1;
      for(re int i=h[u];i+1;i=e[i].nxt)
      {
        re int v=e[i].to;
        if(dis[v]>dis[u]+e[i].w)
    {
      dis[v]=dis[u]+e[i].w;
      Q.push(mk(dis[v],v));
    }
      }
      while(!Q.empty()&&vis[Q.top().se]) Q.pop();
    }
}
il void Dp()
{
  fp(j,0,k)
  fp(o,1,n)
    {
      re int u=sta[o];
    for(re int i=h[u];i+1;i=e[i].nxt)
      {
    re int v=e[i].to;
        if(dis[u]+j-dis[v]+e[i].w>=0&&dis[u]+j-dis[v]+e[i].w<=k)
    (f[dis[u]+j-dis[v]+e[i].w][v]+=f[j][u])%=mod;
      }
    }
}
il bool cmp(re int x,re int y){return dis[x]<dis[y];}
int main()
{
  re int T=gi();
  while(T--)
  {
    memset(h,-1,sizeof(h));memset(f,0,sizeof(f));cnt=0;ans=0;
    n=gi();m=gi();k=gi();mod=gi();
    fp(i,1,m)
      {
    re int u=gi(),v=gi(),w=gi();
    add(u,v,w);
      }
    Dijstra();
    fp(i,1,n) sta[i]=i;sort(sta+1,sta+1+n,cmp);
    f[0][1]=1;
    Dp();
    fp(i,0,k) (ans+=f[i][n])%=mod;
    printf("%d
",ans);
  }
  return 0;
}

(100pts)算法

有(0)边时转移又锅了。
因为会有(dis)和(j)同时相等的点，它们谁先转移，谁后转移需要进一步确定。
仔细想想，发现按拓扑序转移就行了，毕竟只有这样才满足无后效性。

那就进行一遍拓扑排序（只有(0)边提供入度，因为针对(dis)和(j)同时相等的点）。
如果跑完后还有点没进拓扑序，说明有(0)环。
如果有合法路线经过(0)环，就可以(puts("-1"))了（但实际上GGF的数据中，只要有零环就可以puts）。
然后再两重标准给点排序，再照搬(70pts)转移即可。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define ll long long
#define re register
#define il inline
#define pi pair<int,int>
#define mk make_pair
#define fi first
#define se second
#define fp(i,a,b) for(re int i=a;i<=b;++i)
#define fq(i,a,b) for(re int i=a;i>=b;--i)
using namespace std;
const int N=1e5+100;
int f[55][N],n,m,k,mod,h[N],cnt,dis[N],ans,sta[N],d[N],seq[N],top;
bool vis[N],lab[N];
struct Edge{int to,nxt,w;}e[N<<1];
il void add(re int u,re int v,re int w){e[++cnt]=(Edge){v,h[u],w};h[u]=cnt;}
il int gi()
{
  re int x=0,t=1;
  re char ch=getchar();
  while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
  if(ch=='-') t=-1,ch=getchar();
  while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
  return x*t;
}
il void Dijstra()
{
  priority_queue<pi,vector<pi>,greater<pi> >Q;
  fp(i,1,n) dis[i]=1e9,vis[i]=0;
  dis[1]=0;Q.push(mk(0,1));
  while(!Q.empty())
    {
      re int u=Q.top().se;Q.pop();
      vis[u]=1;
      for(re int i=h[u];i+1;i=e[i].nxt)
      {
        re int v=e[i].to;
        if(dis[v]>dis[u]+e[i].w)
    {
      dis[v]=dis[u]+e[i].w;
      Q.push(mk(dis[v],v));
    }
      }
      while(!Q.empty()&&vis[Q.top().se]) Q.pop();
    }
}
il void Dp()
{
  fp(j,0,k)
  fp(o,1,n)
    {
      re int u=sta[o];
    for(re int i=h[u];i+1;i=e[i].nxt)
      {
    re int v=e[i].to;
        if(dis[u]+j-dis[v]+e[i].w>=0&&dis[u]+j-dis[v]+e[i].w<=k)
    (f[dis[u]+j-dis[v]+e[i].w][v]+=f[j][u])%=mod;
      }
    }
}
il int Toposort()
{
  queue<int>Q;
  fp(i,1,n) if(!d[i]) Q.push(i);
  while(!Q.empty())
    {
      re int u=Q.front();Q.pop();seq[u]=++top;
      for(re int i=h[u];i+1;i=e[i].nxt)
    if(!e[i].w)
      {
        re int v=e[i].to;
        if(!--d[v]) Q.push(v);
      }
    }
  fp(i,1,n) if(d[i]) return 0;
  return 1;
}
il bool cmp(re int x,re int y){return dis[x]<dis[y]||(dis[x]==dis[y]&&seq[x]<seq[y]);}
int main()
{
  re int T=gi();
  while(T--)
  {
    memset(h,-1,sizeof(h));memset(f,0,sizeof(f));memset(d,0,sizeof(d));
    cnt=0;ans=0;top=0;
    n=gi();m=gi();k=gi();mod=gi();
    fp(i,1,m)
      {
    re int u=gi(),v=gi(),w=gi();
    add(u,v,w);if(!w) ++d[v];
      }
    if(!Toposort()) {puts("-1");continue;}
    Dijstra();
    fp(i,1,n) sta[i]=i;sort(sta+1,sta+1+n,cmp);
    f[0][1]=1;
    Dp();
    fp(i,0,k) (ans+=f[i][n])%=mod;
    printf("%d
",ans);
  }
  return 0;
}

简洁的正解

想想上面的转移顺序都是些什么鬼。

• 由(j)小的转移到(j)大的
• 由(dis)小的转移到(dis)大的
• 由拓扑序小的转移到拓扑序大的

注意到这个玩意其实很像由起点向终点转移。但是不可行的原因是有后效性。
想想拓扑排序，只要没有入度就没有后效性。
所以我们要在处理一个点的信息前把其前驱的点的信息处理完。

这里不是(DAG)，不能拓扑排序，怎么办呢？
可以倒着记忆化搜索，这样就能在前驱的信息算完后再算自己的信息。
状态中只要有当前位置、(k)值就是对的。
有(0)环的情况就是搜一个点的前驱时搜到了自己，记个(vis)判一下即可。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define ll long long
#define re register
#define il inline
#define pi pair<int,int>
#define mk make_pair
#define fi first
#define se second
#define fp(i,a,b) for(re int i=a;i<=b;++i)
#define fq(i,a,b) for(re int i=a;i>=b;--i)
using namespace std;
const int N=2e5+100;
int f[55][N],n,m,K,mod,h[N],cnt=1,dis[N],ans;
bool vis[N],use[55][N];
struct Edge{int to,nxt,w;}e[N<<1];
il void add(re int u,re int v,re int w){e[++cnt]=(Edge){v,h[u],w};h[u]=cnt;}
il int gi()
{
  re int x=0,t=1;
  re char ch=getchar();
  while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
  if(ch=='-') t=-1,ch=getchar();
  while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
  return x*t;
}
il void Dijstra()
{
  priority_queue<pi,vector<pi>,greater<pi> >Q;
  fp(i,1,n) dis[i]=1e9,vis[i]=0;
  dis[1]=0;Q.push(mk(0,1));
  while(!Q.empty())
    {
      re int u=Q.top().se;Q.pop();
      vis[u]=1;
      for(re int i=h[u];i+1;i=e[i].nxt)
    if(!(i&1))
      {
        re int v=e[i].to;
        if(dis[v]>dis[u]+e[i].w)
    {
      dis[v]=dis[u]+e[i].w;
      Q.push(mk(dis[v],v));
    }
      }
      while(!Q.empty()&&vis[Q.top().se]) Q.pop();
    }
}
il int dfs(re int k,re int u)
{
  if(k<0||k>K) return 0;
  if(use[k][u]) return use[k][u]=0,-1;
  if(~f[k][u]) return f[k][u];
  use[k][u]=1;
  re int s=0;
  for(re int i=h[u];i+1;i=e[i].nxt)
    if(i&1)
    {
      re int v=e[i].to;
      re int x=dfs(dis[u]+k-e[i].w-dis[v],v);
      if(x==-1) return use[k][u]=0,-1;
      (s+=x)%=mod;
    }
  use[k][u]=0;
  if(u==1&&k==0) (++s)%=mod;
  return f[k][u]=s;
}
il int work()
{
  fp(i,0,K)
    {
      re int x=dfs(i,n);
      if(x==-1) return -1;
      (ans+=x)%=mod;
    }
  return ans;
}
int main()
{
  re int T=gi();
  while(T--)
  {
    memset(h,-1,sizeof(h));memset(f,-1,sizeof(f));
    cnt=1;ans=0;
    n=gi();m=gi();K=gi();mod=gi();
    fp(i,1,m)
      {
        re int u=gi(),v=gi(),w=gi();
        add(u,v,w);add(v,u,w);
      }
    Dijstra();
    printf("%d
",work());
  }
  return 0;
}