This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
//#pragma GCC optimize("O3")
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define f first
#define s second
#define all(x) x.begin(),x.end()
#define _ ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
void setIO(string s) {
freopen((s + ".in").c_str(), "r", stdin);
freopen((s + ".out").c_str(), "w", stdout);
}
int main() {_
int n,l;
cin>>n>>l;
vector<int>a(n);
for(int i=0;i<n;i++){
cin>>a[i];
}
vector<pair<pair<int,int>,int>> vec;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(a[i]==a[j]) continue;
int L=max(0,i-(l-max({0,l-(i+1),l-(j+1)}))+1);
int R=i-max({0,l-(n-i),l-(n-j)});
if(R<L) continue;
//cout<<L<<' '<<R<<'\n';
vec.push_back({{L,R},j-i});
}
}
return 0;
vector<vector<int>> st(n),en(n);
for(auto v:vec){
st[v.f.f].push_back(v.s);
if(v.f.s<n-1) en[v.f.s+1].push_back(v.s);
}
int q;
cin>>q;
vector<pair<int,int>> qs;
for(int i=0;i<q;i++){
int x;
cin>>x;
qs.push_back({x,i});
}
sort(all(qs));
sort(all(vec));
vector<int> cnt(3*n);
vector<int> id(l+1);
for(int i=0;i<=l;i++){
int pos=lower_bound(all(qs),make_pair(i,-1))-qs.begin();
id[i]=((pos==q)?-1:pos);
}
return 0;
vector<vector<int>> ans(n,vector<int>(q));
for(int i=0;i<n-l+1;i++){
for(auto v:st[i]){
cnt[v+n]++;
}
for(auto v:en[i]){
cnt[v+n]--;
}
for(int j=-i;j<n-i-l+1;j++){
if(j==0) continue;
//cout<<cnt[n+j]<<' ';
assert(cnt[n+j]>=0 and cnt[n+j]<=l);
if(id[cnt[n+j]]!=-1){
ans[i][id[cnt[n+j]]]++;
}
}
for(int j=1;j<q;j++){
ans[i][j]+=ans[i][j-1];
}
//cout<<'\n';
}
return 0;
vector<int> tmp(q);
for(int i=0;i<q;i++){
tmp[qs[i].s]=i;
}
for(int j=0;j<q;j++){
for(int i=0;i<n-l+1;i++){
cout<<ans[i][tmp[j]]<<' ';
}
cout<<'\n';
}
return 0;
}
//maybe its multiset not set
//yeeorz
//laborz
Compilation message (stderr)
lot.cpp: In function 'void setIO(std::string)':
lot.cpp:15:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
15 | freopen((s + ".in").c_str(), "r", stdin);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
lot.cpp:16:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
16 | freopen((s + ".out").c_str(), "w", stdout);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |