Submission #598545

#	Time	Username	Problem	Language	Result	Execution time	Memory
598545		jiahng	Lottery (CEOI18_lot)	C++14	45 / 100	92 ms	65536 KiB

lot

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

typedef long long ll;
#define ll int
typedef pair<int,int> pi;
typedef vector <int> vi;
typedef vector <pi> vpi;
typedef pair<pi, ll> pii;
typedef set <ll> si;
typedef long double ld;
#define f first
#define s second
#define mp make_pair
#define FOR(i,s,e) for(int i=s;i<=int(e);++i)
#define DEC(i,s,e) for(int i=s;i>=int(e);--i)
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define lbd(x, y) lower_bound(all(x), y)
#define ubd(x, y) upper_bound(all(x), y)
#define aFOR(i,x) for (auto i: x)
#define mem(x,i) memset(x,i,sizeof x)
#define fast ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define maxn 10001
#define INF (ll)(1e9+10)
#define MOD 1000000007
typedef pair <vi, int> pvi;
typedef pair <int,pi> ipi;
typedef vector <pii> vpii;
typedef pair <pi,pi> pipi;
int N,L,A[maxn];
vi occ[maxn];
vpi num[maxn];
int Q,k;
int32_t main(){
    fast;
    cin >> N >> L;
    FOR(i,1,N) cin >> A[i];
    
    FOR(i,2,N+1-L){
       int cur = 0;
       FOR(j,1,L) cur += (A[j] != A[i+j-1]);
       FOR(j,0,N+1-L-i){
            occ[j+1].pb(cur); occ[i+j].pb(cur);
            //cout << i+j << ' ' << j+1  << ' ' << cur << '\n';
            cur -= A[j+1] != A[j+i];
            cur += A[j+1+L] != A[j+i+L];
        }
    }
    
    FOR(i,1,N){
        sort(all(occ[i]));
        int co = 0;
        num[i].pb(pi(-1,0));
        while (co < occ[i].size()){
            int cur = occ[i][co];
            while (co < occ[i].size() && occ[i][co] == cur) co++;
            num[i].pb(pi(cur, co));
        }
    }
    cin >> Q;
    FOR(i,1,Q){
        cin >> k;
        FOR(j,1,N+1-L){
            pi x = *(--upper_bound(all(num[j]), pi(k, 1e9)));
            cout << x.s << ' ';
        }
        cout << '\n';
    }
}

Compilation message (stderr)

lot.cpp: In function 'int32_t main()':
lot.cpp:55:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |         while (co < occ[i].size()){
      |                ~~~^~~~~~~~~~~~~~~
lot.cpp:57:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |             while (co < occ[i].size() && occ[i][co] == cur) co++;
      |                    ~~~^~~~~~~~~~~~~~~

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