109722024-04-24 17:28:44KristófZebra (75 pont)cpp17Wrong answer 0/753ms3264 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;
        int idoseged2;
        if(nulla[0]<egy[0])
            {
            idoseged=nulla[0];
            idoseged2=nulla[0];
            }
        else
            {
            idoseged=egy[0];
            idoseged2=egy[0];
            }
        vector<int>::iterator it;
        sort(nulla.begin(),nulla.end());
        sort(egy.begin(),egy.end());
        for(int i=0;i<n;i++)
            {
            if(idoseged!=idoseged2)
                {
                ans+=vala1.size();
                ans+=vala0.size();

                idoseged2=idoseged;
                ans--;
                //cout<<ans<<" "<<"answer"<<vala0.size()<<" "<<vala1.size()<<endl;
                if(egy.size()<1 || nulla.size()<1)
                    {
                    //cout<<"in&out"<<endl;
                    if(nulla.size()<1)
                {
                for(int i=0;i<egy.size();i++)
                    {
                    vala1.push_back(egy[i]);
                    idoseged2=egy[i];
                    if(idoseged2!=idoseged)
                        {

                        ans+=vala0.size();
                        ans+=vala1.size();
                        idoseged=idoseged2;
                        ans--;
                        }

                    }

                break;
                }
            if(egy.size()<1)
                {
                for(int i=0;i<nulla.size();i++)
                    {
                    vala0.push_back(egy[i]);
                    idoseged2=nulla[i];
                    if(idoseged2!=idoseged)
                        {
                        ans+=vala0.size();
                        ans+=vala1.size();
                        idoseged=idoseged2;
                        ans--;
                        }

                    }


                break;
                }
                    }
                //cout<<"cleared"<<endl;
                if(vala1.size()>0 && vala0.size()>0){
                vala0.clear();
                vala1.clear();
                }
                }
                //cout<<idoseged2<<" "<<idoseged<<endl;


                if(egy[0]<nulla[0])
                {
                vala1.push_back(egy[0]);
                idoseged=egy[0];
                it=egy.begin();
                egy.erase(it);
                }
            else
                {
                vala0.push_back(nulla[0]);
                idoseged=nulla[0];
                it=nulla.begin();
                nulla.erase(it);

                }
            }



            cout<<ans<<endl;

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base0/75
1Accepted0/03ms1956 KiB
2Wrong answer0/03ms1916 KiB
3Wrong answer0/53ms2100 KiB
4Wrong answer0/53ms2320 KiB
5Wrong answer0/53ms2400 KiB
6Wrong answer0/53ms2524 KiB
7Wrong answer0/53ms2396 KiB
8Wrong answer0/53ms2532 KiB
9Wrong answer0/53ms2736 KiB
10Wrong answer0/53ms2912 KiB
11Wrong answer0/53ms2996 KiB
12Wrong answer0/53ms2992 KiB
13Wrong answer0/53ms2988 KiB
14Wrong answer0/53ms2992 KiB
15Wrong answer0/53ms3264 KiB
16Wrong answer0/53ms3240 KiB
17Wrong answer0/53ms3232 KiB