1. 程式人生 > >河南工業“玲瓏杯” 魔法寶石(spfa或者暴力)

河南工業“玲瓏杯” 魔法寶石(spfa或者暴力)

一道偽裝圖論題,比賽連題都沒看,哎,真是想想就氣。

題目描述

小s想要創造n種魔法寶石。小s可以用ai的魔力值創造一棵第i種魔法寶石,或是使用兩個寶石合成另一種寶石(不消耗魔力值)。請你幫小s算出合成某種寶石的所需的最小花費。

輸入

第一行為資料組數T(1≤T≤3)。
對於每組資料,首先一行為n,m(1≤n,m≤10^5)。分別表示魔法寶石種類數和合成魔法的數量。
之後一行n個數表示a1到an。(1≤ai≤10^9)。a_i表示合成第i種寶石所需的魔力值。
之後n行,每行三個數a,b,c(1≤a,b,c≤n),表示一個第a種寶石和第b種寶石,可以合成一個第c種寶石。

輸出

每組資料輸出一行n個數,其中第i個數表示合成第i種寶石的魔力值最小花費。

樣例輸入

1
3 1
1 1 10
1 2 3
樣例輸出

1 1 2

可以看出是求每個合成寶石的最小魔力值,不過這裡是有些兩個寶石可以另一個寶石,故我們只要對這些求一下是直接製造這個寶石魔力值小,還是用兩個寶石合成魔力值小。所以,TM可以暴力

第一種暴力寫法:

#include<iostream>
#include<algorithm>
using namespace std;
const int MAX = 1e5 + 10;
typedef long long ll;
ll dis[MAX];
ll a[MAX],b[MAX],c[MAX];
int
main(void) { int t; cin >> t; while(t--){ int n,m; cin >> n >> m; for(int i=1;i<=n;i++) dis[i] = 0; for(int i=1;i<=n;i++) cin >> dis[i]; for(int i=1;i<=m;i++){ cin >> a[i] >> b[i] >> c[i]; } for
(int i=1;i<=n;i++){ bool checked = false;//判斷每次是否被更新 for(int j=1;j<=m;j++){ if(dis[c[j]] > dis[a[j]] + dis[b[j]]){ checked = true; dis[c[j]] = dis[a[j]] + dis[b[j]]; } } if(!checked) break;//如果這一圈沒有更新,就直接跳出 } for(int i=1;i<=n-1;++i) cout << dis[i] << " ";//這裡要注意輸出格式,最後一位數後面不能有空格 cout << dis[n] << endl; } return 0; }

第二種SPFA演算法 :

#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
using namespace std;
const int MAX = 1e5 + 10;
typedef long long ll;
bool used[MAX];//用來標記這個點是否在佇列中
struct Edge{
    int b;
    int to;
};
ll dis[MAX];
vector<Edge> G[MAX];
int main(void)
{
    int t,n,m;
    cin >> t;
    while(t--){
        queue<ll> q;
        cin >> n >> m;
        for(int i=1;i<=n;i++){
            G[i].clear();
            dis[i] = 0;
            used[i] = false;
        }
        for(int i=1;i<=n;i++)
            cin >> dis[i];
        ll a,b,c;
        Edge e;
        for(int i=1;i<=m;i++){
            cin >> a >> b >> c;
            if(dis[c] > dis[a] + dis[b]){
                dis[c] = dis[a] + dis[b];
                q.push(c);
                used[c] = true;//如果這個點發生更新,就把這個點放入佇列中。
            }
            e.b = b;e.to = c;//注意這裡要用鄰接表存兩個邊
            G[a].push_back(e);
            e.b = a;e.to = c;
            G[b].push_back(e);
        }
        while(!q.empty()){
            ll k = q.front();
            for(int i=0;i<G[k].size();i++){
                e = G[k][i];
                if(dis[k] + dis[e.b] < dis[e.to]){
                    dis[e.to] = dis[k] + dis[e.b];
                    if(used[e.to] == false){
                        q.push(e.to);
                        used[e.to] = true;
                    }
                }
            }
            q.pop();
            used[k] = false;
        }
        for(int i=1;i<=n-1;i++)
            cout << dis[i] << " ";//輸出格式
        cout << dis[n] << endl;
    }
    return 0;
}