oj.uz

Submission #1046798

# Submission time Handle Problem Language Result Execution time Memory
1046798 2024-08-07T00:17:21 Z nightfal Comparing Plants (IOI20_plants) C++17
100 / 100
 1206 ms 271272 KB 
// #include "plants.h"
#include <iostream>
#include <cstdio>
#include <cassert>
#include <vector>
#include <utility>
#include <algorithm>


// using namespace std;
#define maxVal 1'000'000'001
static std:: vector<int> rank,lz,lzr;
static std:: vector<std::pair<int,int>> t,tr;
static std:: vector<std::vector<int>> g;
static std:: vector<std::vector<std::vector<int>>> power;

static int l,m;
void print(std::pair<int,int> p) {std::cout << "(" << p.first << "," << p.second << ") "; };
void print(std::vector<int> &v) {for(int elem: v) std::cout << elem << " "; std::cout << std::endl;}
void print(std::vector<std::vector<int>> &v) {for (auto elem: v) print(elem); std::cout << std::endl;}
void print(std::vector<std::pair<int,int>> &v) {for(auto [a,b]: v) std::cout << "(" << a << "," << b << ") ";std::cout << std::endl;}
void print(std::vector<std::vector<std::vector<int>>> &v) {for (auto elem: v) print(elem); std::cout << std::endl;}

std::pair<int,int> update(std::vector<std::pair<int,int>> &t, std::vector<int> &lz, int node,int start,int end,int left,int right,int diff) {
    if (left <= start and end <= right) lz[node] += diff;
    if (start!=end) { lz[2*node] += lz[node]; lz[2*node+1] += lz[node];}
    t[node].first += lz[node]; lz[node]=0;
    if (right < start or end < left) return {maxVal,m+1};
    if (left <= start and end <= right) return t[node];

    int mid=(start+end)/2;
    auto ans1  = update(t,lz,2*node,start,mid,left,right,diff);
    auto ans2  = update(t,lz,2*node+1,mid+1,end,left,right,diff);
    auto ans = std::min(ans1,ans2);
    t[node] = std::min(t[2*node],t[2*node+1]);
    return ans;    
}
std::pair<int,int> findMin(std::vector<std::pair<int,int>> &t, std::vector<int> &lz, int node, int start, int end, int left, int right) {
    if (start!=end) {lz[2*node] += lz[node]; lz[2*node+1] += lz[node];}
    t[node].first += lz[node]; lz[node]=0;
    if (right < start or end < left) return {maxVal,m+1};
    if (left <= start and end <= right) return t[node];
    
    int mid=(start+end)/2;
    auto min1 = findMin(t,lz,2*node,start,mid,left,right);
    auto min2 = findMin(t,lz,2*node+1,mid+1,end,left,right);
    auto ans = std::min(min1,min2);
    t[node] = std::min(t[2*node],t[2*node+1]);    
    return ans;
}
void segInit(std::vector<int> &r, std::vector<std::pair<int,int>> &t, int node, int start, int end) {
    if (start==end) {t[node]={r[start],start}; return;}
    int mid=(start+end)/2;
    segInit(r,t, 2*node,start,mid);
    segInit(r,t, 2*node+1,mid+1,end);
    t[node] = std::min(t[2*node],t[2*node+1]);
    return;
}
int findPreMinI(std::vector<std::pair<int,int>> &t, std::vector<int> &lz, int k,int minI1) {
    int ans = minI1;
    if (minI1>0) {
        auto [min2,minI2] = findMin(t,lz,1,0,m-1,std::max(0,minI1-k+1),minI1-1);
        if (min2==0) ans = minI2;
    }
    if (minI1-k+1<0) {
        auto [min2,minI2] = findMin(t,lz,1,0,m-1,minI1-k+1+m,m-1);
        if (min2==0) ans = minI2;
    }
    return ans;
}
int computeRank(std::vector<std::pair<int,int>> &t, std::vector<int> &lz, int minI1, int k, int ranking) {

    if (rank[minI1]!=m) return ranking;
    int minI2 = findPreMinI(t,lz,k,minI1);

    while (minI2 != minI1) {
        ranking = computeRank(t,lz,minI2,k,ranking); 
        minI2 = findPreMinI(t,lz,k,minI1);
    }
    rank[minI1] = ranking--;
    update(t,lz,1,0,m-1,minI1,minI1,m-1);
    if(minI1>0) update(t,lz,1,0,m-1,std::max(0,minI1-k+1),minI1-1,-1);
    if(minI1-k+1<0) update(t,lz,1,0,m-1,minI1-k+1+m,m-1,-1);

    return ranking;
}

