poj3662 Telephone Lines

 Konjac's code, WA dropped

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
int n,p,k;int len;
int head,tail;
int lin[1100],next[1100],ver[1100],v[1100];
int d[1100][1100];
int q[201000];bool f[1100]={};
void add(int x,int y,int z)
{
    next[++len]=lin[x];
    ver [len] = y; v [len] = z;
    lin [x] = len;
}
intmain ()
{
    scanf("%d%d%d",&n,&p,&k);
    memset(d,10,sizeof(d));
    for(int i=1;i<=p;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);
        if(x==1)
        {
            for(int j=1;j<=k;j++)
            d[y][j] = 0 ;
        }
        else if(y==1)
        {
            for(int j=1;j<=k;j++)
            d[x][j]=0;
        }
    }
    q[++tail]=1;f[1]=1;
    for(int i=0;i<=k;i++)
        d[1][i]=0;
    while(head++<tail)
    {
        int tmp=q[head];f[tmp]=0;
        for(int i=lin[tmp];i;i=next[i])
        {
            int y=ver[i];
            bool flag=false;
            for(int j=0;j<=k;j++)
            {
                if(j>0)
                    d[y][j]=min(d[y][j-1],d[y][j]);
                if(d[y][j]>max(d[tmp][j],v[i]))
                    d[y][j]=max(d[tmp][j],v[i]),flag=1;
                if(d[y][j]>d[tmp][j-1])
                    d[y][j]=d[tmp][j-1],flag=1;
            }
            if(flag)
                q[++tail]=y;
        }
    }
    printf("%d\n",d[n][k]);
    return 0;
}

 

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=326169292&siteId=291194637