Submission #1193931

#	Time	Username	Problem	Language	Result	Execution time	Memory
1193931		simona1230	카니발 티켓 (IOI20_tickets)	C++20	11 / 100	4 ms	6724 KiB

tickets

#include "tickets.h"
#include <vector>
#include <bits/stdc++.h>
using namespace std;

int nn,mm;
int a[1501][1501];

int z;
struct help
{
    int x,y,i;
    help(){}
    help(int _x,int _y,int _i)
    {
        x=_x;
        y=_y;
        i=_i;
    }

    bool operator<(const help&h)const
    {
        return max(abs(h.y-z),abs(h.x-z))>max(abs(x-z),abs(y-z));
    }
};

vector<vector<int> > v;
int id[200001],maxx[200001],minn[200001];
long long try_(int x,int p)
{
    //cout<<x<<" !"<<endl;
    z=x;
    priority_queue<help> q;
    if(a[p][minn[p]]==x)id[p]=1;
    else id[p]=2;
    long long ans=0;
    int l=0,m=0;
    for(int i=0;i<nn;i++)
    {
        if(i==p)continue;
        if(a[i][maxx[i]]<x)
        {
            ans+=x-a[i][minn[i]];
            id[i]=1;
            l++;
        }
        else if(a[i][minn[i]]>x)
        {
            ans+=a[i][maxx[i]]-x;
            id[i]=2;
            m++;
        }
        else
        {
            q.push({a[i][minn[i]],a[i][maxx[i]],i});
        }
    }

    int l1=nn/2;
    if(nn%2==0)l1--;
    int l2=nn/2;

    if(l>l1)return -1;
    if(m>l2)return -1;
    //cout<<"pos"<<endl;

    while(q.size())
    {
        help h=q.top();
        q.pop();

        if(h.y-z>z-h.x&&m<l2||l==l1)
        {
            m++;
            id[h.i]=2;
            ans+=h.y-z;
        }
        else
        {
            l++;
            id[h.i]=1;
            ans+=z-h.x;
        }
    }

    return ans;
}
long long find_maximum(int k, std::vector<std::vector<int>> x)
{
    v=x;
	nn=x.size();
	mm=x[0].size();

	for(int i=0;i<nn;i++)
    {
        for(int j=0;j<mm;j++)
        {
            v[i][j]=-1;
            a[i][j]=x[i][j];
        }
    }

    if(mm==1)
    {
        vector<int> h;
        for(int i=0;i<nn;i++)
            h.push_back(a[i][0]);
        sort(h.begin(),h.end());
        long long sz=h.size()/2,ans=0;
        for(int i=0;i<nn;i++)
            ans+=abs(h[i]-h[sz]);
        for(int i=0;i<nn;i++)
            v[i][0]=0;
        allocate_tickets(v);
        return ans;
    }

    for(int i=0;i<nn;i++)
    {
        for(int j=1;j<mm;j++)
        {
            if(a[i][maxx[i]]<a[i][j])
                maxx[i]=j;
            if(a[i][minn[i]]>a[i][j])
                minn[i]=j;
        }
    }

    long long ans=-1;

    for(int i=0;i<nn;i++)
    {
        long long t1=try_(a[i][maxx[i]],i);
        if(ans<t1)
        {
            ans=t1;
            for(int j=0;j<nn;j++)
                if(id[j]==1)v[j][minn[j]]=0,v[j][maxx[j]]=-1;
                else v[j][minn[j]]=-1,v[j][maxx[j]]=0;
        }
        long long t2=try_(a[i][minn[i]],i);
        if(ans<t2)
        {
            ans=t2;
            for(int j=0;j<nn;j++)
                if(id[j]==1)v[j][minn[j]]=0,v[j][maxx[j]]=-1;
                else v[j][minn[j]]=-1,v[j][maxx[j]]=0;
        }
    }

    allocate_tickets(v);
	return ans;
}

• Subtask 1: 11 / 11
• Subtask 2: 0 / 16
• Subtask 3: 0 / 14
• Subtask 4: 0 / 14
• Subtask 5: 0 / 12
• Subtask 6: 0 / 23
• Subtask 7: 0 / 10