Submission #709529

#	Time	Username	Problem	Language	Result	Execution time	Memory
709529		TlenekWodoru	Catfish Farm (IOI22_fish)	C++17	100 / 100	228 ms	55512 KiB

fish

#include <bits/stdc++.h>
#include "fish.h"
using namespace std;
struct RYBA
{
    int y,c;
    long long wynik=0,wynik2=0;
    long long l=0,s=0,p=0;
    long long suma=0;
    RYBA(int yy=0, int cc=0)
    {
        y=yy;
        c=cc;
        wynik=0;wynik2=0;
        l=0;s=0,p=0;
        suma=0;
    }
    bool operator <(const RYBA &B)const
    {
        return y<B.y;
    }
};
vector<RYBA>V[100009];
long long max_weights(int N, int M, vector<int>X, vector<int>Y, vector<int>W)
{
    for(int i=0;i<M;i++)
    {
        V[X[i]+1].push_back(RYBA(Y[i]+1,W[i]));
        //cout<<"x="<<X[i]+1<<" y="<<Y[i]+1<<" w="<<W[i]<<endl;
    }
    for(int i=1;i<=N;i++)
    {
        V[i].push_back(RYBA(N+1,0));
        V[i].push_back(RYBA(0,0));
        sort(V[i].begin(),V[i].end());
    }
    for(int i=1;i<=N;i++)
    {
        int Hl=0,Hp=0;
        long long l=0,p=0,s=0;
        for(int j=0;j<V[i].size();j++)
        {
            while(Hl<V[i-1].size()&&V[i-1][Hl].y<=V[i][j].y-1)
            {
                l+=V[i-1][Hl].c;
                Hl++;
            }
            while(Hp<V[i+1].size()&&V[i+1][Hp].y<=V[i][j].y-1)
            {
                p+=V[i+1][Hp].c;
                Hp++;
            }
            V[i][j].l=l;
            V[i][j].s=s;
            V[i][j].p=p;
            V[i][j].suma=l+p;
            //cout<<"     x="<<i-1<<" y="<<V[i][j].y-1<<" c="<<V[i][j].c<<" l="<<l<<" s="<<s<<" p="<<p<<endl;
            s+=V[i][j].c;
        }
    }
    for(int i=0;i<V[1].size();i++)
    {
        V[1][i].wynik=V[1][i].suma;
        V[1][i].wynik2=V[1][i].suma;
    }
    long long WYNIK=0;
    for(int i=2;i<=N;i++)
    {
        int h;
        long long maxx;
        if(i>=2)
        {
            h=0;maxx=0;
            for(int j=0;j<V[i].size();j++)
            {
                while(h<V[i-2].size()&&V[i-2][h].y<=V[i][j].y)
                {
                    maxx=max(maxx,V[i-2][h].wynik-V[i-2][h].p);
                    h++;
                }
                V[i][j].wynik=max(V[i][j].wynik,maxx+V[i][j].suma);
            }
            h=V[i-2].size()-1;maxx=0;
            for(int j=V[i].size()-1;j>=0;j--)
            {
                while(h>=0&&V[i][j].y<=V[i-2][h].y)
                {
                    maxx=max(maxx,V[i-2][h].wynik);
                    h--;
                }
                V[i][j].wynik=max(V[i][j].wynik,maxx+V[i][j].p);
                V[i][j].wynik2=V[i][j].wynik;
            }
        }
        h=0;maxx=0;
        for(int j=0;j<V[i].size();j++)
        {
            while(h<V[i-1].size()&&V[i-1][h].y<=V[i][j].y)
            {
                maxx=max(maxx,V[i-1][h].wynik2-V[i-1][h].p-V[i-1][h].s);
                h++;
            }
            V[i][j].wynik=max(V[i][j].wynik,maxx+V[i][j].suma);
            V[i][j].wynik2=max(V[i][j].wynik2,maxx+V[i][j].suma);
        }
        h=V[i-1].size()-1;maxx=0;
        for(int j=V[i].size()-1;j>=0;j--)
        {
            while(h>=0&&V[i][j].y<=V[i-1][h].y)
            {
                maxx=max(maxx,V[i-1][h].wynik);
                h--;
            }
            V[i][j].wynik=max(V[i][j].wynik,maxx-V[i][j].s+V[i][j].p);
            WYNIK=max(WYNIK,V[i][j].wynik);
        }
    }
    return WYNIK;
}

Compilation message (stderr)

fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:41:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |         for(int j=0;j<V[i].size();j++)
      |                     ~^~~~~~~~~~~~
fish.cpp:43:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |             while(Hl<V[i-1].size()&&V[i-1][Hl].y<=V[i][j].y-1)
      |                   ~~^~~~~~~~~~~~~~
fish.cpp:48:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |             while(Hp<V[i+1].size()&&V[i+1][Hp].y<=V[i][j].y-1)
      |                   ~~^~~~~~~~~~~~~~
fish.cpp:61:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |     for(int i=0;i<V[1].size();i++)
      |                 ~^~~~~~~~~~~~
fish.cpp:74:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |             for(int j=0;j<V[i].size();j++)
      |                         ~^~~~~~~~~~~~
fish.cpp:76:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   76 |                 while(h<V[i-2].size()&&V[i-2][h].y<=V[i][j].y)
      |                       ~^~~~~~~~~~~~~~
fish.cpp:96:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   96 |         for(int j=0;j<V[i].size();j++)
      |                     ~^~~~~~~~~~~~
fish.cpp:98:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<RYBA>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   98 |             while(h<V[i-1].size()&&V[i-1][h].y<=V[i][j].y)
      |                   ~^~~~~~~~~~~~~~

• Subtask 1: 3 / 3
• Subtask 2: 6 / 6
• Subtask 3: 9 / 9
• Subtask 4: 14 / 14
• Subtask 5: 21 / 21
• Subtask 6: 17 / 17
• Subtask 7: 14 / 14
• Subtask 8: 16 / 16