Algoogle

Algorithm for Programming Contest

AOJ 1311 Test Case Tweaking

Category: AOJ Tag: dijkstra

Test Case Tweaking

問題概要


有向グラフの辺の重みを変更してスタートからゴールまでの重みの総和の最小をC以下にしたい.
最小何回の変更で達成できるか.

解法


各ノードに到達するまでに変更した回数とそのときの最小コストをメモしてdijkstraする.
変更はすべて辺の重みを0にすればよい.
ゴールに到達するまでにたどる辺の数は最大でも頂点の数なのでそれより多い回数の変更は許さない.

コード


(1311.cpp) download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#include <bits/stdc++.h>
using namespace std;
#define repi(i,a,b) for(int i = (int)(a); i < (int)(b); i++)
#define rep(i,a) repi(i,0,a)
#define pb push_back
typedef pair<int,int> pii;
struct node{
    int v, d, c;
    bool operator<(const node &n)const{ return d > n.d;}
};
vector<pii> G[128];
int d[128][128];
int N, M, C;

int solve(){
    memset(d, -1, sizeof(d));
    priority_queue<node> q;
    q.push((node){0,0,0});
    while(!q.empty()){
        int v = q.top().v, td = q.top().d, c = q.top().c;
        q.pop();
        if(d[v][c] >= 0) continue;
        d[v][c] = td;
        rep(i,G[v].size()){
            int &nv = G[v][i].first, &nd = G[v][i].second;
            q.push((node){nv, td+nd, c});
            if(c < N) q.push((node){nv, td, c+1});
        }
    }
    rep(i,M) if(0 <= d[N-1][i] and d[N-1][i] <= C) return i;
    return M;
}

bool input(){
    cin >> N >> M >> C;
    rep(i,N) G[i].clear();
    rep(i,M){
        int f, t, c; cin >> f >> t >> c;
        G[f-1].pb(pii(t-1,c));
    }
    return N;
}
signed main(){
    while(input()) cout << solve() << endl;
    return 0;
}

Comments