BZOJ 2763 JLOI2011 飞行路线 分层图+堆优化SPFA

题目大意：给定一个无向图，求源点到汇点的最短路，其中有k次机会把边权变为0

非常水的分层图。。话说所谓分层图其实就是多一维的SPFA。。。还起了这么个高大上的名字

这题裸SPFA过不去 要加堆优化 我的堆优化一定是写的有毛病 把heap[top--]=heap[0]改成top--就死活过不去 把魔法森林改一下测试了一下结果居然WA了

总之贴代码

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define M 10100using namespace std;struct abcd{    int to,f,next;}table[100100];typedef pair<int,int> ABCD;ABCD heap[M*10];unsigned short r,h;int head[M],tot,top;int n,m,k,s,t,ans=0x3f3f3f3f;int f[M][11],pos[M][11];void push_up(int t){    while( t>1 && f[heap[t].first][heap[t].second]<f[heap[t>>1].first][heap[t>>1].second] )        swap(heap[t],heap[t>>1]),swap(pos[heap[t].first][heap[t].second],pos[heap[t>>1].first][heap[t>>1].second]),t>>=1;}void insert(ABCD x){    heap[++top]=x;    pos[x.first][x.second]=top;    push_up(top);}void pop(){    pos[heap[1].first][heap[1].second]=0;    heap[1]=heap[top];    //top--;    heap[top--]=heap[0];    pos[heap[1].first][heap[1].second]=1;    int t=2;    while(t<=top)    {        if( f[heap[t].first][heap[t].second]>f[heap[t+1].first][heap[t+1].second] && t<top )            t++;        if( f[heap[t].first][heap[t].second]<f[heap[t>>1].first][heap[t>>1].second] )            swap(heap[t],heap[t>>1]),swap(pos[heap[t].first][heap[t].second],pos[heap[t>>1].first][heap[t>>1].second]),t<<=1;        else            break;    }}void SPFA(){    int i;    ABCD x;    memset(f,0x3f,sizeof f);    insert( make_pair(s,0) );    f[s][0]=0;    while(top)    {        x=heap[1];pop();        for(i=head[x.first];i;i=table[i].next)        {            if( f[x.first][x.second]+table[i].f<f[table[i].to][x.second] )            {                f[table[i].to][x.second]=f[x.first][x.second]+table[i].f;                if(!pos[table[i].to][x.second])                    insert( make_pair(table[i].to,x.second) );                else                    push_up( pos[table[i].to][x.second] );            }            if( x.second<k && f[x.first][x.second]<f[table[i].to][x.second+1] )            {                f[table[i].to][x.second+1]=f[x.first][x.second];                if(!pos[table[i].to][x.second+1])                    insert( make_pair(table[i].to,x.second+1) );                else                    push_up( pos[table[i].to][x.second+1] );            }        }    }    for(i=0;i<=k;i++)        ans=min(ans,f[t][i]);}void add(int x,int y,int z){    table[++tot].to=y;    table[tot].f=z;    table[tot].next=head[x];    head[x]=tot;}int main(){    int i,x,y,z;    cin>>n>>m>>k>>s>>t;s++;t++;    for(i=1;i<=m;i++)        scanf("%d%d%d",&x,&y,&z),x++,y++,add(x,y,z),add(y,x,z);    SPFA();    cout<<ans<<endl;}