Submission #877324

#	Time	Username	Problem	Language	Result	Execution time	Memory
877324		efedmrlr	Olympic Bus (JOI20_ho_t4)	C++17	5 / 100	1046 ms	7588 KiB

ho_t4

#include <bits/stdc++.h>
using namespace std;

#define n_l '\n'
#define dbg(...) cout << "[" << #__VA_ARGS__ << "]: "; cout << to_string(__VA_ARGS__) << endl
template <typename T, size_t N> int SIZE(const T (&t)[N]){ return N; } template<typename T> int SIZE(const T &t){ return t.size(); } string to_string(const string s, int x1=0, int x2=1e9){ return '"' + ((x1 < s.size()) ? s.substr(x1, x2-x1+1) : "") + '"'; } string to_string(const char* s) { return to_string((string) s); } string to_string(const bool b) { return (b ? "true" : "false"); } string to_string(const char c){ return string({c}); } template<size_t N> string to_string(const bitset<N> &b, int x1=0, int x2=1e9){ string t = ""; for(int __iii__ = min(x1,SIZE(b)),  __jjj__ = min(x2, SIZE(b)-1); __iii__ <= __jjj__; ++__iii__){ t += b[__iii__] + '0'; } return '"' + t + '"'; } template <typename A, typename... C> string to_string(const A (&v), int x1=0, int x2=1e9, C... coords); int l_v_l_v_l = 0, t_a_b_s = 0; template <typename A, typename B> string to_string(const pair<A, B> &p) { l_v_l_v_l++; string res = "(" + to_string(p.first) + ", " + to_string(p.second) + ")"; l_v_l_v_l--; return res; } template <typename A, typename... C> string to_string(const A (&v), int x1, int x2, C... coords) { int rnk = rank<A>::value; string tab(t_a_b_s, ' '); string res = ""; bool first = true; if(l_v_l_v_l == 0) res += n_l; res += tab + "["; x1 = min(x1, SIZE(v)), x2 = min(x2, SIZE(v)); auto l = begin(v); advance(l, x1); auto r = l; advance(r, (x2-x1) + (x2 < SIZE(v))); for (auto e = l; e != r; e = next(e)) { if (!first) { res += ", "; } first = false; l_v_l_v_l++; if(e != l){ if(rnk > 1) { res += n_l; t_a_b_s = l_v_l_v_l; }; } else{ t_a_b_s = 0; } res += to_string(*e, coords...); l_v_l_v_l--; } res += "]"; if(l_v_l_v_l == 0) res += n_l; return res; } void dbgm(){;} template<typename Heads, typename... Tails> void dbgm(Heads H, Tails... T){ cout << to_string(H) << " | "; dbgm(T...); } 
#define dbgm(...) cout << "[" << #__VA_ARGS__ << "]: "; dbgm(__VA_ARGS__); cout << endl


#define int long long int
#define MP make_pair
#define pb push_back
#define REP(i,n) for(int (i) = 0; (i) < (n); (i)++)

void fastio() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
}


const double EPS = 0.00001;
const int INF = 1e16;
const int N = 205;
const int M = 1005;
const int ALPH = 26;
const int LGN = 25;
const int MOD = 1e9+7;
int n,m,q;
vector<array<int,4> > edges;
vector<int> spA, spA_r, spB, spB_r;
vector<int> node_dijA[N], node_dijB[N], node_dijA_r[N], node_dijB_r[N];
int ans = INF;

int mult(int x, int y) {
    return x*y%MOD;
} 
int add(int x, int y) {
    return (x+y)%MOD;
}



