Submission #812838

#TimeUsernameProblemLanguageResultExecution timeMemory
812838andrei_boacaThousands Islands (IOI22_islands)C++17
31 / 100
58 ms38604 KiB
#include "islands.h"

#include <variant>
#include <vector>
#include <algorithm>
#include <assert.h>
//#include <bits/stdc++.h>
//#include "grader.cpp"
using namespace std;
struct date
{
    int nod,a,b;
};
vector<date> muchii[100005];
vector<int> c[1005][1005];
bool useedge[300005],usenod[300005];
int n,m;
bool good=0;
int who;
vector<int> edges,v;
vector<int> nodes,w;
vector<date> ee[100005];
void dfs(int nod)
{
    if(good)
        return;
    if(muchii[nod].size()>=3)
    {
        who=nod;
        good=1;
        return;
    }
    usenod[nod]=1;
    for(date i:muchii[nod])
    {
        if(good)
            return;
        if(!usenod[i.nod])
        {
            edges.push_back(i.a);
            dfs(i.nod);
            if(good)
                return;
        }
    }
}
vector<int> sol;
void build(int x,int y,int z)
{
    assert(x>=0&&y>=0&&z>=0);
    int A=c[x][y][0];
    int B=c[y][x][0];
    int C=c[x][z][0];
    int D=c[z][x][0];
    if(y==z)
    {
        C=c[x][z][1];
        D=c[z][x][1];
    }
    vector<int> aux={A,B,C,D,B,A,D,C};
    for(int i:aux)
        sol.push_back(i);
}
bool use[200005];
void calc(int nod)
{
    use[nod]=1;
    for(date i:ee[nod])
    {
        if(use[i.nod])
        {
            good=1;
            return;
        }
        calc(i.nod);
    }
}
std::variant<bool, std::vector<int>> find_journey(int N, int M, std::vector<int> U, std::vector<int> V)
{

    n=N;
    m=M;
    for(int i=0;i<m;i++)
    {
        int a=U[i];
        int b=V[i];
        c[a][b].push_back(i);
    }
    if(n==2)
    {
        if(c[0][1].size()>=2&&c[1][0].size()>=1)
        {
            int A=c[1][0][0];
            int B=c[0][1][0];
            int C=c[0][1][1];
            sol={B,A,C,B,A,C};
            return sol;
        }
        return false;
    }
    for(int a=0;a<n;a++)
        for(int b=0;b<n;b++)
            if(a!=b)
            {
                for(int i=0;i<c[a][b].size()&&i<c[b][a].size();i++)
                {
                    int x=c[a][b][i];
                    int y=c[b][a][i];
                    muchii[a].push_back({b,x,y});
                }
            }
    if(muchii[0].size()>=2)
    {
        int x=0;
        int y=muchii[0][0].nod;
        int z=muchii[0][1].nod;
        build(x,y,z);
        return sol;
    }
    dfs(0);
    if(good)
    {
        for(int i:edges)
            sol.push_back(i);
        int x=who;
        int y=-1;
        int z=-1;
        for(date i:muchii[who])
            if(!usenod[i.nod])
            {
                if(y==-1)
                    y=i.nod;
                else
                    z=i.nod;
            }
        build(x,y,z);
        reverse(edges.begin(),edges.end());
        for(int i:edges)
            sol.push_back(i);
        return sol;
    }
    for(int a=0;a<n;a++)
        for(int b=0;b<n;b++)
        {
            for(int i=0;i+1<c[a][b].size();i+=2)
            {
                int x=c[a][b][i];
                int y=c[a][b][i+1];
                ee[a].push_back({b,x,y});
            }
        }
    good=0;
    calc(0);
    if(good)
        return true;
    return false;
}

Compilation message (stderr)

islands.cpp: In function 'std::variant<bool, std::vector<int, std::allocator<int> > > find_journey(int, int, std::vector<int>, std::vector<int>)':
islands.cpp:105:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  105 |                 for(int i=0;i<c[a][b].size()&&i<c[b][a].size();i++)
      |                             ~^~~~~~~~~~~~~~~
islands.cpp:105:48: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  105 |                 for(int i=0;i<c[a][b].size()&&i<c[b][a].size();i++)
      |                                               ~^~~~~~~~~~~~~~~
islands.cpp:145:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  145 |             for(int i=0;i+1<c[a][b].size();i+=2)
      |                         ~~~^~~~~~~~~~~~~~~
#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...