oj.uz

Submission #520735

# Submission time Handle Problem Language Result Execution time Memory
520735 2022-01-31T00:51:47 Z christinelynn Swap (BOI16_swap) C++17
100 / 100
 965 ms 233796 KB 
#include<bits/stdc++.h>
#define ll int
#define ld long double
#define fs first
#define sc second
#define debug(y) cout<<y,exit(0)
using namespace std;
typedef pair<ll,ll> LL;
const ll N = 2e5 + 9;
const ll inf = 1e9 + 7;
 
vector<ll> dp[N][18][2];
ll n,a[N];
bool need[N][18][2];
 
void Merge(vector<ll> &res,vector<ll> &a,vector<ll> &b){
    for (ll i = 0,j = 1;i < a.size();i += j,j *= 2){
        for (ll k = i;k < i + j && k < a.size();k++) res.push_back(a[k]);
        for (ll k = i;k < i + j && k < b.size();k++) res.push_back(b[k]);
    }
}
 
int main(){
    ios_base::sync_with_stdio(NULL); cin.tie(0); cout.tie(0);
    #define task "test"
    if (fopen(task".inp","r")){
        freopen(task".inp","r",stdin);
        //freopen(task".out","w",stdout);
    }
    cin>>n;
    for (ll i = 1;i <= n;i++) cin>>a[i];
    need[1][0][1] = 1;
 
    for (ll node = 1;node <= n/2;node++){
        for (ll i = 0;node >> i;i++){
            for (auto j : {0,1}){
                if (!need[node][i][j]) continue;
                ll p = ((node >> i + 1) << 1) + j;
                ll mn = min({a[p],a[node*2],a[node*2 + 1]});
                if (mn == a[node*2])
                    need[2*node][i + 1][j] = need[2*node + 1][0][1] = 1;
                else if (mn == a[p])
                    need[2*node][0][0] = need[2*node + 1][0][1] = 1;
                else
                    need[2*node][0][0] = need[2*node + 1][0][0] = 1,
                    need[2*node][i + 1][j] = need[2*node + 1][i + 1][j] = 1;
            }
        }
    }
 
    for (ll node = n;node;node--){
        for (ll i = 0;node >> i;i++){
            for (auto j : {0,1}){
                if (!need[node][i][j]) continue;
 
                ll p = ((node >> i + 1) << 1) + j;
                if (2*node > n) dp[node][i][j] = {a[p]};
                else if (2*node == n) dp[node][i][j] = {min(a[p],a[node*2]),max(a[p],a[node*2])};
                else{
                    ll mn = min({a[p],a[2*node],a[2*node + 1]});
                    if (mn == a[2*node + 1]){
                        vector<ll> tmp1,tmp2; tmp1 = tmp2 = {a[2*node + 1]};
                        Merge(tmp1,dp[2*node][0][0],dp[2*node + 1][i + 1][j]);
                        Merge(tmp2,dp[2*node][i + 1][j],dp[2*node + 1][0][0]);
                        dp[node][i][j] = min(tmp1,tmp2);
                    }
                    else{
                        dp[node][i][j] = {min(a[p],a[2*node])};
                        if (mn == a[p])
                            Merge(dp[node][i][j],dp[2*node][0][0],dp[2*node + 1][0][1]);
                        else
                            Merge(dp[node][i][j],dp[2*node][i + 1][j],dp[2*node + 1][0][1]);
                    }
                }
            }
        }
        if (2*node < n){
            for (ll i = 0;2*node >> i;i++){
                for (auto j : {0,1})
                    vector<ll>().swap(dp[2*node][i][j]),
                    vector<ll>().swap(dp[2*node + 1][i][j]);
            }
        }
    }
    for (auto i : dp[1][0][1]) cout<<i<<" ";
}
 
/* stuff you should look for
  * int overflow, array bounds
  * special cases (n=1?)
  * do smth instead of nothing and stay organized
  * WRITE STUFF DOWN
  * DON'T GET STUCK ON ONE APPROACH
*/

Compilation message

