图论一顿套模版

图论一顿套模版

https://ac.nowcoder.com/acm/contest/283/H

题目描述

由于临近广西大学建校90周年校庆，西大开始了喜闻乐见的校园修缮工程！

然后问题出现了，西大内部有许许多多的道路，据统计有N栋楼和M条道路（单向），每条路都有“不整洁度”W，现在校方想知道从S楼到T楼的所有路径中，“不整洁度”乘积最小是多少。

由于答案可能很大，所以你需要将最后的答案对10⁹+7取模。

输入描述:

第一行为四个整数N、M、S、T，意义如上。

第2至第M+1行每行表示一条道路，有三个整数，分别表示每条道路的起点u，终点v和“不整洁度”W。

输入保证没有自环，可能有重边。

其中W一定是2的整数次幂。

输出描述:

输出一个整数，表示最小的不整洁度之乘积对10⁹+7取模的结果。

若无解请输出 -1

示例1

输入

4 4 1 3
1 2 8
1 3 65536
2 4 2
4 3 16

输出

256

因为W是2的整数次幂，所以可以把W=log2(W),由乘法变成加法

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<cmath>
#include<vector>
#include<queue>
#define maxn 200005
#define INF 0x3f3f3f3f3f3f3f3f
const long long MOD=1e9+7;
using namespace std;
 
int n,m,k;
struct sair{
    long long len,pos;
    friend bool operator<(sair a,sair b){
        return a.len>b.len;
    }
};
long long dis[maxn];
long long vis[maxn];
 
long long pow_mod(long long sum){
    long long ans=1;
    long long b=2;
    while(sum){
        if(sum&1) ans=(b*ans)%MOD;
        b=(b*b)%MOD;
        sum>>=1;
    }
    return ans;
}
 
vector<pair<long long,long long> >v[maxn];
void Dijstra(int s,int t){
    priority_queue<sair>q;
    sair tmp;
    tmp.len=0,tmp.pos=s;
    dis[s]=0;
    q.push(tmp);
    while(!q.empty()){
        tmp=q.top();
        q.pop();
        long long pos=tmp.pos;
        if(vis[pos]){
            continue;
        }
        vis[pos]=1;
        for(int i=0;i<v[pos].size();i++){
            long long f=v[pos][i].first;
            long long len=v[pos][i].second;
 
            if(dis[f]>dis[pos]+len){
                dis[f]=dis[pos]+len;
                tmp.pos=f;
                tmp.len=dis[f];
                q.push(tmp);
            }
        }
    }
 
    if(dis[t]==INF) cout<<-1<<endl;
    else cout<<pow_mod(dis[t])<<endl;
}
 
 
 
int main(){
 
    std::ios::sync_with_stdio(false);
    int s,t;
    cin>>n>>m>>s>>t;
    for(int i=0;i<=n;i++){
        dis[i]=INF;
        vis[i]=0;
    }
    for(int i=0;i<=n;i++){
        v[i].clear();
    }
    int a,b;
    long long c;
    int j;
    for(int i=1;i<=m;i++){
        cin>>a>>b>>c;
        c=log2(c);
        for(j=0;j<v[a].size();j++){
            if(v[a][j].first==b){
                if(v[a][j].second>c){
                    v[a][j].second=c;
                }
                break;
            }
        }
        if(j==v[a].size()){
            v[a].push_back(make_pair(b,c));
        }
    }
    Dijstra(s,t);
    system("pause");
}