Submission #799143

#	Time	Username	Problem	Language	Result	Execution time	Memory
799143		Blagoj	Watering can (POI13_kon)	C++17	50 / 100	4072 ms	10652 KiB

kon

#include <bits/stdc++.h>

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast,unroll-loops")

using namespace std;
#define ll long long
const int nax = 300'001;
int sq = 500;
ll N, K, t[nax], zers[nax];
 
vector<ll>SQE[nax / 500 + 5];
vector<ll>ps[nax / 500 + 5]; 
 
void inicjuj(int n, int k, int *D)
{
        
        sq = (int)sqrt(n) + 1;
       for (int i = 0; i < n; i++)
       {  
            t[i] = D[i];
            SQE[i / sq].push_back(t[i]);
        }
        N = n, K = k;
 
        for (int i = 0; i <= n / sq; i++){
            ps[i].resize(SQE[i].size());
            sort(SQE[i].begin(), SQE[i].end());
            SQE[i].erase(unique(SQE[i].begin(),SQE[i].end()),SQE[i].end());
            for (int j = i * sq; j < n && j /sq == i; j++){
                int pos = lower_bound(SQE[i].begin(), SQE[i].end(), t[j]) - SQE[i].begin();
                ps[i][pos]++;
            }
            for (int j = ps[i].size() - 2; j >= 0; j--){
                ps[i][j] += ps[i][j + 1];
            }
        }
}
 
void podlej(int a, int b)
{
    //--a,--b;
    for (int i = a / sq; i * sq <=  b; i++){
       // cout << i << endl;
        if (a <= i * sq && b >= (i + 1) * sq - 1){
            zers[i]++;
        }else{
            SQE[i].clear();
            for (int j = i*sq; j / sq == i && j < N; j++){
                t[j]+=zers[i];
                if (j >= a && j <= b)t[j]++;
                SQE[i].push_back(t[j]);
               // t[j]++;
            }
            zers[i] = 0;
            sort(SQE[i].begin(), SQE[i].end());
            SQE[i].erase(unique(SQE[i].begin(),SQE[i].end()),SQE[i].end());
            ps[i].clear();
            ps[i].resize(SQE[i].size());
            for (int j = i*sq; j / sq == i && j < N; j++){
                int pos = lower_bound(SQE[i].begin(), SQE[i].end(), t[j]) - SQE[i].begin();
                ps[i][pos]++;
            } 
            for (int j = ps[i].size() - 2; j >= 0; j--){
                ps[i][j] += ps[i][j + 1];
            }
        }
    }
}
 
int dojrzale(int a, int b)
{
   // --a,--b;
    int ans = 0;
    
   
    for (int i = a / sq; i <= b / sq; i++){
        if (a <= i * sq && b >= (i + 1) * sq - 1){
            int pos = lower_bound(SQE[i].begin(), SQE[i].end(), K - zers[i]) - SQE[i].begin();
            if (pos < SQE[i].size())ans += ps[i][pos];
        }else{
           for (int j = max(a, i * sq); j / sq == i && j <= b; j++){
                if (t[j] + zers[i] >= K)++ans;
           }
        }
    } 
    return ans;
}

Compilation message (stderr)

kon.cpp: In function 'int dojrzale(int, int)':
kon.cpp:81:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   81 |             if (pos < SQE[i].size())ans += ps[i][pos];
      |                 ~~~~^~~~~~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 0 / 10
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 10
• Subtask 10: 0 / 10