結果

問題 No.1194 Replace
ユーザー やむなくやむなく
提出日時 2020-08-22 14:22:22
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,397 bytes
コンパイル時間 1,895 ms
コンパイル使用メモリ 184,932 KB
実行使用メモリ 107,304 KB
最終ジャッジ日時 2024-10-15 08:37:44
合計ジャッジ時間 10,623 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 TLE -
testcase_02 WA -
testcase_03 WA -
testcase_04 TLE -
testcase_05 WA -
testcase_06 WA -
testcase_07 TLE -
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] = -1;
    for(auto &nx : e[x]){
        if(id.find(nx) == id.end()){
            dfs(nx, num);
        }
    }
    id[x] = num++;
}

void dfs2(int x, int c){
    col[x] = c;
    for(auto &nx : ne[x]){
        if(col.find(nx) == col.end()){
            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]);
    }
    int sz = 0;
    for(auto &p:e){
        if(id.find(p.first) == id.end()){
            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.find(a[i]) == col.end()){
            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