109792024-04-25 15:17:38KristófZebra (75 pont)cpp17Wrong answer 0/753ms3892 KiB
#include <iostream>
#include <vector>
#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n;
    cin>>n;
    int old[n];
    int ido[n];
    for(int i=0;i<n;i++)
        {
        cin>>old[i];

        }
    for(int i=0;i<n;i++)
        {
        cin>>ido[i];
        }
    vector<int>egy;
    vector<int>nulla;
    vector<int>vala1;
    vector<int>vala0;
    int ans=0;
    for(int i=0;i<n;i++)
        {
        if(old[i]==1)
            {
            egy.push_back(ido[i]);
            }
        else
            {
            nulla.push_back(ido[i]);
            }
        }
        int idoseged=-1;
        int idoseged2=-1;
        vector<int>::iterator it;
        sort(nulla.begin(),nulla.end());
        sort(egy.begin(),egy.end());
        for(int i=0;i<n;i++)
            {
            if(egy[0]<nulla[0])
                {
                //cout<<"egy"<<" "<<egy[0]<<" "<<idoseged2<<endl;
                vala1.push_back(egy[0]);
                idoseged=egy[0];
                it=egy.begin();
                egy.erase(it);
                }
            else
                {
                //cout<<"nulla"<<endl;
                vala0.push_back(nulla[0]);
                idoseged=nulla[0];
                it=nulla.begin();
                nulla.erase(it);
                }
            //cout<<idoseged<<" "<<idoseged2<<endl;
            if(idoseged!=idoseged2)
                {
                ans+=vala1.size();
                ans+=vala0.size();
                ans--;
                idoseged2=idoseged;
                //cout<<ans<<" "<<idoseged<<" "<<idoseged2<<endl;
                }

            if(nulla.size()<1)
                {
                for(int i=0;i<egy.size();i++)
                    {
                    vala1.push_back(egy[i]);
                    idoseged=egy[i];
                    if(idoseged!=idoseged2)
                        {
                        ans+=vala1.size();
                        ans+=vala0.size();
                        ans--;
                        idoseged2=idoseged;
                        }
                    }
                break;
                }
            if(egy.size()<1)
                {
                for(int i=0;i<nulla.size();i++)
                    {
                    vala1.push_back(egy[i]);
                    idoseged=egy[i];
                    if(idoseged!=idoseged2)
                        {
                        ans+=vala1.size();
                        ans+=vala0.size();
                        ans--;
                        idoseged2=idoseged;
                        }
                    }
                break;
                }
             if(vala0.size()>0 && vala1.size()>0)
                {
                vala0.clear();
                vala1.clear();
                }
            }
            cout<<ans;
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base0/75
1Accepted0/03ms1680 KiB
2Wrong answer0/03ms1844 KiB
3Wrong answer0/53ms2092 KiB
4Wrong answer0/53ms2344 KiB
5Wrong answer0/53ms2516 KiB
6Wrong answer0/53ms2600 KiB
7Wrong answer0/53ms2820 KiB
8Wrong answer0/53ms2868 KiB
9Wrong answer0/53ms3104 KiB
10Wrong answer0/53ms3184 KiB
11Wrong answer0/53ms3200 KiB
12Wrong answer0/53ms3284 KiB
13Wrong answer0/53ms3416 KiB
14Wrong answer0/53ms3656 KiB
15Wrong answer0/53ms3640 KiB
16Wrong answer0/53ms3764 KiB
17Wrong answer0/53ms3892 KiB