Submission #301487

#TimeUsernameProblemLanguageResultExecution timeMemory
301487daniel920712Comparing Plants (IOI20_plants)C++14
5 / 100
4070 ms21496 KiB
#include "plants.h"
#include <stdio.h>
#include <vector>
#include <utility>
using namespace std;
vector < pair < int , int > > con[200005];
vector < int > all;
vector < int > Next[200005];
int N,tt=0;
bool use[2000005];
void init(int k,vector<int> r)
{
    int a=0,b=0,i,j,ok=0;
    all=r;
    N=r.size();
    if(k==2) tt=1;
    for(i=0;i<N;i++)
    {
        if(k==2)
        {
            if(r[(i+N-1)%N]!=r[i])
            {
                b=0;
                a++;
                con[i].push_back(make_pair(a,0));
                for(j=(i+1)%N;r[(j+N-1)%N]==r[i];j=(j+1)%N)
                {
                    if(j==N-1) ok=1;
                    if(r[i]==0) b--;
                    else b++;
                    con[j].push_back(make_pair(a,b));
                }
            }
        }
        else
        {
            for(i=0;i<N;i++)
            {
                if(r[(i+N-1)%N]!=r[i])
                {
                    for(j=(i+1)%N;r[j]==r[i];j=(j+1)%N)
                    {
                        //printf("%d %d\n",i,j);
                        if(r[i]>r[(i+N-1)%N]) Next[(j+N-1)%N].push_back(j);
                        else Next[j].push_back((j+N-1)%N);
                    }
                }
                for(j=i+1;j<N;j++)
                {
                    if(r[i]>r[j]) Next[i].push_back(j);
                    if(r[i]<r[j]) Next[j].push_back(i);

                }
            }
        }
    }
	return;
}
bool F(int a,int b)
{
    if(a==b) return 1;
    if(use[a]) return 0;
    use[a]=1;
    for(auto i:Next[a]) if(F(i,b)) return 1;
    return 0;
}
int compare_plants(int x, int y)
{
    int i;
    if(tt)
    {
        for(auto i:con[x])
        {
            for(auto j:con[y])
            {
                if(i.first==j.first)
                {
                    if(i.second<j.second) return -1;
                    return 1;
                }
            }
        }

    }
    else
    {
        for(i=0;i<N;i++) use[i]=0;
        if(F(x,y)) return -1;
        else if(F(y,x)) return 1;
    }

	return 0;
}

Compilation message (stderr)

plants.cpp: In function 'void init(int, std::vector<int>)':
plants.cpp:13:21: warning: variable 'ok' set but not used [-Wunused-but-set-variable]
   13 |     int a=0,b=0,i,j,ok=0;
      |                     ^~
#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...