bool inRangeLeft(int z, int y, int left) {return y < z && (z < left || left <= y) || z < y && (z < left && left <= y);}
bool inRangeRight(int z, int y, int right) {return z < y && (right < z || y <= right) || y < z && (y <= right && right < z);}

int compare(int x, int y) {
    int left, i=0, z=x;
    int found = -1;
    while(true) {
        left = power[i][z][0];
        if(left==-1) break;
        if(left==z || inRangeLeft(z,y,left)) {found = left; break;}
        i++;
    }
    for(int j=i-1; j>=0; j--) {
        left = power[j][z][0];
        if(left==-1) continue;
        if(left==z || inRangeLeft(z,y,left)) {found = left; continue;}
        else z = left; 
    }
    left = power[0][z][0];
    // assert(found==left);
    if (left!=-1 && inRangeLeft(z,y,left) && -rank[z]>-rank[y]) return 1;

    int right; 
    i=0; z= x; found = -1;
    while(true) {
        right = power[i][z][1];
        if(right==-1) break;
        if(right==z || inRangeRight(z,y,right)) {found = right; break;}
        i++;
    }
    for(int j=i-1; j>=0; j--) {
        right = power[j][z][1];
        if(right==-1) continue;
        if(right==z || inRangeRight(z,y,right)) {found = right; continue;}
        else z = right;
    }
    right = power[0][z][1];
    // assert(found==right);
    if(right!=-1 && inRangeRight(z,y,right) && -rank[z]>-rank[y]) return 1;

    return 0;
}
void calPower() {
    int x = m-1, cnt=1;
    while (x) {x>>=1; cnt++;}
    cnt++;
    power.resize(cnt,std::vector<std::vector<int>>(m,std::vector<int>(2,-1)));
    for(int j=0; j<m; j++)
        for(int x=0; x<2; x++)
            power[0][j][x] = g[j][x];
    
    for (int i=1; i<cnt; i++)
        for(int j=0; j<m; j++)
            for(int x=0; x<2; x++) {
                int  prev = power[i-1][j][x]; 
                int  &next = power[i][j][x];
                if (prev == -1 || prev == j) {next = prev; continue;}
                next = power[i-1][prev][x];
                if (next == -1 or next == j) continue;
                // if(x==0 && (prev < next && next < j || j < prev && (next < j || prev < next)) || 
                // (x==1 && (j < next && next < prev || prev < j && (j < next || next < prev))))
                if(x==0 && inRangeLeft(j,next,prev) || x==1 && inRangeRight(j,next,prev))
                next = j;
            }
    // std::cout << "rank: "<<std::endl; print(rank);
    // std::cout << "power: "<<std::endl; print(power);
    // std::cout << "rangeLR: "<<std::endl; print(rangeLR);
}
void calReachability(int k) {
    tr.resize(4*m); lzr.resize(4*m);
    std::vector<std::pair<int,int>> rankInv;
    for(int i=0; i<m; i++) {rank[i] = -rank[i]; rankInv.push_back({rank[i],i}); }
    sort(rankInv.begin(),rankInv.end());
    segInit(rank,tr,1,0,m-1);

    g.resize(m,std::vector<int>(2,-1));
    
    for(auto [val,i]: rankInv) {
        std::pair<int,int> temp1 ={maxVal,m+1}, temp2={maxVal,m+1};
        if (i>0) temp1 = findMin(tr,lzr,1,0,m-1,std::max(0,i-k+1),i-1);
        if (i-k+1<0) temp2 = findMin(tr,lzr,1,0,m-1,i-k+1+m,m-1);
        temp1 = std::min(temp1,temp2); 
        if (temp1.first <=0) g[i][0] = temp1.second;

        temp1 ={maxVal,m+1}, temp2={maxVal,m+1};
        if (i+1<m) temp1 = findMin(tr,lzr,1,0,m-1,i+1,std::min(m-1,i+k-1));
        if (i+k-1>m-1) temp2 = findMin(tr,lzr,1,0,m-1,0,i+k-1-m);
        temp1 = std::min(temp1,temp2);
        if (temp1.first <= 0) g[i][1] = temp1.second;
        update(tr,lzr,1,0,m-1,i,i,m);
    }
    calPower();
    return;
}
void init(int k, std::vector<int> r) {
    int n = r.size(); 
    m=n; l=k;
    t.resize(4*n); lz.resize(4*n); 
    segInit(r,t,1,0,n-1);
    rank.resize(n,n);
    int ranking = n-1;
    while (ranking >=0) {
        auto [min1,minI1] = findMin(t,lz,1,0,n-1,0,n-1);
        ranking = computeRank(t,lz,minI1,k,ranking);
    }
    calReachability(k);
	return;
}
int compare_plants(int x, int y) {
    int ans = compare(x,y);
    if (ans) return ans;
    else return -compare(y,x);
}

