Submission #539143

#	Time	Username	Problem	Language	Result	Execution time	Memory
539143		vector	Rope (JOI17_rope)	C++17	80 / 100	2552 ms	147948 KiB

rope

#include<bits/stdc++.h>
#define SIZE 1000001
using namespace std;
int N,M,A[SIZE],B[SIZE],C[SIZE],D[SIZE],ans[SIZE];
vector<int>v1[SIZE],v2[SIZE];
multiset<pair<int,int> >st;
int main()
{
    ios_base::sync_with_stdio(0),cin.tie(0);
    cin>>N>>M;
    for(int i=1;i<=N;i++){
        cin>>A[i];
        B[A[i]]++;
    }
    for(int i=1;i<=M;i++){
        st.insert({-B[i],i});
        ans[i]=1e9;
    }
    for(int i=1;i<N;i+=2){
        v1[A[i]].push_back(A[i+1]);
        v1[A[i+1]].push_back(A[i]);
    }
    for(int i=1;i<=M;i++){
        set<int>u;
        C[i]=-B[i];
        u.insert(i);
        for(int j:v1[i]){
            if(u.find(j)==u.end()){
                C[j]=-B[j];
                u.insert(j);
            }
            st.erase({C[j],j});
            st.insert({++C[j],j});
        }
        st.erase({C[i],i});
        if(!st.empty())ans[i]=min(ans[i],N-B[i]+st.begin()->first);
        else ans[i]=0;
        st.insert({C[i],i});
        for(int j:u){
            st.erase({C[j],j});
            st.insert({-B[j],j});
        }
    }
    for(int i=2;i<N;i+=2){
        v2[A[i]].push_back(A[i+1]);
        v2[A[i+1]].push_back(A[i]);
    }
    for(int i=1;i<=M;i++){
        set<int>u;
        C[i]=-B[i];
        u.insert(i);
        for(int j:v2[i]){
            if(u.find(j)==u.end()){
                C[j]=-B[j];
                u.insert(j);
            }
            st.erase({C[j],j});
            st.insert({++C[j],j});
        }
        st.erase({C[i],i});
        if(!st.empty())ans[i]=min(ans[i],N-B[i]+st.begin()->first);
        else ans[i]=0;
        st.insert({C[i],i});
        for(int j:u){
            st.erase({C[j],j});
            st.insert({-B[j],j});
        }
        printf("%d\n",ans[i]);
    }
}

• Subtask 1: 15 / 15
• Subtask 2: 30 / 30
• Subtask 3: 10 / 10
• Subtask 4: 25 / 25
• Subtask 5: 0 / 20