Submission #431846

#	Time	Username	Problem	Language	Result	Execution time	Memory
431846		Bench0310	Seats (IOI18_seats)	C++17	5 / 100	4081 ms	133792 KiB

seats

#include <bits/stdc++.h>
#include "seats.h"

using namespace std;
typedef long long ll;

int n,h,w;
vector<array<int,2>> v;
vector<int> ar,br,ac,bc;
int res=0;
const int N=1000;
set<int> row[N];
vector<int> rowmn(N,N*N);
set<int> col[N];
vector<int> colmn(N,N*N);

int ok(int i){return ((br[i]-ar[i]+1)*(bc[i]-ac[i]+1)==i+1);}

void give_initial_chart(int nh,int nw,vector<int> r,vector<int> c)
{
    h=nh;
    w=nw;
    n=h*w;
    v.assign(n,{0,0});
    for(int i=0;i<n;i++) v[i]={r[i],c[i]};
    ar.assign(n,v[0][0]);
    br.assign(n,v[0][0]);
    ac.assign(n,v[0][1]);
    bc.assign(n,v[0][1]);
    res=ok(0);
    for(int i=1;i<n;i++)
    {
        ar[i]=min(ar[i-1],v[i][0]);
        br[i]=max(br[i-1],v[i][0]);
        ac[i]=min(ac[i-1],v[i][1]);
        bc[i]=max(bc[i-1],v[i][1]);
        res+=ok(i);
    }
    for(int i=0;i<h;i++) row[i].insert(h*w);
    for(int i=0;i<w;i++) col[i].insert(h*w);
    for(int i=0;i<n;i++)
    {
        row[v[i][0]].insert(i);
        col[v[i][1]].insert(i);
    }
    for(int i=0;i<h;i++) rowmn[i]=(*row[i].begin());
    for(int i=0;i<w;i++) colmn[i]=(*col[i].begin());
}

int swap_seats(int a,int b)
{
    if(a>b) swap(a,b);
    row[v[a][0]].erase(a);
    col[v[a][1]].erase(a);
    row[v[b][0]].erase(b);
    col[v[b][1]].erase(b);
    swap(v[a],v[b]);
    row[v[a][0]].insert(a);
    col[v[a][1]].insert(a);
    row[v[b][0]].insert(b);
    col[v[b][1]].insert(b);
    rowmn[v[a][0]]=(*row[v[a][0]].begin());
    rowmn[v[b][0]]=(*row[v[b][0]].begin());
    colmn[v[a][1]]=(*col[v[a][1]].begin());
    colmn[v[b][1]]=(*col[v[b][1]].begin());
    if(h>1000||w>1000)
    {
        for(int i=a;i<=b;i++)
        {
            res-=ok(i);
            ar[i]=min((i>0?ar[i-1]:n),v[i][0]);
            br[i]=max((i>0?br[i-1]:0),v[i][0]);
            ac[i]=min((i>0?ac[i-1]:n),v[i][1]);
            bc[i]=max((i>0?bc[i-1]:0),v[i][1]);
            res+=ok(i);
        }
        return res;
    }
    else
    {
        int r=0;
        int now=0;
//        int cnt=0;
//        cout << "in" << endl;
        while(now<n)
        {
//            cout << now << " ";
            int c=h;
            for(int i=h-1;i>=0;i--) if(rowmn[i]<=now) c=i;
            int d=-1;
            for(int i=0;i<h;i++) if(rowmn[i]<=now) d=i;
            int e=w;
            for(int i=w-1;i>=0;i--) if(colmn[i]<=now) e=i;
            int f=-1;
            for(int i=0;i<w;i++) if(colmn[i]<=now) f=i;
//            cout << "[" << c << "," << d << "," << e << "," << f << "] ";
//            cout.flush();
            if((d-c+1)*(f-e+1)==now+1) r++;
            now=max(now+1,(d-c+1)*(f-e+1)-1);
//            cnt++;
        }
//        assert(cnt<=2*(w+h));
//        cout << endl;
        return r;
    }
}

//int main()
//{
//    
//    return 0;
//}

• Subtask 1: 5 / 5
• Subtask 2: 0 / 6
• Subtask 3: 0 / 20
• Subtask 4: 0 / 6
• Subtask 5: 0 / 33
• Subtask 6: 0 / 30