poj 3613 floyd

来源:互联网 发布:java webservice demo 编辑:程序博客网 时间:2024/06/05 23:00


题意: 求进过路径数量为k的已知起点的最短路 ;

先进行在线离散化。

然后Floyd  矩阵快速幂。

注意 如果结构体中数组大小开太大 (大概极限是350乘350)的话不能动态使用,就是将结构体导入导出函数,需要在外部声明一个常量的结构体,对其进行改变。


//#include <bits/stdc++.h>#include <map>#include <cstdio>#include <algorithm>#include <cstring>#include <iostream>using namespace std ;const int inf = 0x3f3f3f3f ;map<int , int> pq ;int pq_num ;struct Matrix{    int m[250][250] ;    void init(){        memset(m , 0x3f , sizeof(inf)) ;    }    Matrix operator * (const Matrix & temp) const {        Matrix c ;        memset(c.m , inf , sizeof(c.m)) ;        for(int i = 1 ; i < pq_num ; i ++ ){            for(int j = 1 ; j < pq_num ; j ++ ){                for(int k = 1 ; k < pq_num ; k ++ )                    c.m[i][j] = min(c.m[i][j] , m[i][k] + temp.m[k][j]) ;            }        }        return c ;    }};Matrix ans , read ;void power(int mi){    //Matrix ans ; ans.init() ;    //n -- ;    ans = read ;    mi -- ;    while( mi ){        if(mi & 1){            ans = ans * read ;        }        mi >>= 1 ;        //base = base * base ;        read = read * read ;    }    //return ans ;}int main(){    int n , T , st , ed , a , b , w;    //cout << "fsagfads" ;        scanf("%d %d %d %d" , &n , &T , &st , &ed) ;        pq.clear() ;        pq_num = 1 ;        ///read.init() ;        //cout << read.m[0][0] << endl ;        memset(read.m , 0x3f , sizeof(read.m)) ;        //cout << read.m[0][0] << endl ;        //cout << read.m[0][0] << endl ;        for(int i = 0 ; i < T ; i ++ ){            scanf("%d %d %d" , &w , &a , &b) ;            if(!pq[a]) pq[a] = pq_num ++ ;            if(!pq[b]) pq[b] = pq_num ++ ;            a = pq[a] , b = pq[b] ;            read.m[a][b] = read.m[b][a] = w ;        }        st = pq[st] , ed = pq[ed] ;        //power(read , n) ;        power(n) ;        printf("%d\n" , ans.m[st][ed]) ;    //}    return 0 ;}

原创粉丝点击