Compilation message

plants.cpp: In function 'bool inRangeLeft(int, int, int)':
plants.cpp:88:56: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   88 | bool inRangeLeft(int z, int y, int left) {return y < z && (z < left || left <= y) || z < y && (z < left && left <= y);}
      |                                                  ~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~
plants.cpp: In function 'bool inRangeRight(int, int, int)':
plants.cpp:89:58: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   89 | bool inRangeRight(int z, int y, int right) {return z < y && (right < z || y <= right) || y < z && (y <= right && right < z);}
      |                                                    ~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
plants.cpp: In function 'int compare(int, int)':
plants.cpp:93:9: warning: variable 'found' set but not used [-Wunused-but-set-variable]
   93 |     int found = -1;
      |         ^~~~~
plants.cpp: In function 'void calPower()':
plants.cpp:149:25: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
  149 |                 if(x==0 && inRangeLeft(j,next,prev) || x==1 && inRangeRight(j,next,prev))
      |                    ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 30 ms 4240 KB Output is correct
7 Correct 128 ms 28112 KB Output is correct
8 Correct 483 ms 269504 KB Output is correct
9 Correct 549 ms 269456 KB Output is correct
10 Correct 656 ms 269508 KB Output is correct
11 Correct 757 ms 269504 KB Output is correct
12 Correct 794 ms 269512 KB Output is correct
13 Correct 881 ms 269404 KB Output is correct
14 Correct 991 ms 269404 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 600 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 3 ms 1372 KB Output is correct
7 Correct 50 ms 10244 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 3 ms 1368 KB Output is correct
10 Correct 52 ms 10328 KB Output is correct
11 Correct 74 ms 10176 KB Output is correct
12 Correct 115 ms 10420 KB Output is correct
13 Correct 49 ms 10324 KB Output is correct

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 600 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 3 ms 1372 KB Output is correct
7 Correct 50 ms 10244 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 3 ms 1368 KB Output is correct
10 Correct 52 ms 10328 KB Output is correct
11 Correct 74 ms 10176 KB Output is correct
12 Correct 115 ms 10420 KB Output is correct
13 Correct 49 ms 10324 KB Output is correct
14 Correct 109 ms 28100 KB Output is correct
15 Correct 801 ms 270260 KB Output is correct
16 Correct 105 ms 28108 KB Output is correct
17 Correct 799 ms 270272 KB Output is correct
18 Correct 890 ms 269760 KB Output is correct
19 Correct 1151 ms 270272 KB Output is correct
20 Correct 759 ms 270276 KB Output is correct

Subtask #4
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 78 ms 4940 KB Output is correct
4 Correct 1090 ms 271048 KB Output is correct
5 Correct 1030 ms 269760 KB Output is correct
6 Correct 948 ms 269892 KB Output is correct
7 Correct 887 ms 270020 KB Output is correct
8 Correct 828 ms 270216 KB Output is correct
9 Correct 904 ms 269504 KB Output is correct
10 Correct 970 ms 269504 KB Output is correct
11 Correct 868 ms 269508 KB Output is correct
12 Correct 1069 ms 269504 KB Output is correct
13 Correct 968 ms 269508 KB Output is correct

Subtask #5
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 436 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 1 ms 604 KB Output is correct
7 Correct 11 ms 1628 KB Output is correct
8 Correct 9 ms 1624 KB Output is correct
9 Correct 12 ms 1544 KB Output is correct
10 Correct 10 ms 1624 KB Output is correct
11 Correct 10 ms 1540 KB Output is correct
12 Correct 10 ms 1592 KB Output is correct
13 Correct 8 ms 1628 KB Output is correct