void dijkstra(vector<vector<array<int,2> > > &adj, vector<int> &sp, int src, vector<vector<int> > &nds) {
    priority_queue<array<int,3> , vector<array<int,3> >, greater<array<int,3> > > pq;
    nds.assign(n+5, vector<int>());
    sp.assign(n+5, INF);
    pq.push({0, src, 0});
    while(pq.size()) {
        auto cur = pq.top();
        pq.pop();
        if(sp[cur[1]] != INF) {
            if(cur[0] == sp[cur[1]]) {
                nds[cur[1]].pb(cur[2]);
            }   
            continue;
        }
        sp[cur[1]] = cur[0];
        nds[cur[1]].pb(cur[2]);
        for(auto c : adj[cur[1]]) {

            pq.push({cur[0] + c[1], c[0], cur[1]});
        }
    }
}
void dfs(int node, vector<int> &vis, vector<int> &topo, vector<vector<int> > &lst, int fin, vector<vector<int> > &nadj) {
    vis[node] = 1;
    if(node == fin) vis[node] = 2;
    for(auto c : lst[node]) {
        if(vis[c] == 2) {
            vis[node] = 2;
            nadj[node].pb(c);
        } 
        if(vis[c]) continue;
        dfs(c, vis, topo, lst, fin, nadj);
        if(vis[c] == 2) {
            vis[node] = 2;
            nadj[node].pb(c);
        }
    }
    if(vis[node] == 2)
    {
        topo.pb(node);
    }
}
void find_dp(vector<int> &dp, vector<vector<int> > &nds, int src, int fin) {
    vector<int> topo;
    vector<int> vis(n+3, 0);
    vector<vector<int> > nadj(n+3, vector<int>());
    
    dfs(src, vis, topo, nds, fin, nadj);
    
    swap(nds, nadj);
    reverse(topo.begin(), topo.end());
    dp.assign(n+3, 0);
    dp[src] = 1ll;
    for(auto i : topo) {
        for(auto c : nds[i]) {
            dp[c] = add(dp[c], dp[i]);
        }
    }
    
}
void find_dps(vector<int> &dp, vector<int> &dpr, vector<vector<int> > &nds, vector<vector<int> > &nds_r, int src, int fin) {
    find_dp(dp, nds, src, fin);
    find_dp(dpr, nds_r, fin, src);
    
}

bool calcX(array<int,4> &edge, vector<int> &dp, vector<int> &dpr, int src, int fin, vector<vector<int> > &dag) {
    bool is_crit = 0;
    if(mult(dp[edge[0]] , dpr[edge[1]]) == dp[fin] && dag[edge[0]].size() == 1 && dag[edge[0]][0] == edge[1]) {
        is_crit = 1;
    }
    return is_crit;
}
int dijkstra_again(int src, int fin, array<int,4> &eg) {
    vector<vector<array<int,2> > > adj(n+3, vector<array<int,2> >());
    vector<int> sp;
    vector<vector<int> > trash;
    adj[eg[1]].pb({eg[0], eg[2]});
    for(auto &c : edges) {
        if(&c == &eg) continue;
        adj[c[0]].pb({c[1], c[2]});
    }
    
    dijkstra(adj, sp, src, trash);
    return sp[fin];
}
inline void solve() {
    cin>>n>>m;
    edges.resize(m);
    REP(i,m) {
        cin>>edges[i][0]>>edges[i][1]>>edges[i][2]>>edges[i][3];

    }
    
    vector<vector<array<int,2> > > adj(n + 3, vector<array<int,2> >());
    vector<vector<int> > dagA, dagA_r, dagB, dagB_r;
    vector<int> dp, dpr;
    for(auto c : edges) {
        adj[c[0]].pb({c[1], c[2]});
    }
    dijkstra(adj, spA, 1, dagA);
    dijkstra(adj, spB, n, dagB);
    
    adj.assign(n+3, vector<array<int,2> >(0));
    for(auto c : edges) {
        adj[c[1]].pb({c[0], c[2]});
    }
    dijkstra(adj, spA_r, 1, dagA_r);
    dijkstra(adj, spB_r, n, dagB_r);

    ans = min(ans, spA[n] + spB[1]);
    // cout<<"init:\n";
    // dbg(ans);
    
    for(int i = 1; i<=n; i++ ) {
        adj.assign(n+3, vector<array<int,2> >(0));
        for(auto eg : edges) {
            if(eg[0] == i || eg[1] == i) continue;
            adj[eg[0]].pb({eg[1], eg[2]});

        }
        vector<vector<int> > trash;
        dijkstra(adj, node_dijA[i], 1, trash);
        dijkstra(adj, node_dijB[i], n, trash);      
        adj.assign(n+3, vector<array<int,2> >(0));
        
        for(auto eg : edges) {
            if(eg[0] == i || eg[1] == i) continue;
            adj[eg[1]].pb({eg[0], eg[2]});
        }

        dijkstra(adj, node_dijA_r[i], 1, trash);
        dijkstra(adj, node_dijB_r[i], n, trash);

    }

    find_dps(dp, dpr, dagB_r, dagA, 1, n);
    
    for(auto eg : edges) {
        // dbg(node_dijA[eg[0]][eg[1]]); dbg(node_dijB_r[eg[1]][eg[0]]); dbg(node_dijB[eg[0]][eg[1]]); dbg(node_dijA_r[eg[1]][eg[0]]);
        int ret = (node_dijA[eg[0]][eg[1]] + node_dijB_r[eg[1]][eg[0]] + eg[2]) + (node_dijB[eg[0]][eg[1]] + node_dijA_r[eg[1]][eg[0]] + eg[2]);
        // dbg(eg);
        ret += eg[3];
        // dbg(ret);
        ans = min(ans , ret); 
        
    }
    // cout<<"first part:"<<endl;
    for(auto &eg : edges) {
        bool crit = calcX(eg, dp, dpr, 1, n, dagB_r);
        // dbg(eg);
        // cout<<"crit:"<<crit<<endl;
        int costBA = node_dijB[eg[0]][eg[1]] + node_dijA_r[eg[1]][eg[0]] + eg[2] + eg[3];
        // cout<<"costBA:"<<costBA<<endl;
        int ret = costBA;
        if(crit) {
            ret += dijkstra_again(1, n, eg);
            ans = min(ans, ret);
        }  
        else {
            ret += spA[n];
            ans = min(ans , ret);
        }
        // dbg(ret);
    }
    find_dps(dp, dpr, dagA_r, dagB, n, 1);
    for(auto &eg : edges) {
        bool crit = calcX(eg, dp, dpr, n, 1, dagA_r);
        // dbg(eg);
        // cout<<"crit:"<<crit<<endl;
        int costAB = node_dijA[eg[0]][eg[1]] + node_dijB_r[eg[1]][eg[0]] + eg[2] + eg[3];
        // cout<<"costAB:"<<costAB<<endl;
        int ret = costAB;
        if(crit) {
            ret += dijkstra_again(n, 1, eg);
            ans = min(ans ,ret);
        }  
        else {
            ret += spB[1];
            ans = min(ans , ret);
        }
        // dbg(ret);
    }
    if(ans == INF) ans = -1;
    cout<<ans<<"\n";
}
 
