1. 程式人生 > >2018-南京網絡賽icpc-L題(分層最短路)

2018-南京網絡賽icpc-L題(分層最短路)

cpp ans main -- pri long long cst pre return

題意:給你n個點,m條邊的有向帶權圖,然後你每次可以選<=k條邊的邊權變成0,問你1到n的最短路;

解題思路:這道題基本上就是原題了呀,bzoj2763(無向圖),解法就是拆點跑分層的最短路,比如這道題,你用了一次變為0,就相當於進入了下一個層次;

我們把每個點都拆成k個層次點,每個相同層次的點按輸入的邊權連接,每個點可以向它能連接到的點的下一個層次連接一條邊權為0的邊;

意思就是:輸入x到y權值為w的邊

    for(int i=1;i<=m;i++)
        {
            scanf("%lld%lld%lld",&x,&y,&w);
            for(int j=0;j<=k;j++)
            {
                add(x+j*n,y+j*n,w);同一層次的點;
                if(j!=k)
                add(x+j*n,y+(j+1)*n,0);不同層次的點,可以向它能連接的下一個層次連一個邊權為0;
            }
        }
完整代碼:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
#define ll long long
using namespace std;
const int maxm=3000500;
const ll inf=0x7f7f7f7f;
const int maxn=100500;
struct node
{
ll num;
ll dist;
node(ll _num,ll _dist):num(_num),dist(_dist){}
friend bool operator<(node a,node b)
{
return a.dist>b.dist;
}
};
struct Edge
{
ll next;
ll to;
ll w;
}edge[maxm*2];
int head[maxm];
ll dist[maxm];
int cnt;
int n,m,k;
void add(ll u,ll v,ll w)
{
edge[cnt].next=head[u];
edge[cnt].to=v;
edge[cnt].w=w;
head[u]=cnt++;
}
void dij(int u)
{
priority_queue<node>q;
memset(dist,inf,sizeof(dist));
dist[u]=0;
q.push(node(u,dist[u]));
while(!q.empty())
{
node now=q.top();
//cout<<now.num<<" "<<now.dist<<endl;
q.pop();
int x=now.num;
for(int i=head[x];i!=-1;i=edge[i].next)
{
int v=edge[i].to;
if(dist[v]>dist[x]+edge[i].w)
{
dist[v]=dist[x]+edge[i].w;
q.push(node(v,dist[v]));
}
}
}
}
int main()
{
int tt;
ll x,y,w;
scanf("%d",&tt);
while(tt--)
{
memset(head,-1,sizeof(head));cnt=0;
scanf("%d%d%d",&n,&m,&k);
if(k>=m)
{
printf("0\n");
continue;
}
for(int i=1;i<=m;i++)
{
scanf("%lld%lld%lld",&x,&y,&w);
for(int j=0;j<=k;j++)
{
add(x+j*n,y+j*n,w);
if(j!=k)
add(x+j*n,y+(j+1)*n,0);
}
}
dij(1);
ll ans=inf;
for(int i=0;i<=k;i++)
{
ans=min(ans,dist[n+i*n]);
}
printf("%lld\n",ans);
}
}

2018-南京網絡賽icpc-L題(分層最短路)