Subtask #6
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 3 ms 1116 KB Output is correct
6 Correct 692 ms 268760 KB Output is correct
7 Correct 804 ms 268940 KB Output is correct
8 Correct 754 ms 268996 KB Output is correct
9 Correct 814 ms 269348 KB Output is correct
10 Correct 649 ms 268724 KB Output is correct
11 Correct 698 ms 269248 KB Output is correct
12 Correct 736 ms 270520 KB Output is correct
13 Correct 817 ms 268996 KB Output is correct
14 Correct 754 ms 268956 KB Output is correct
15 Correct 835 ms 269116 KB Output is correct
16 Correct 652 ms 268632 KB Output is correct
17 Correct 723 ms 268996 KB Output is correct

Subtask #7
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 30 ms 4240 KB Output is correct
7 Correct 128 ms 28112 KB Output is correct
8 Correct 483 ms 269504 KB Output is correct
9 Correct 549 ms 269456 KB Output is correct
10 Correct 656 ms 269508 KB Output is correct
11 Correct 757 ms 269504 KB Output is correct
12 Correct 794 ms 269512 KB Output is correct
13 Correct 881 ms 269404 KB Output is correct
14 Correct 991 ms 269404 KB Output is correct
15 Correct 0 ms 600 KB Output is correct
16 Correct 0 ms 348 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 1 ms 348 KB Output is correct
20 Correct 3 ms 1372 KB Output is correct
21 Correct 50 ms 10244 KB Output is correct
22 Correct 1 ms 348 KB Output is correct
23 Correct 3 ms 1368 KB Output is correct
24 Correct 52 ms 10328 KB Output is correct
25 Correct 74 ms 10176 KB Output is correct
26 Correct 115 ms 10420 KB Output is correct
27 Correct 49 ms 10324 KB Output is correct
28 Correct 109 ms 28100 KB Output is correct
29 Correct 801 ms 270260 KB Output is correct
30 Correct 105 ms 28108 KB Output is correct
31 Correct 799 ms 270272 KB Output is correct
32 Correct 890 ms 269760 KB Output is correct
33 Correct 1151 ms 270272 KB Output is correct
34 Correct 759 ms 270276 KB Output is correct
35 Correct 1 ms 348 KB Output is correct
36 Correct 0 ms 348 KB Output is correct
37 Correct 78 ms 4940 KB Output is correct
38 Correct 1090 ms 271048 KB Output is correct
39 Correct 1030 ms 269760 KB Output is correct
40 Correct 948 ms 269892 KB Output is correct
41 Correct 887 ms 270020 KB Output is correct
42 Correct 828 ms 270216 KB Output is correct
43 Correct 904 ms 269504 KB Output is correct
44 Correct 970 ms 269504 KB Output is correct
45 Correct 868 ms 269508 KB Output is correct
46 Correct 1069 ms 269504 KB Output is correct
47 Correct 968 ms 269508 KB Output is correct
48 Correct 1 ms 348 KB Output is correct
49 Correct 0 ms 348 KB Output is correct
50 Correct 0 ms 348 KB Output is correct
51 Correct 0 ms 436 KB Output is correct
52 Correct 1 ms 348 KB Output is correct
53 Correct 1 ms 604 KB Output is correct
54 Correct 11 ms 1628 KB Output is correct
55 Correct 9 ms 1624 KB Output is correct
56 Correct 12 ms 1544 KB Output is correct
57 Correct 10 ms 1624 KB Output is correct
58 Correct 10 ms 1540 KB Output is correct
59 Correct 10 ms 1592 KB Output is correct
60 Correct 8 ms 1628 KB Output is correct
61 Correct 55 ms 6736 KB Output is correct
62 Correct 106 ms 27856 KB Output is correct
63 Correct 550 ms 269400 KB Output is correct
64 Correct 735 ms 269504 KB Output is correct
65 Correct 937 ms 269760 KB Output is correct
66 Correct 866 ms 270016 KB Output is correct
67 Correct 809 ms 270272 KB Output is correct
68 Correct 797 ms 269764 KB Output is correct
69 Correct 884 ms 270016 KB Output is correct
70 Correct 1098 ms 271272 KB Output is correct
71 Correct 1206 ms 269764 KB Output is correct
72 Correct 962 ms 269764 KB Output is correct
73 Correct 887 ms 270016 KB Output is correct
74 Correct 593 ms 269568 KB Output is correct
75 Correct 805 ms 269704 KB Output is correct