signed main() {
    fastio();
    solve();   
}

Compilation message (stderr)

ho_t4.cpp: In function 'std::string to_string(std::string, int, int)':
ho_t4.cpp:6:208: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    6 | template <typename T, size_t N> int SIZE(const T (&t)[N]){ return N; } template<typename T> int SIZE(const T &t){ return t.size(); } string to_string(const string s, int x1=0, int x2=1e9){ return '"' + ((x1 < s.size()) ? s.substr(x1, x2-x1+1) : "") + '"'; } string to_string(const char* s) { return to_string((string) s); } string to_string(const bool b) { return (b ? "true" : "false"); } string to_string(const char c){ return string({c}); } template<size_t N> string to_string(const bitset<N> &b, int x1=0, int x2=1e9){ string t = ""; for(int __iii__ = min(x1,SIZE(b)),  __jjj__ = min(x2, SIZE(b)-1); __iii__ <= __jjj__; ++__iii__){ t += b[__iii__] + '0'; } return '"' + t + '"'; } template <typename A, typename... C> string to_string(const A (&v), int x1=0, int x2=1e9, C... coords); int l_v_l_v_l = 0, t_a_b_s = 0; template <typename A, typename B> string to_string(const pair<A, B> &p) { l_v_l_v_l++; string res = "(" + to_string(p.first) + ", " + to_string(p.second) + ")"; l_v_l_v_l--; return res; } template <typename A, typename... C> string to_string(const A (&v), int x1, int x2, C... coords) { int rnk = rank<A>::value; string tab(t_a_b_s, ' '); string res = ""; bool first = true; if(l_v_l_v_l == 0) res += n_l; res += tab + "["; x1 = min(x1, SIZE(v)), x2 = min(x2, SIZE(v)); auto l = begin(v); advance(l, x1); auto r = l; advance(r, (x2-x1) + (x2 < SIZE(v))); for (auto e = l; e != r; e = next(e)) { if (!first) { res += ", "; } first = false; l_v_l_v_l++; if(e != l){ if(rnk > 1) { res += n_l; t_a_b_s = l_v_l_v_l; }; } else{ t_a_b_s = 0; } res += to_string(*e, coords...); l_v_l_v_l--; } res += "]"; if(l_v_l_v_l == 0) res += n_l; return res; } void dbgm(){;} template<typename Heads, typename... Tails> void dbgm(Heads H, Tails... T){ cout << to_string(H) << " | "; dbgm(T...); }
      |                                                                                                                                                                                                             ~~~^~~~~~~~~~
ho_t4.cpp: In function 'void solve()':
ho_t4.cpp:13:26: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   13 | #define REP(i,n) for(int (i) = 0; (i) < (n); (i)++)
      |                          ^
ho_t4.cpp:132:5: note: in expansion of macro 'REP'
  132 |     REP(i,m) {
      |     ^~~

• Subtask 1: 5 / 5
• Subtask 2: 0 / 11
• Subtask 3: 0 / 21
• Subtask 4: 0 / 63