oj.uz

Submission #592024

# Submission time Handle Problem Language Result Execution time Memory
592024 2022-07-08T11:49:11 Z andrei_boaca Pipes (CEOI15_pipes) C++14
100 / 100
 2252 ms 12764 KB 
#include <bits/stdc++.h>
#pragma GCC optimize("O3")
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
vector<vector<int>> muchii;
int comp[100005],myroot[100005],dp[17][100005],val[100005],par[100005],niv[100005],lg[100005];
vector<int> used;
int n,m;
int what;
int LCA(int a,int b)
{
    if(niv[a]<niv[b])
        swap(a,b);
    int dif=niv[a]-niv[b];
    int loga=log2(dif);
    for(int i=0;i<=loga;i++)
        if((dif>>i)&1)
            a=dp[i][a];
    if(a==b)
        return a;
    loga=log2(niv[a]);
    for(int i=loga;i>=0;i--)
        if(dp[i][a]!=0&&dp[i][b]!=0)
            if(dp[i][a]!=dp[i][b])
            {
                a=dp[i][a];
                b=dp[i][b];
            }
    return par[a];
}
ll baza=100013;
const ll mod=1e9+7;
int gethash(int a,int b)
{
    ll x=(1LL*baza*a)%mod;
    x+=b;
    if(x>=mod)
        x-=mod;
    return x;
}
void dfscalc(int nod)
{
    for(int q:muchii[nod])
    {
        int i=abs(q);
        if(i!=par[nod])
        {
            dfscalc(i);
            val[nod]+=val[i];
        }
    }
    if(val[nod])
    {
        int a=min(nod,par[nod]);
        int b=max(nod,par[nod]);
        used.push_back(gethash(a,b));
    }
    comp[nod]=what;
}
void dfsbuild(int nod)
{
    val[nod]=0;
    dp[0][nod]=par[nod];
    int loga=log2(niv[nod])+1;
    for(int i=1;i<=loga;i++)
    {
        dp[i][nod]=dp[i-1][dp[i-1][nod]];
        if(dp[i][nod]==0)
            break;
    }
    for(int q:muchii[nod])
    {
        int i=abs(q);
        if(i!=par[nod])
        {
            par[i]=nod;
            niv[i]=niv[nod]+1;
            dfsbuild(i);
        }
    }
}
void dsumerge(int a, int b)
{
    int c1=comp[a],c2=comp[b];
    if(lg[c1]<lg[c2])
    {
        swap(a,b);
        swap(c1,c2);
    }
    what=c1;
    dfscalc(myroot[c2]);
    par[b]=a;
    lg[c1]+=lg[c2];
    lg[c2]=0;
    niv[b]=niv[a]+1;
    muchii[a].push_back(b);
    muchii[b].push_back(a);
    dfsbuild(b);
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin>>n>>m;
    muchii.resize(n+1);
    for(int i=1;i<=n;i++)
    {
        niv[i]=1;
        lg[i]=1;
        comp[i]=i;
        myroot[i]=i;
        par[i]=0;
        val[i]=0;
    }
    int cnt=0;
    for(int z=1;z<=m;z++)
    {
        int a,b;
        cin>>a>>b;
        if(comp[a]!=comp[b])
        {
            cnt++;
            dsumerge(a,b);
        }
        else
        {
            int lca=LCA(a,b);
            val[a]++;
            val[b]++;
            val[lca]-=2;
        }
    }
    for(int i=1;i<=n;i++)
        if(lg[i])
            dfscalc(myroot[i]);
    sort(used.begin(),used.end());
    used.erase(unique(used.begin(),used.end()),used.end());
    int poz=0;
    for(int i=1;i<=n;i++)
    {
        for(int j:muchii[i])
            if(j>i)
            {
                ll h=gethash(i,j);
                int st=0;
                int dr=used.size();
                dr--;
                bool ok=1;
                while(st<=dr)
                {
                    int mij=(st+dr)/2;
                    if(h==used[mij])
                    {
                        ok=0;
                        break;
                    }
                    if(h>used[mij])
                        st=mij+1;
                    else
                        dr=mij-1;
                }
                if(ok)
                    cout<<i<<' '<<j<<'\n';
            }
    }
    return 0;
}

Compilation message

pipes.cpp: In function 'int main()':
pipes.cpp:138:9: warning: unused variable 'poz' [-Wunused-variable]
  138 |     int poz=0;
      |         ^~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 852 KB Output is correct
2 Correct 6 ms 852 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 158 ms 752 KB Output is correct
2 Correct 140 ms 844 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 271 ms 1484 KB Output is correct
2 Correct 337 ms 1700 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 512 ms 3104 KB Output is correct
2 Correct 420 ms 3992 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 770 ms 8964 KB Output is correct
2 Correct 631 ms 9460 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1189 ms 10280 KB Output is correct
2 Correct 1033 ms 10352 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1500 ms 12292 KB Output is correct
2 Correct 1448 ms 12756 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1940 ms 12240 KB Output is correct
2 Correct 1875 ms 12748 KB Output is correct

Subtask #10
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2252 ms 11692 KB Output is correct
2 Correct 2093 ms 12764 KB Output is correct