习题日常第二十三练

1[BJWC2012]冻结（题目链接）

这个题是一个多层图的模板题。注意多层图之间的每一层是有方向的，建图的时候需要特别注意。代码如下。

#include<cmath>
#include <iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#include<vector>
#include<set>
#include<map>
#include<cstring>
#include<math.h>
#include<stack>
#include<algorithm>
#include<queue>
#include<bitset>
#include<sstream>
#define  ll long long int
const ll mod=998244353;
using namespace std;
struct node
{
    int to,next,w;
}t[400010];
int a[200010],vis[200010],dis[200010],n,m,tot=0,k;
void add(int x,int y,int z)
{
    t[++tot].to=y;
    t[tot].next=a[x];
    t[tot].w=z;
    a[x]=tot;
    return;
}
priority_queue<pair<int,int> >p;
void dist()
{
    memset(vis,0,sizeof(vis));
    dis[1]=0;
    p.push(make_pair(0,1));
    while(!p.empty())
    {
        int x=p.top().second;
        p.pop();
        if(vis[x])
            continue;
        vis[x]=1;
        for(int i=a[x];i!=-1;i=t[i].next)
        {
            if(dis[t[i].to]>dis[x]+t[i].w)
            {
                dis[t[i].to]=dis[x]+t[i].w;
                p.push(make_pair(-dis[t[i].to],t[i].to));
            }
        }
    }
    return;
}
int main()
{
    ios::sync_with_stdio(false);
    ll i,j;
    cin>>n>>m>>k;
    for(i=0;i<=n*(k+1)+1;i++)
    {
        a[i]=-1;
        dis[i]=1e8;
    }
    for(i=0;i<m;i++)
    {
        int x,y,z;
        cin>>x>>y>>z;
        for(j=0;j<=k;j++)
        {
            add(j*n+x,j*n+y,z);
            add(j*n+y,j*n+x,z);
            add(j*n+x,(j+1)*n+y,z/2);
            add(j*n+y,(j+1)*n+x,z/2);
        }
    }
    dist();
    int ans=1e8;
    for(i=0;i<=k;i++)
    {
        ans=min(ans,dis[i*n+n]);
    }
    cout<<ans;
  return 0;
}

2飞行路线（题目链接）

这个题和上个题一样，也是多层图的问题，但是数据规模更大，所以数组开空间的时候一定要尽可能的大。代码如下。

#include<cmath>
#include <iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#include<vector>
#include<set>
#include<map>
#include<cstring>
#include<math.h>
#include<stack>
#include<algorithm>
#include<queue>
#include<bitset>
#include<sstream>
#define  ll long long int
const ll mod=998244353;
using namespace std;
struct node
{
    ll to,next,w;
}t[800010];
ll a[400010],vis[400010],dis[400010],n,m,tot=0,k,s,z;
void add(ll x,ll y,ll z)
{
    t[++tot].to=y;
    t[tot].next=a[x];
    t[tot].w=z;
    a[x]=tot;
    return;
}
priority_queue<pair<int,int> >p;
void dist()
{
    memset(vis,0,sizeof(vis));
    dis[s]=0;
    p.push(make_pair(0,s));
    while(!p.empty())
    {
        ll x=p.top().second;
        p.pop();
        if(vis[x])
            continue;
        vis[x]=1;
        for(int i=a[x];i!=-1;i=t[i].next)
        {
            if(dis[t[i].to]>dis[x]+t[i].w)
            {
                dis[t[i].to]=dis[x]+t[i].w;
                p.push(make_pair(-dis[t[i].to],t[i].to));
            }
        }
    }
    return;
}
int main()
{
    /*ios::sync_with_stdio(false);*/
    ll i,j;
    /*cin>>n>>m>>k>>s>>z;*/
    scanf("%lld%lld%lld%lld%lld",&n,&m,&k,&s,&z);
    for(i=0;i<=n*(k+1)+1;i++)
    {
        a[i]=-1;
        dis[i]=1e12;
    }
    for(i=0;i<m;i++)
    {
        ll x,y,z;
        scanf("%lld%lld%lld",&x,&y,&z);
        for(j=0;j<=k;j++)
        {
            add(j*n+x,j*n+y,z);
            add(j*n+y,j*n+x,z);
            add(j*n+x,(j+1)*n+y,0);
            add(j*n+y,(j+1)*n+x,0);
        }
    }
    dist();
    ll ans=1e12;
    for(i=0;i<=k;i++)
    {
        ans=min(ans,dis[i*n+z]);
    }
    printf("%lld",ans);
  return 0;
}