QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#218097#7079. ArrayKLPP#TL 0ms11864kbC++203.2kb2023-10-17 18:28:402023-10-17 18:28:41

Judging History

你现在查看的是最新测评结果

  • [2023-10-17 18:28:41]
  • 评测
  • 测评结果:TL
  • 用时:0ms
  • 内存:11864kb
  • [2023-10-17 18:28:40]
  • 提交

answer

#include<bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
 
using namespace std;
using namespace __gnu_pbds;
typedef long long int lld;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> ordered_set;
#define rep(i,a,b) for(int i=a;i<b;i++)
#define trav(a,v) for(auto a:v)
int n;
lld arr[2000000];
int nxt[2000000];
int prv[2000000];
map<lld,int> m;
set<lld> s;
lld sm(lld x, lld y){
    lld ans=(y-x+1);
    ans*=(x+y);
    ans/=2;
    return ans;
}
vector<lld >V;
const int MX=2'000'000;
class ST{
    lld mn[MX*4];
    public:
    void build(int a, int b, int node){
        mn[node]=0;
        if(a==b)return;
        int mid=(a+b)/2;
        build(a,mid,2*node);
        build(mid+1,b,2*node+1);
    }
    void update(int a, int b, int node, int pos, lld val){
        if(pos<a || b<pos)return;
        if(a==b){
            mn[node]=val;
            return;
        }
        int mid=(a+b)/2;
        update(a,mid,2*node,pos,val);
        update(mid+1,b,2*node+1,pos,val);
        mn[node]=min(mn[2*node],mn[2*node+1]);
    }
    lld query(int a, int b, int node, int l, int r){
        if(r<a || b<l)return 1e18;
        if(l<=a && b<=r)return mn[node];
        int mid=(a+b)/2;
        return min(query(a,mid,2*node,l,r),query(mid+1,b,2*node+1,l,r));
    }
};
ST S1,S2;
void solve(){
    m.clear();
    s.clear();
    cin>>n;
    rep(i,0,n)cin>>arr[i];
    rep(i,0,n)V.push_back(arr[i]);
    sort(V.begin(),V.end());
    V.resize(unique(V.begin(),V.end())-V.begin());
    int M=V.size();
    S1.build(0,M-1,1);
    S2.build(0,M-1,1);
    for(int i=n-1;i>-1;i--){
        if(m.find(arr[i])==m.end())nxt[i]=n;
        else nxt[i]=m[arr[i]];
        m[arr[i]]=i;
    }
    m.clear();
    rep(i,0,n){
        if(m.find(arr[i])==m.end())prv[i]=-1;
        else prv[i]=m[arr[i]];
        m[arr[i]]=i;
    }
    trav(a,m){
        lld X=a.second;
        S1.update(0,M-1,1,lower_bound(V.begin(),V.end(),a.first)-V.begin(),-(X*(X+1))/2-a.first);
        S2.update(0,M-1,1,lower_bound(V.begin(),V.end(),a.first)-V.begin(),-(X*(X+1))/2+a.first);
    }
    lld best=0;
    rep(i,0,n){
        S1.update(0,M-1,1,lower_bound(V.begin(),V.end(),arr[i])-V.begin(),1e18);
        S2.update(0,M-1,1,lower_bound(V.begin(),V.end(),arr[i])-V.begin(),1e18);
        if(prv[i]==-1){
            if((int)s.size()>0){
                set<lld>::iterator it=s.lower_bound(arr[i]);
                if(it!=s.end()){
                    best=min(best,*it-arr[i]-sm(i+1,nxt[i]));
                }
                if(it!=s.begin()){
                    best=min(best,arr[i]-*prev(it)-sm(i+1,nxt[i]));
                }
            }
            best=min(best,S1.query(0,M-1,1,0,lower_bound(V.begin(),V.end(),arr[i])-V.begin())+arr[i]);
            best=min(best,S2.query(0,M-1,1,lower_bound(V.begin(),V.end(),arr[i])-V.begin(),M-1)-arr[i]);
        }
        s.insert(arr[i]);
    }
    s.clear();
    lld ans=0;
    rep(i,0,n){
        s.insert(arr[i]);
        lld can=i+1;
        can*=s.size();
        ans+=can;
    }
    cout<<ans+best<<"\n";
}

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	int tt=1;
	cin>>tt;
	while(tt--){
		solve();
	}
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 11864kb

input:

1
4
1 2 3 4

output:

22

result:

ok "22"

Test #2:

score: -100
Time Limit Exceeded

input:

100000
10
873324878 873324878 873324878 873324878 891656676 891656676 615245360 873324878 873324878 873324878
10
560723194 560723194 797429144 797429144 560723194 797429144 819647695 560723194 797429144 560723194
10
750627649 746781323 756277046 756277046 750627649 750627649 756277046 750627649 9142...

output:

134
-819647554
-756276866
-872600887
-423989387
-610290228
-507254268
-510756791
-640888920
-706795646
-881357583
-980047416
-704065665
-486154446
-506414938
-876412033
-917181620
-792686654
-935042515
-382037532
-878758402
-905001347
-857119927
-720743105
-478678121
-966789263
-971371692
-749444536...

result: