結果

問題 No.1194 Replace
ユーザー やむなくやむなく
提出日時 2020-08-22 14:26:03
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,411 bytes
コンパイル時間 2,101 ms
コンパイル使用メモリ 185,568 KB
実行使用メモリ 130,240 KB
最終ジャッジ日時 2024-04-23 08:46:53
合計ジャッジ時間 8,974 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

//
// Created by yamunaku on 2020/08/22.
//

#include <bits/stdc++.h>

using namespace std;

#define rep(i, n) for(int i = 0; i < (n); i++)
#define repl(i, l, r) for(int i = (l); i < (r); i++)
#define per(i, n) for(int i = ((n)-1); i >= 0; i--)
#define perl(i, l, r) for(int i = ((r)-1); i >= (l); i--)
#define all(x) (x).begin(),(x).end()
#define MOD9 998244353
#define MOD1 1000000007
#define IINF 1000000000
#define LINF 1000000000000000000
#define SP <<" "<<
#define CYES cout<<"Yes"<<endl
#define CNO cout<<"No"<<endl
#define CFS cin.tie(0);ios::sync_with_stdio(false)
#define CST(x) cout<<fixed<<setprecision(x)

using ll = long long;
using ld = long double;
using vi = vector<int>;
using mti = vector<vector<int>>;
using vl = vector<ll>;
using mtl = vector<vector<ll>>;
using pi = pair<int, int>;
using pl = pair<ll, ll>;
template<typename T>
using heap = priority_queue<T, vector<T>, function<bool(const T, const T)>>;

map<int, vector<int>> e;
map<int, vector<int>> ne;
map<int, int> id;
map<int, int> col;

void dfs(int x, int &num){
    id[x] = -2;
    for(auto &nx : e[x]){
        if(id[nx] == -1){
            dfs(nx, num);
        }
    }
    id[x] = num++;
}

void dfs2(int x, int c){
    col[x] = c;
    for(auto &nx : ne[x]){
        if(col[nx] == -1){
            dfs2(nx, c);
        }
    }
}

int main(){
    CFS;
    ll n;
    int m;
    cin >> n >> m;
    vi b(m), c(m);
    rep(i, m){
        cin >> b[i] >> c[i];
        e[b[i]].push_back(c[i]);
        ne[c[i]].push_back(b[i]);
        id[b[i]] = id[c[i]] = -1;
        col[b[i]] = col[c[i]] = -1;
    }
    int sz = 0;
    for(auto &p:e){
        if(id[p.first] == -1){
            dfs(p.first, sz);
        }
    }
    ll ans = n * (n + 1) / 2;
    vi a(sz);
    map<int, int> ma;
    for(auto &p : id){
        a[p.second] = p.first;
        ans -= p.first;
    }
    per(i, sz){
        if(col[a[i]] == -1){
            dfs2(a[i], a[i]);
        }
    }
    rep(i, sz){
        ma[col[a[i]]] = max(ma[col[a[i]]], a[i]);
    }
    map<int, vector<int>> ee;
    rep(i, m){
        ee[col[b[i]]].push_back(col[c[i]]);
    }
    set<int> f;
    rep(i, sz){
        int cc = col[a[i]];
        if(f.find(cc) != f.end()) continue;
        f.insert(cc);
        for(auto &nx : ee[cc]){
            ma[cc] = max(ma[cc], ma[nx]);
        }
    }
    rep(i, sz){
        ans += ma[col[a[i]]];
    }
    cout << ans << endl;
    return 0;
}
0