oj.uz

Submission #1020389

# Submission time Handle Problem Language Result Execution time Memory
1020389 2024-07-12T03:17:56 Z vjudge1 Tree Rotations (POI11_rot) C++17
100 / 100
 197 ms 64712 KB 
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
// #define int long long
// #define double long double
#define pb push_back
 
#define endl '\n'
#define fastIO ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
#define setmin(x,y) x=min((x),(y))
#define setmax(x,y) x=max((x),(y))
#define sqr(x) (x)*(x)
#define fi first
#define se second
 
mt19937 hdp(chrono::high_resolution_clock::now().time_since_epoch().count());
int rand(int l,int r){return l+((hdp()%(r-l+1))+r-l+1)%(r-l+1);}
 
const int N = 4e5 + 5;
const int mod = 998244353;
const int SQ = 450;
const int inf = 1e9;
const int bound = 1e7;
 
vector<int> g[N];
int n,m,c[N],sz[N],par[N];
vector<int> V[N];
stack<int> st;
long long dp[N];
 
int bit[N];
void upd(int i)
{
    for(;i<=n;i+=i&-i) 
    {
        st.push(i);
        bit[i]++;
    }
}
int get(int i)
{
    int res=0;
    for(;i;i-=i&-i) res+=bit[i];
    return res;
}
 
void dfs(int u=1,int p=0,bool keep=0)
{
    int bc=0;
    for(int v:g[u]) if(v!=p&&sz[v]>sz[bc]) bc=v;
    for(int v:g[u]) if(v!=p&&v!=bc) dfs(v,u,0);
    if(bc)
    {
        dfs(bc,u,1);
        swap(V[u],V[bc]);
        dp[u]=dp[bc];
    }
    for(int v:g[u]) if(v!=p&&v!=bc) dp[u]+=dp[v];
    long long sum1=0,sum2=0,xx=V[u].size();
    for(int v:g[u]) if(v!=bc) for(auto t:V[v])
    {
        V[u].pb(t);
        int x=get(t);
        sum1+=x;
        sum2+=xx-x;
    }
    dp[u]+=min(sum1,sum2);
    for(int v:g[u]) if(v!=bc) for(auto t:V[v]) upd(t);
    if(c[u])
    {
        V[u].pb(c[u]);
        upd(c[u]);
    }
    if(!keep) 
    {
        while(!st.empty())
        {
            bit[st.top()]=0;
            st.pop();
        }
    }
}
 
void dfs_sz(int u=1,int p=0)
{
    sz[u]=1;
    for(auto v:g[u]) if(v!=p)
    {
        dfs_sz(v,u);
        sz[u]+=sz[v];
    }
}
 
signed main()
{
    if (fopen("in.txt", "r")) {
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    }
    fastIO
    cin>>n;
    int u=0,v=0;
    for(int m=1;m<2*n;m++)
    {
        cin>>c[m];
        g[u].pb(m);
        par[m]=u;
        if(c[m]) 
        {
            while(g[u].size()==2) u=par[u];
        }
        else u=m;
    }
    dfs_sz();
    dfs();
    cout<<dp[1];
}

Compilation message

rot.cpp: In function 'int main()':
rot.cpp:103:13: warning: unused variable 'v' [-Wunused-variable]
  103 |     int u=0,v=0;
      |             ^
rot.cpp:98:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   98 |         freopen("in.txt","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~
rot.cpp:99:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   99 |         freopen("out.txt","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

Subtask #1
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 19288 KB Output is correct
2 Correct 11 ms 19036 KB Output is correct
3 Correct 11 ms 19216 KB Output is correct
4 Correct 10 ms 19088 KB Output is correct
5 Correct 10 ms 19032 KB Output is correct

Subtask #2
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 19292 KB Output is correct
2 Correct 11 ms 19292 KB Output is correct
3 Correct 11 ms 19292 KB Output is correct
4 Correct 11 ms 19292 KB Output is correct

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 19292 KB Output is correct
2 Correct 12 ms 19192 KB Output is correct
3 Correct 11 ms 19288 KB Output is correct
4 Correct 11 ms 19292 KB Output is correct

Subtask #4
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 20060 KB Output is correct
2 Correct 13 ms 19800 KB Output is correct
3 Correct 12 ms 20316 KB Output is correct
4 Correct 13 ms 20316 KB Output is correct

Subtask #5
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 16 ms 22364 KB Output is correct
2 Correct 19 ms 21084 KB Output is correct
3 Correct 34 ms 24156 KB Output is correct
4 Correct 17 ms 22164 KB Output is correct

Subtask #6
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 46 ms 27596 KB Output is correct
2 Correct 34 ms 29400 KB Output is correct
3 Correct 40 ms 32352 KB Output is correct
4 Correct 41 ms 32452 KB Output is correct

Subtask #7
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 52 ms 42956 KB Output is correct
2 Correct 60 ms 38604 KB Output is correct
3 Correct 60 ms 35532 KB Output is correct
4 Correct 49 ms 33228 KB Output is correct

Subtask #8
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 70 ms 37592 KB Output is correct
2 Correct 67 ms 39880 KB Output is correct
3 Correct 70 ms 46024 KB Output is correct
4 Correct 64 ms 45772 KB Output is correct

Subtask #9
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 155 ms 54580 KB Output is correct
2 Correct 111 ms 50408 KB Output is correct
3 Correct 103 ms 49556 KB Output is correct
4 Correct 109 ms 49092 KB Output is correct
5 Correct 160 ms 49604 KB Output is correct

Subtask #10
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 110 ms 50568 KB Output is correct
2 Correct 113 ms 62400 KB Output is correct
3 Correct 129 ms 58536 KB Output is correct
4 Correct 107 ms 63640 KB Output is correct
5 Correct 190 ms 54140 KB Output is correct

Subtask #11
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 139 ms 51652 KB Output is correct
2 Correct 139 ms 53696 KB Output is correct
3 Correct 160 ms 52540 KB Output is correct
4 Correct 136 ms 52168 KB Output is correct
5 Correct 113 ms 64712 KB Output is correct
6 Correct 197 ms 55368 KB Output is correct