結果

問題 No.654 Air E869120
コンテスト
ユーザー どらら
提出日時 2018-02-24 00:02:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,321 bytes
コンパイル時間 2,639 ms
コンパイル使用メモリ 209,356 KB
実行使用メモリ 24,064 KB
最終ジャッジ日時 2024-10-10 05:11:37
合計ジャッジ時間 4,111 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 30 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)
#define rep(i,n) REP(i,0,n)
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)
#define ALLOF(c) (c).begin(), (c).end()
typedef long long ll;
typedef unsigned long long ull;

const ll INF = (ll)(1e19);

class Dinic {
  ll MAX_V;
  struct edge{ ll to, cap, rev, icap, flow; };

  vector< vector<edge> > G;
  vector<ll> level; //sからの距離
  vector<ll> iter; //どこまで調べたか

  void max_flow_bfs(ll s){
    fill(level.begin(), level.end(), -1);
    queue<ll> que;
    level[s] = 0;
    que.push(s);
    while(!que.empty()){
      ll v = que.front(); que.pop();
      for(ll i=0; i<G[v].size(); i++){
        edge &e = G[v][i];
        if(e.cap>0 && level[e.to]<0){
          level[e.to] = level[v] + 1;
          que.push(e.to);
        }
      }
    }
  }
  ll max_flow_dfs(ll v, ll t, ll f){
    if(v==t) return f;
    for(ll &i=iter[v]; i<G[v].size(); i++){
      edge &e = G[v][i];
      if(e.cap>0 && level[v]<level[e.to]){
        ll d = max_flow_dfs(e.to, t, min(f, e.cap));
        if(d>0){
          e.cap -= d;
          G[e.to][e.rev].cap += d;
          e.flow += d;
          return d;
        }
      }
    }
    return 0;
  }

public:
  Dinic(ll N):MAX_V(N),G(N),level(N),iter(N){
  }

  void add_edge(ll from, ll to, ll cap){
    G[from].push_back((edge){to, cap, (ll)G[to].size(), cap, 0});
    G[to].push_back((edge){from, 0, (ll)G[from].size()-1, 0, 0});
  }

  ll get_flow(ll from, ll to){ //untried
    rep(i,G[from].size()){
      if(G[from][i].to == to){
        return G[from][i].flow;
      }
    }
    return -1;
  }

  ll max_flow(ll s, ll t){
    ll flow = 0;
    while(true){
      max_flow_bfs(s);
      if(level[t]<0) return flow;
      fill(iter.begin(), iter.end(), 0);
      ll f;
      while((f = max_flow_dfs(s, t, INF))>0){
        flow += f;
      }
    }
  }
};

int N, M;
ll d;
vector<vector<ll>> air;


int main(){
  cin >> N >> M >> d;
  rep(i,M){
    ll u, v, p, q, w;
    cin >> u >> v >> p >> q >> w;
    air.push_back({u,v,p,q,w});
  }

  {
    map<vector<ll>,ll> check;
    rep(i,air.size()){
      vector<ll> tmp = {air[i][0], air[i][1], air[i][2], air[i][3]};
      check[tmp] += air[i][4];
    }
    air.clear();
    FOR(it,check){
      vector<ll> tmp = it->first;
      tmp.push_back(it->second);
      air.push_back(tmp);
    }
  }

  map<pair<int,ll>,int> memo;
  
  vector<set<int>> G(N+1);
  rep(i,air.size()){
    G[air[i][0]].insert(air[i][2]);
    G[air[i][1]].insert(air[i][3]);
  }
  int id = 0;
  rep(i,G.size()){
    FOR(it,G[i]){
      memo[make_pair(i,*it)] = id++;
    }
  }

  Dinic dinic(id+2);
  rep(i,G.size()){
    FOR(it1,G[i]){
      FOR(it2,G[i]){
        int prev = *it1;
        int next = *it2;
        if(next-prev>=d){
          dinic.add_edge(memo[make_pair(i,prev)], memo[make_pair(i,next)], INF);
        }
      }
    }
  }
  
  rep(i,air.size()){
    dinic.add_edge(memo[make_pair(air[i][0],air[i][2])], memo[make_pair(air[i][1],air[i][3])], air[i][4]);
  }

  rep(i,G.size()){
    FOR(it,G[i]){
      if(i==1) dinic.add_edge(id, memo[make_pair(i,*it)], INF);
      if(i==N) dinic.add_edge(memo[make_pair(i,*it)], id+1, INF);
    }
  }
  cout << dinic.max_flow(id, id+1) << endl;

  return 0;
}
0