swap.cpp: In function 'void Merge(std::vector<int>&, std::vector<int>&, std::vector<int>&)':
swap.cpp:17:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   17 |     for (ll i = 0,j = 1;i < a.size();i += j,j *= 2){
      |                         ~~^~~~~~~~~~
swap.cpp:18:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |         for (ll k = i;k < i + j && k < a.size();k++) res.push_back(a[k]);
      |                                    ~~^~~~~~~~~~
swap.cpp:19:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |         for (ll k = i;k < i + j && k < b.size();k++) res.push_back(b[k]);
      |                                    ~~^~~~~~~~~~
swap.cpp: In function 'int main()':
swap.cpp:38:36: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   38 |                 ll p = ((node >> i + 1) << 1) + j;
      |                                  ~~^~~
swap.cpp:56:36: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   56 |                 ll p = ((node >> i + 1) << 1) + j;
      |                                  ~~^~~
swap.cpp:27:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |         freopen(task".inp","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 169424 KB Output is correct
2 Correct 76 ms 169384 KB Output is correct
3 Correct 94 ms 169312 KB Output is correct
4 Correct 75 ms 169284 KB Output is correct
5 Correct 75 ms 169476 KB Output is correct

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 169424 KB Output is correct
2 Correct 76 ms 169384 KB Output is correct
3 Correct 94 ms 169312 KB Output is correct
4 Correct 75 ms 169284 KB Output is correct
5 Correct 75 ms 169476 KB Output is correct
6 Correct 78 ms 169436 KB Output is correct
7 Correct 84 ms 169284 KB Output is correct
8 Correct 88 ms 169320 KB Output is correct
9 Correct 75 ms 169404 KB Output is correct
10 Correct 74 ms 169400 KB Output is correct

Subtask #3
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 169424 KB Output is correct
2 Correct 76 ms 169384 KB Output is correct
3 Correct 94 ms 169312 KB Output is correct
4 Correct 75 ms 169284 KB Output is correct
5 Correct 75 ms 169476 KB Output is correct
6 Correct 78 ms 169436 KB Output is correct
7 Correct 84 ms 169284 KB Output is correct
8 Correct 88 ms 169320 KB Output is correct
9 Correct 75 ms 169404 KB Output is correct
10 Correct 74 ms 169400 KB Output is correct
11 Correct 77 ms 169404 KB Output is correct
12 Correct 77 ms 169484 KB Output is correct
13 Correct 78 ms 169484 KB Output is correct
14 Correct 87 ms 169668 KB Output is correct
15 Correct 84 ms 169612 KB Output is correct

Subtask #4
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 169424 KB Output is correct
2 Correct 76 ms 169384 KB Output is correct
3 Correct 94 ms 169312 KB Output is correct
4 Correct 75 ms 169284 KB Output is correct
5 Correct 75 ms 169476 KB Output is correct
6 Correct 78 ms 169436 KB Output is correct
7 Correct 84 ms 169284 KB Output is correct
8 Correct 88 ms 169320 KB Output is correct
9 Correct 75 ms 169404 KB Output is correct
10 Correct 74 ms 169400 KB Output is correct
11 Correct 77 ms 169404 KB Output is correct
12 Correct 77 ms 169484 KB Output is correct
13 Correct 78 ms 169484 KB Output is correct
14 Correct 87 ms 169668 KB Output is correct
15 Correct 84 ms 169612 KB Output is correct
16 Correct 149 ms 174052 KB Output is correct
17 Correct 135 ms 174104 KB Output is correct
18 Correct 128 ms 174008 KB Output is correct
19 Correct 261 ms 183892 KB Output is correct
20 Correct 264 ms 183924 KB Output is correct

Subtask #5
32.0 / 32.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 169424 KB Output is correct
2 Correct 76 ms 169384 KB Output is correct
3 Correct 94 ms 169312 KB Output is correct
4 Correct 75 ms 169284 KB Output is correct
5 Correct 75 ms 169476 KB Output is correct
6 Correct 78 ms 169436 KB Output is correct
7 Correct 84 ms 169284 KB Output is correct
8 Correct 88 ms 169320 KB Output is correct
9 Correct 75 ms 169404 KB Output is correct
10 Correct 74 ms 169400 KB Output is correct
11 Correct 77 ms 169404 KB Output is correct
12 Correct 77 ms 169484 KB Output is correct
13 Correct 78 ms 169484 KB Output is correct
14 Correct 87 ms 169668 KB Output is correct
15 Correct 84 ms 169612 KB Output is correct
16 Correct 149 ms 174052 KB Output is correct
17 Correct 135 ms 174104 KB Output is correct
18 Correct 128 ms 174008 KB Output is correct
19 Correct 261 ms 183892 KB Output is correct
20 Correct 264 ms 183924 KB Output is correct
21 Correct 329 ms 188152 KB Output is correct
22 Correct 316 ms 188224 KB Output is correct
23 Correct 307 ms 188200 KB Output is correct
24 Correct 965 ms 233636 KB Output is correct
25 Correct 929 ms 233796 KB Output is correct