제출 #1151435

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1151435		irmuun	Group Photo (JOI21_ho_t3)	C++20	100 / 100	2949 ms	196672 KiB

Main

#include <bits/stdc++.h>

using namespace std;

#define ll long long
#define pb push_back
#define ff first
#define ss second
#define all(s) s.begin(),s.end()
#define rall(s) s.rbegin(),s.rend()

struct segtree{
    vector<int>d;
    segtree(int n){
        d.resize(4*n);
        build(1,1,n);
    }
    void build(int v,int l,int r){
        if(l==r){
            d[v]=0;
            return;
        }
        int m=(l+r)>>1;
        build(v<<1,l,m);
        build(v<<1|1,m+1,r);
        d[v]=d[v<<1]+d[v<<1|1];
    }
    int query(int v,int l,int r,int L,int R){
        if(R<L||r<L||R<l) return 0;
        if(L<=l&&r<=R) return d[v];
        int m=(l+r)>>1;
        return query(v<<1,l,m,L,R)+query(v<<1|1,m+1,r,L,r);
    }
    void update(int v,int l,int r,int p,int k){
        if(r<p||p<l) return;
        if(l==r){
            d[v]=k;
            return;
        }
        int m=(l+r)>>1;
        update(v<<1,l,m,p,k);
        update(v<<1|1,m+1,r,p,k);
        d[v]=d[v<<1]+d[v<<1|1];
    }
};

int main(){
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    int n;
    cin>>n;
    int p[n+5],f[n+5];
    for(int i=1;i<=n;i++){
        cin>>p[i];
        f[p[i]]=i;
    }
    int need[n+5][n+5];
    memset(need,0,sizeof need);
    segtree sg(n);
    for(int r=1;r<=n;r++){
        int cur=0;
        need[r][r]=0;
        for(int l=r-1;l>=1;l--){
            sg.update(1,1,n,f[l+1],1);
            cur+=sg.query(1,1,n,f[l]+1,n);
            need[l][r]=cur;
        }
        for(int l=1;l<=r;l++){
            sg.update(1,1,n,f[l],0);
        }
    }
    int cnt[n+5][n+5];
    memset(cnt,0,sizeof cnt);
    for(int l=1;l<=n;l++){
        int left[n+5],len=0;
        int pre[n+5],pos[n+5];
        for(int i=1;i<=n;i++){
            if(p[i]>=l){
                len++;
                pos[p[i]]=len;
            }
        }
        pre[0]=0;
        for(int i=l;i<=n;i++){
            pre[i-l+1]=pre[i-l]+pos[i];
        }
        int cur=0;
        for(int r=l;r<=n;r++){
            int x=r-l+1;
            cnt[l][r]=need[l][r]+pre[x]-(x*(x+1)/2);
        }
    }
    int dp[n+5];
    fill(dp,dp+n+1,1e9);
    dp[0]=0;
    for(int l=1;l<=n;l++){
        for(int r=l;r<=n;r++){
            dp[r]=min(dp[r],dp[l-1]+cnt[l][r]);
        }
    }
    cout<<dp[n];
}

• Subtask 1: 5 / 5
• Subtask 2: 7 / 7
• Subtask 3: 32 / 32
• Subtask 4: 20 / 20
• Subtask 5: 36 / 36