Submission #1286512

#TimeUsernameProblemLanguageResultExecution timeMemory
1286512eri16Obstacles for a Llama (IOI25_obstacles)C++20
24 / 100
88 ms34112 KiB
#include <bits/stdc++.h>
using namespace std;

int arr1[200005];

int arr2[2][200005];
int vis[2][200005]={0};

int dx[4]={0,0,-1,1};
int dy[4]={-1,1,0,0};

int kk=0;

void dfs(int i, int j, int col, int n) {
    if (i < 0 || i >= 2 || j < 0 || j >= n) return;
    if (arr2[i][j]==0) return;
    if (vis[i][j]) return;

    vis[i][j] = 1;
    arr2[i][j] = col;

    for (int z = 0; z < 4; z++) {
        dfs(i + dx[z], j + dy[z], col, n);
    }
}


void initialize(std::vector<int> T, std::vector<int> H){
    
    if (T.size()==1){
        kk=1;
        for (int i=0; i<H.size(); i++){
            arr1[i]=(T[0]>H[i]) ? 1 : 0;
        }
        
        int k=0;
        
        for (int i=0; i<H.size(); i++){
            
            if (arr1[i]==1){arr1[i]=k;}
            else{
                arr1[i]=-1;
                if (i>0){
                    if (arr1[i-1]!=-1){
                        k++;
                    }
                }
            }
        }
    }
    
    else{
        kk=2;
        for (int i=0; i<H.size(); i++){
            arr2[0][i]=(T[0]>H[i]) ? 1 : 0;
        }
        
        for (int i=0; i<H.size(); i++){
            arr2[1][i]=(T[T.size()-1]>H[i]) ? 1 : 0;
        }        
        
        int k=1;
        
        vis[0][H.size()]=1;
        vis[1][H.size()]=1;
        
        for (int i=0; i<2; i++){
            for (int j=0; j<H.size(); j++){if (vis[i][j]==0){dfs(i,j,k,H.size());k++;}
        }}
    }
}


bool can_reach(int L, int R, int S, int D){
    if (kk==0){
        return true;
    }
    if (kk==1){
        if (arr1[S]==arr1[D] && arr1[S]!=-1){return true;}
    }
    if (kk==2){
        if (arr2[0][S]==arr2[0][D] && arr2[0][S]!=0){return true;}        
    }
    
    return false;
}
    
    
    
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...