Submission #912757

#	Time	Username	Problem	Language	Result	Execution time	Memory
912757		imarn	Duathlon (APIO18_duathlon)	C++14	10 / 100	84 ms	29184 KiB

count_triplets

#include<bits/stdc++.h>
#define f first
#define s second
#define ll long long
#define pb push_back
#define pii pair<int,int>
#define pll pair<ll,ll>
#define sz(x) (ll)x.size()
#define all(x) x.begin(),x.end()
#define vi vector<int>
#define vvi vector<vi>
using namespace std;
const int N=1e5+5;
vi g[N];stack<int>st;
vvi cmp,bst;
ll d[N]{0},lo[N]{0},t=0,ap[N]{0},id[N]{0},cur=0,isap[N]{0},sz[N],dp[N],vis[N]{0};
void dfs(int u,int p){
    lo[u]=d[u]=++t;st.push(u);
    for(auto v:g[u]){
        if(!d[v]){
            dfs(v,u);
            lo[u]=min(lo[u],lo[v]);
            if(d[u]<=lo[v]){
                ap[u] = (d[u]>1||d[v]>2);cmp.pb({u});
                while(cmp.back().back()!=v){
                    cmp.back().pb(st.top());st.pop();
                }
            }
        }
        else if(v!=p)lo[u]=min(lo[u],d[v]);
    }
}
void build(int n){
    for(int i=1;i<=n;i++)if(ap[i])id[i]=cur++,isap[id[i]]=1,bst.pb({}),sz[id[i]]=1;
    for(auto it : cmp){
        bst.pb({});sz[cur]=it.size();
        for(auto ij : it){
            if(!ap[ij])id[ij]=cur;
            else {
                bst[cur].pb(id[ij]);
                bst[id[ij]].pb(cur);
            }
        }cur++;
    }
}ll ans=0,n,sss=0;
void dd(int u,int p){
    sss+=sz[u];
    sss-=(isap[u]?0:sz(bst[u]));
    for(auto v:bst[u])if(v!=p)dd(v,u);
}
void solve(int u,int p){vis[u]=1;
    dp[u]=sz[u];if(!isap[u])ans+=(sz[u]-sz(bst[u]))*(sz[u]-sz(bst[u])-1)*(sz[u]-sz(bst[u])-2);
    ll tt=0,cc=0;
    for(auto v:bst[u]){
        if(v==p)continue;
        solve(v,u);
        if(isap[u])ans+=2*(dp[v]-1)*(tt)+(sz[v]-1)*(sz[v]-2);
        else ans+=2*(dp[v])*(tt+cc)*(sz[u]-sz(bst[u]));
        tt+=dp[v]-1;cc++;
    }dp[u]+=tt;
    if(!isap[u]&&isap[p])ans+=2*(sss-dp[u]+1)*(sz[u]-sz(bst[u]))*(dp[u]-sz[u]+sz(bst[u])-1)+2*(sz[u]-sz(bst[u]))*(sz[u]-sz(bst[u])-1)*(sss-dp[u]+1)+2*(sz[u]-sz(bst[u]))*(sz[u]-sz(bst[u])-1)*(dp[u]-sz[u]+sz(bst[u])-1);
    else if(!isap[u]&&!isap[p])ans+=+2*(sz[u]-sz(bst[u]))*(sz[u]-sz(bst[u])-1)*(dp[u]-sz[u]+sz(bst[u]));
    else if(isap[u])ans+=2*(sss-dp[u])*(dp[u]-1);
}
int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    int m;cin>>n>>m;
    for(int i=1;i<=m;i++){
        int u,v;cin>>u>>v;
        g[u].pb(v);g[v].pb(u);
    }for(int i=1;i<=n;i++)if(!d[i])dfs(i,i),t=0;
    build(n);for(int i=0;i<cur;i++){
        if(!vis[i]){
            sss=0;dd(i,i);solve(i,i);
        }
    }
    cout<<ans;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 11
• Subtask 3: 0 / 8
• Subtask 4: 10 / 10
• Subtask 5: 0 / 13
• Subtask 6: 0 / 15
• Subtask 7: 0 / 20
• Subtask 8: 0 / 8
• Subtask 9: 0 / 10