JZYZOJ1355 [usaco2007]奶牛赛跑 矩阵乘法 离散化

http://172.20.6.3/Problem_Show.asp?id=1355

 

写的时候本来想离散化，“1000^2的数组放一两个到函数里而已嘛，指定承受得住”，然后没离散化，然后就爆栈了，第一次知道直接爆栈是不报错的，只会运行之后return value 3221225477，学习了orz。

然后重写了一份离散化的……其实我觉得不离散化，数组就在外面声明也未尝不可啊，n<1000，n^3*logn，大概，不会超时吧（宛如那戏台上的老将军，背上插满了flag）。

 

代码

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm>  
#include<cmath>
using namespace std;
const int maxn=210;
int n,m,s,t,siz;
struct nod{int x,y,v;}aa[maxn];
struct mat{
    int e[maxn][maxn];
    mat(){memset(e,63,sizeof(e));}
};mat a;
int bb[maxn]={};
mat mul(mat x,mat y){
    mat z;
    for(int i=1;i<=siz;i++){
        for(int j=1;j<=siz;j++){
            for(int k=1;k<=siz;k++){
                z.e[i][j]=min(z.e[i][j],x.e[i][k]+y.e[k][j]);
            }
        }
    }return z;
}
mat pow(mat x,int k){
    mat z;
    for(int i=1;i<=siz;i++){
        z.e[i][i]=0;
    }
    while(k){
        if(k&1)z=mul(z,x);
        x=mul(x,x);
        k/=2;
    }
    return z;
}
int main(){
    scanf("%d%d%d%d",&n,&m,&s,&t);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&aa[i].v,&aa[i].x,&aa[i].y);
        bb[2*i-1]=aa[i].x;bb[2*i]=aa[i].y;
    }sort(bb+1,bb+1+m*2);
    siz=unique(bb+1,bb+1+m*2)-bb-1;
    s=lower_bound(bb+1,bb+1+siz,s)-bb;
    t=lower_bound(bb+1,bb+1+siz,t)-bb;
    for(int i=1;i<=m;i++){
        aa[i].x=lower_bound(bb+1,bb+1+siz,aa[i].x)-bb;
        aa[i].y=lower_bound(bb+1,bb+1+siz,aa[i].y)-bb;
        a.e[aa[i].x][aa[i].y]=aa[i].v;
        a.e[aa[i].y][aa[i].x]=aa[i].v;
    }
    mat z=pow(a,n);
    printf("%d
",z.e[s][t]);
    return 0;
}