QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#754106#6433. Klee in Solitary Confinementnahida_qaqWA 1ms3860kbC++231.9kb2024-11-16 14:13:472024-11-16 14:13:48

Judging History

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

  • [2024-11-16 14:13:48]
  • 评测
  • 测评结果:WA
  • 用时:1ms
  • 内存:3860kb
  • [2024-11-16 14:13:47]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define pb push_back
#define ff first
#define ss second
typedef pair<int, int> PII;
const int N = 1000007;
const int inf = 1e18;
const int mod = 1e9+7;

int a[N];
unordered_map<int,int>vis;
unordered_map<int,vector<int>>v;
void solve() {
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        v[a[i]].pb(i);
    }
    int ans=0;
    if(k==0){
        for(auto i:v){
            int tt=i.ss.size();
            ans=max(ans,tt);
        }
        cout<<ans<<'\n';
        return;
    }
    for(int i=1;i<=n;i++){
        if(vis[a[i]])continue;
        vis[a[i]]=1;
        int x=a[i],y=a[i]-k;
        if(!v.count(y)){
            int z=v[x].size();
            ans=max(ans,z);
            continue;
        }
        vector<PII>cnt;
        for(auto j:v[x])cnt.pb({j,1});
        for(auto j:v[y])cnt.pb({j,2});
        sort(cnt.begin(),cnt.end());
        vector<int>pr1(cnt.size()+1);
        vector<int>pr2(cnt.size()+1);
        pr1[0]=(cnt[0].ss==1);
        pr2[0]=(cnt[0].ss==2);
        for(int j=1;j<cnt.size();j++){
            pr1[j]=pr1[j-1]+(cnt[j].ss==1);
            pr2[j]=pr2[j-1]+(cnt[j].ss==2);
        }
        int l=0;
        int sum=0;
        for(int j=0;j<cnt.size();j++){
            if(cnt[j].ss==1)sum--;
            else sum++;
            if(sum<0)sum=0;
            if(sum==1){
                l=j;
            }
            if(cnt[j].ss==2){
                int tmp=0;
                tmp+=pr2[j];
                if(l)tmp+=pr1[l-1]-pr2[l-1];
                tmp+=pr1[cnt.size()-1]-pr1[j];
               // if(tmp>ans)cout<<l<<' '<<j<<'\n';
                ans=max(ans,tmp);
            }
        }
    }
    cout<<ans<<'\n';
}
signed main() {
    ios;
    int _ = 1;
    //cin >> _;
    while (_--) {
        solve();
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

5 2
2 2 4 4 4

output:

5

result:

ok 1 number(s): "5"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3556kb

input:

7 1
3 2 3 2 2 2 3

output:

6

result:

ok 1 number(s): "6"

Test #3:

score: 0
Accepted
time: 0ms
memory: 3616kb

input:

7 1
2 3 2 3 2 3 3

output:

5

result:

ok 1 number(s): "5"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3560kb

input:

9 -100
-1 -2 1 2 -1 -2 1 -2 1

output:

3

result:

ok 1 number(s): "3"

Test #5:

score: 0
Accepted
time: 1ms
memory: 3616kb

input:

200 121649
0 527189 -1000000 -306471 -998939 527189 -1000000 -1000000 0 527189 0 527189 0 527189 -306471 -998939 -306471 -306471 -306471 0 0 527189 527189 1000000 527189 -1000000 1000000 648838 -1000000 -998939 -998939 -998939 0 1000000 -1000000 -998939 527189 1000000 648838 -1000000 1000000 648838 ...

output:

37

result:

ok 1 number(s): "37"

Test #6:

score: 0
Accepted
time: 0ms
memory: 3600kb

input:

200 -454379
-385892 454379 -1000000 373644 -665078 -1000000 -1000000 454379 0 1000000 373644 -1000000 1000000 -385892 -1000000 373644 0 -665078 0 -665078 -1000000 -665078 -385892 -665078 -385892 454379 -665078 -385892 -1000000 454379 1000000 -385892 373644 454379 -1000000 -385892 -1000000 -385892 -1...

output:

40

result:

ok 1 number(s): "40"

Test #7:

score: 0
Accepted
time: 0ms
memory: 3564kb

input:

200 0
451272 -1000000 677452 677452 0 18908 451272 677452 -233144 677452 451272 18908 -1000000 18908 -1000000 0 451272 0 -233144 677452 1000000 451272 1000000 18908 -1000000 0 -233144 451272 1000000 18908 677452 0 677452 0 677452 1000000 -233144 18908 451272 -1000000 -233144 18908 1000000 0 0 -23314...

output:

35

result:

ok 1 number(s): "35"

Test #8:

score: 0
Accepted
time: 0ms
memory: 3620kb

input:

200 -705945
-586687 198791 0 198791 0 705945 198791 -1000000 705945 705945 1000000 705945 0 -1000000 699023 0 705945 -586687 -1000000 198791 -1000000 1000000 198791 -1000000 198791 705945 -1000000 1000000 1000000 198791 198791 -1000000 699023 0 0 699023 -586687 705945 -586687 705945 699023 0 705945 ...

output:

34

result:

ok 1 number(s): "34"

Test #9:

score: 0
Accepted
time: 0ms
memory: 3656kb

input:

200 0
344286 0 -230711 532652 -589794 344286 532652 -1000000 -589794 344286 0 532652 -1000000 344286 344286 1000000 0 -1000000 532652 532652 532652 -230711 -1000000 344286 532652 532652 0 532652 1000000 0 0 1000000 -589794 -1000000 1000000 -1000000 -1000000 344286 0 344286 344286 1000000 -1000000 -1...

output:

34

result:

ok 1 number(s): "34"

Test #10:

score: 0
Accepted
time: 0ms
memory: 3860kb

input:

200 -230027
-1000000 -662604 0 1000000 59253 1000000 1000000 -662604 -662604 0 -70266 1000000 -300293 -300293 59253 1000000 -1000000 1000000 0 1000000 -662604 59253 -1000000 -70266 -1000000 -662604 -662604 59253 59253 -1000000 -70266 -70266 1000000 -300293 59253 59253 1000000 -70266 -1000000 -70266 ...

output:

33

result:

ok 1 number(s): "33"

Test #11:

score: 0
Accepted
time: 0ms
memory: 3632kb

input:

200 197410
1000000 -665711 0 -271800 -701554 -271800 -1000000 -665711 -665711 -1000000 -271800 1000000 -701554 1000000 -701554 -1000000 1000000 1000000 -1000000 -701554 1000000 -271800 -1000000 -197410 -665711 -701554 1000000 -1000000 -665711 -701554 -701554 -271800 -1000000 -1000000 -271800 -701554...

output:

37

result:

ok 1 number(s): "37"

Test #12:

score: 0
Accepted
time: 0ms
memory: 3644kb

input:

200 -96015
791544 -1000000 791544 -1000000 -668818 695529 1000000 -552146 -1000000 -668818 1000000 1000000 1000000 695529 695529 0 791544 -552146 0 1000000 791544 -668818 0 -1000000 0 -1000000 791544 -1000000 791544 1000000 791544 0 -1000000 0 -668818 695529 791544 1000000 695529 695529 -668818 -100...

output:

38

result:

ok 1 number(s): "38"

Test #13:

score: -100
Wrong Answer
time: 0ms
memory: 3620kb

input:

200 -1000000
1000000 127969 0 0 0 -970611 0 0 293224 293224 293224 293224 -1000000 -970611 1000000 -1000000 820037 820037 820037 127969 127969 0 0 -970611 1000000 -970611 820037 -1000000 -970611 -1000000 820037 -970611 -1000000 127969 -970611 293224 1000000 293224 1000000 293224 293224 -1000000 8200...

output:

37

result:

wrong answer 1st numbers differ - expected: '39', found: '37'