結果

問題 No.1194 Replace
ユーザー pockyny
提出日時 2020-08-23 01:35:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,067 ms / 2,000 ms
コード長 1,909 bytes
コンパイル時間 1,111 ms
コンパイル使用メモリ 86,024 KB
最終ジャッジ日時 2025-01-13 12:09:10
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <map>
#include <utility>
using namespace std;
vector<int> G[400010],rG[400010],vis;
bool used[400010];
int cmp[400010];
typedef long long ll;
void add_edge(int n){
    for(int i=0;i<n;i++){
        for(int v:G[i]){
            rG[v].push_back(i);
        }
    }
}

void dfs(int s){
    used[s] = true;
    for(int v:G[s]){
        if(!used[v]) dfs(v);
    }
    vis.push_back(s);
}

void rdfs(int s,int k){
    used[s] = true; cmp[s] = k;
    for(int v:rG[s]){
        if(!used[v]) rdfs(v,k);
    }
}

int scc(int n){
    add_edge(n);
    for(int i=0;i<n;i++) used[i] = false;
    vis.clear();
    for(int i=0;i<n;i++){
        if(!used[i]) dfs(i);
    }
    for(int i=0;i<n;i++) used[i] = false;
    int k = 0;
    for(int i=n - 1;i>=0;i--){
        if(!used[vis[i]]){
            rdfs(vis[i],k); k++;
        }
    }
    return k;
}

map<int,int> mp;
pair<int,int> p[200010];
vector<int> z;
vector<int> dag[400010];
int mx[400010] = {},dp[400010];
bool used2[400010];
void solve(int s){
    if(used2[s]) return;
    used2[s] = true;
    for(int v:dag[s]){
        solve(v);
        dp[s] = max(dp[s],dp[v]);
    }
}

int main(){
    ll i,n,m; cin >> n >> m;
    ll ans = n*(n + 1)/2;
    for(i=0;i<m;i++){
        int b,c; cin >> b >> c;
        mp[b] = 1; mp[c] = 1; p[i] = {b,c};
    }
    for(auto x:mp){
        ans -= x.first; z.push_back(x.first);
    }
    for(i=0;i<z.size();i++) mp[z[i]] = i;
    for(i=0;i<m;i++){
        G[mp[p[i].first]].push_back(mp[p[i].second]);
    }
    int sz = scc((int)z.size());
    for(i=0;i<z.size();i++){
        mx[cmp[i]] = max(mx[cmp[i]],z[i]);
    }
    for(i=0;i<z.size();i++){
        used[cmp[i]] = false;
        dp[cmp[i]] = mx[cmp[i]];
        for(int v:G[i]) dag[cmp[i]].push_back(cmp[v]);
    }
    for(i=0;i<z.size();i++) solve(i);
    for(i=0;i<z.size();i++) ans += dp[cmp[i]];
    cout << ans << endl;
}
0