Submission #1283271

#TimeUsernameProblemLanguageResultExecution timeMemory
1283271MMihalevMosaic (IOI24_mosaic)C++20
0 / 100
78 ms16840 KiB
#include<iostream>
#include<algorithm>
#include<vector>
#include "mosaic.h"
using namespace std;
const int MAX_N=2e5+5;
int n,q;
vector<long long>ans;
int a1[3][MAX_N];
int a2[MAX_N][3];
std::vector<long long> mosaic(std::vector<int> X, std::vector<int> Y,std::vector<int> T, std::vector<int> B,std::vector<int> L, std::vector<int> R)
{
    n=X.size();
    q=T.size();
    ans.resize(q);

    for(int i=0;i<n;i++)
    {
        a1[0][i]=X[i];
    }
    for(int i=0;i<n;i++)
    {
        a2[i][0]=Y[i];
    }

    a1[1][0]=a2[1][0];
    for(int i=1;i<n;i++)
    {
        a1[1][i]=(a1[1][i-1]+a1[0][i]==0 ? 1 : 0);
    }
    a2[0][1]=a1[0][1];
    for(int i=1;i<n;i++)
    {
        a2[i][1]=(a2[i-1][i]+a2[i][0]==0 ? 1 : 0);
    }

    a1[2][0]=a2[2][0];a1[2][1]=a2[2][1];
    for(int i=2;i<n;i++)
    {
        a1[2][i]=(a1[2][i-1]+a1[1][i]==0 ? 1 : 0);
    }
    a2[0][2]=a1[0][2];a2[1][2]=a1[1][2];
    for(int i=2;i<n;i++)
    {
        a2[i][2]=(a2[i-1][2]+a2[i][1]==0 ? 1 : 0);
    }

    for(int i=1;i<=q;i++)
    {
        if(T[i-1]<3)
        {
            ans[i-1]=a1[T[i-1]][L[i-1]];
            continue;
        }
        if(L[i-1]<3)
        {
            ans[i-1]=a2[T[i-1]][L[i-1]];
            continue;
        }

        if(T[i-1]<L[i-1])
        {
            int x=T[i-1]-2;
            ans[i-1]=a1[2][L[i-1]-x];
        }
        else
        {
            int x=L[i-1]-2;
            ans[i-1]=a2[T[i-1]-x][2];
        }
    }

    return ans;
}
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...