[HDU 6331]Walking Plan

HDU 6331

题解

#include<bits/stdc++.h>
#define LL long long
#define ull unsigned long long
#define ULL ull
#define mp make_pair
#define pii pair<int,int>
#define piii pair<int, pii >
#define pll pair <ll,ll>
#define pb push_back
#define big 20160116
#define INF 2147483647
#define pq priority_queue
using namespace std;
inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
namespace Mymath{
    LL qp(LL x,LL p,LL mod){
        LL ans=1;
        while (p){
            if (p&1) ans=ans*x%mod;
            x=x*x%mod;
            p>>=1;
        }
        return ans;
    }
    LL inv(LL x,LL mod){
        return qp(x,mod-2,mod);
    }
    LL C(LL N,LL K,LL fact[],LL mod){
        return fact[N]*inv(fact[K],mod)%mod*inv(fact[N-K],mod)%mod;
    }
    template <typename Tp> Tp gcd(Tp A,Tp B){
        if (B==0) return A;
        return gcd(B,A%B);
    }
    template <typename Tp> Tp lcm(Tp A,Tp B){
        return A*B/gcd(A,B);
    }
};
namespace fwt{
    using namespace Mymath;
    void FWT(int a[],int n,LL mod)
    {
        for(int d=1;d<n;d<<=1)
            for(int m=d<<1,i=0;i<n;i+=m)
                for(int j=0;j<d;j++)
                {
                    int x=a[i+j],y=a[i+j+d];
                    a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod;
                    //xor:a[i+j]=x+y,a[i+j+d]=x-y;
                    //and:a[i+j]=x+y;
                    //or:a[i+j+d]=x+y;
                }
    }

    void UFWT(int a[],int n,LL mod)
    {
        LL rev=inv(2,mod);
        for(int d=1;d<n;d<<=1)
            for(int m=d<<1,i=0;i<n;i+=m)
                for(int j=0;j<d;j++)
                {
                    int x=a[i+j],y=a[i+j+d];
                    a[i+j]=1LL*(x+y)*rev%mod,a[i+j+d]=(1LL*(x-y)*rev%mod+mod)%mod;
                    //xor:a[i+j]=(x+y)/2,a[i+j+d]=(x-y)/2;
                    //and:a[i+j]=x-y;
                    //or:a[i+j+d]=y-x;
                }
    }
    void solve(int a[],int b[],int n,LL mod)
    {
        FWT(a,n,mod);
        FWT(b,n,mod);
        for(int i=0;i<n;i++) a[i]=1LL*a[i]*b[i]%mod;
        UFWT(a,n,mod);
    }
};
const int Maxn=55;
int n,m;
int dist[Maxn][Maxn];
int dp[105][Maxn][Maxn];
int dp2[105][Maxn][Maxn];
void mian(){
    n=read();m=read();
    for (int i=1;i<=n;i++){
        for (int j=1;j<=n;j++){
            dist[i][j]=-1;
        }
    }
    for (int i=0;i<m;i++){
        int u=read(),v=read(),w=read();
        if (dist[u][v]==-1) dist[u][v]=w;
        else dist[u][v]=min(dist[u][v],w); 
    }
    for (int i=0;i<105;i++){
        for (int j=0;j<Maxn;j++){
            for (int k=0;k<Maxn;k++) dp[i][j][k]=1e9;
        }
    }
    for (int i=1;i<=n;i++){
        dp[0][i][i]=0;
    }
    for (int i=1;i<=100;i++){
        for (int j=1;j<=n;j++){
            for (int k=1;k<=n;k++){
                for (int l=1;l<=n;l++){
                    if (dist[l][k]!=-1){
                        dp[i][j][k]=min(dp[i][j][k],dp[i-1][j][l]+dist[l][k]);
                    }
                }
            }
        }
    }
    for (int i=0;i<105;i++){
        for (int j=0;j<Maxn;j++){
            for (int k=0;k<Maxn;k++) dp2[i][j][k]=1e9;
        }
    }
    for (int i=1;i<=n;i++){
        dp2[0][i][i]=0;
    }
    for (int i=1;i<=101;i++){
        for (int j=1;j<=n;j++){
            for (int k=1;k<=n;k++){
                for (int l=1;l<=n;l++){
                    if (dp[100][l][k]!=-1){
                        dp2[i][j][k]=min(dp2[i][j][k],dp2[i-1][j][l]+dp[100][l][k]);
                    }
                }
            }
        }
    }
    for (int i=99;i>=0;i--){
        for (int j=1;j<=n;j++){
            for (int k=1;k<=n;k++){
                dp[i][j][k]=min(dp[i][j][k],dp[i+1][j][k]);
            }
        }
    }
    int q;
    scanf("%d",&q);
    while (q--){
        int s,t,k;
        s=read();t=read();k=read();
        int v=k/100;
        int u=k%100;
        int ans=1e9;
        for (int ww=1;ww<=n;ww++){
            ans=min(ans,dp2[v][s][ww]+dp[u][ww][t]);
        }
        for (int ww=1;ww<=n;ww++){
            ans=min(ans,dp2[v+1][s][ww]+dp[0][ww][t]);
        }
        if (ans==1e9){
            printf("-1\n");
        }
        else{
            printf("%d\n",ans);
        }
    }
}
int main(){
    int tc;
    tc=read();
    while (tc--){
        mian();
    }
}