제출 #529029

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
529029		Evang	Growing Trees (BOI11_grow)	C++17	100 / 100	118 ms	3768 KiB

grow

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

#ifdef _DEBUG
#define dout(x) clog << "Line " << __LINE__ << ": " << #x << "=" << (x) << el
#else
#define dout(x)
#endif

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
#define uid(a,b) uniform_int_distribution<int>(a,b)(rng)

#define ins insert
#define ssize(x) (int((x).size()))
#define bs(args...) binary_search(args)
#define lb(args...) lower_bound(args)
#define ub(args...) upper_bound(args)
#define all(x) (x).begin(),(x).end()
#define mp(a, b) make_pair(a, b)
#define mt(args...) make_tuple(args)
#define pb(x) push_back(x)
#define eb(args...) emplace_back(args)
#define ff first
#define ss second
#define die exit(0)

template<typename T>
using vc = vector<T>;
template<typename T>
using uset = unordered_set<T>;
template<typename A, typename B>
using umap = unordered_map<A, B>;
template<typename T, typename Comp>
using pq = std::priority_queue<T, vc<T>, Comp>;
template<typename T>
using maxpq = pq<T, less<T>>;
template<typename T>
using minpq = pq<T, greater<T>>;
template<typename T>
using oset = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

using db = double;
using ld = long double;
using ll = long long;
using ull = unsigned long long;
using pi = pair<int, int>;
using pll = pair<ll, ll>;
using vi = vc<int>;
using vll = vc<ll>;
using vpi = vc<pi>;
using vpll = vc<pll>;
using str = string;

constexpr char el = '\n';
constexpr char sp = ' ';
constexpr int inf = 0x3f3f3f3f;
constexpr ll llinf = 0x3f3f3f3f3f3f3f3fLL;
// ---------------------------------------------------------------------


const int N = 1e5+5;
int n, m, a[N];
ll bit[N];

void upd(int i, int x){
    while(i<=n){
        bit[i] += x;
        i += i&-i;
    }
}

ll at(int i){
    ll ans = 0;
    while(i>0){
        ans += bit[i];
        i -= i&-i;
    }
    return ans;
}

signed main() {
    ios::sync_with_stdio(0); cin.tie(0);
    cout << fixed; clog << fixed; clog << unitbuf;
#ifdef _DEBUG
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    freopen("debug.txt", "w", stderr);
#else
    //freopen(".in", "r", stdin);
    //freopen(".out", "w", stdout);
#endif

    cin >> n >> m;
    for(int i = 1; i <= n; ++i)
        cin >> a[i];
    sort(a+1, a+1+n);
    for(int i = 1; i <= n; ++i)
        upd(i, a[i]-a[i-1]);

    while(m--){
        char t;
        int c, h;
        cin >> t >> c >> h;
        if(t=='F'){
            int i = n+1;
            for(int u = n; u > 0; u /= 2)
                while(i - u > 0 && at(i-u)>=h)
                    i -= u;
            int j = min(n, i+c-1);
            int k = j;
            for(int u = k; u > 0; u /= 2)
                while(k-u>0&&at(k-u)==at(j))
                    k -= u;
            int e = j;
            for(int u = n; u > 0; u /= 2)
                while(e+u<=n&&at(e+u)==at(j))
                    e += u;

            upd(i, 1);
            upd(k, -1);
            upd(e-(j-k+1)+1, 1);
            upd(e+1, -1);
        }else{
            int i = n;
            for(int u = i; u > 0; u /= 2)
                while(i-u>0&&at(i-u)>=c)
                    i -= u;
            int j = 1;
            for(int u = n; u > 0; u /= 2)
                while(j+u<=n&&at(j+u)<=h)
                    j += u;
            if(i==j&&at(i)>h||at(i)<c)
                cout << 0 << el;
            else
                cout << j-i+1 << el;
        }
    }
}

컴파일 시 표준 에러 (stderr) 메시지

grow.cpp: In function 'int main()':
grow.cpp:133:20: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
  133 |             if(i==j&&at(i)>h||at(i)<c)
      |                ~~~~^~~~~~~~~

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