제출 #1171634

#제출 시각아이디문제언어결과실행 시간메모리
1171634ByeWorldDiversity (CEOI21_diversity)C++20
컴파일 에러
0 ms0 KiB
#include <bits/stdc++.h>
// #pragma GCC optimize("O3", "Ofast")
// #define int long long
#define ll long long
#define pb push_back
#define fi first
#define se second
#define lf (id<<1)
#define rg ((id<<1)|1)
#define md ((l+r)>>1)
#define ld long double
using namespace std;
typedef pair<int,int> pii;
typedef pair<char,char> pcc;
typedef pair<pii,int> ipii;
typedef pair<pii,pii> ipiii;
const int MAXN = 3e5+10;

int n, q, a[MAXN], cnt[MAXN];
vector <ipii> que;

int l=1,r=0, dif[MAXN];
int nxl[MAXN], nxr[MAXN];
ll ans[MAXN];

signed main(){
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n>>q;
    for(int i=1; i<=n; i++)
        cin>>a[i]; 
    
    for(int p=1; p<=q; p++){
        int l,r;cin>>l>>r;
        que.pb({{l,r}, p});
    }
    sort(que.begin(), que.end(), [&](auto a, auto b){
        if(a.fi.fi/500 != b.fi.fi/500) a.fi.fi/500 < b.fi.fi/500;
        if((a.fi.fi/500) & 1) a.fi.se > b.fi.se;
        return a.fi.se < b.fi.se; 
    });

    nxl[0] = 0, nxr[0] = n+1, nxl[n+1] = 0, nxr[n+1] = n+1;

    for(auto [xx, idx] : que){
        int le = xx.fi, ri = xx.se;
        while (r + 1 <= qr[i].fi.se) {
            r++;

            int x = a[r];
            if (cnt[x] != 0) {
                    cnt2[cnt[x]] -= 1;
                    if (cnt2[cnt[x]] == 0) {
                            if (cnt2[cnt[x] + 1] == 0) {
                                    nxr[cnt[x] + 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] + 1;
                                    nxl[cnt[x] + 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] + 1;
                            } else {
                                    nxl[cnt[x] + 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] + 1;
                            }

                            nxr[cnt[x]] = 0;
                            nxl[cnt[x]] = 0;
                    } else {
                            if (cnt2[cnt[x] + 1] == 0) {
                                    nxr[cnt[x] + 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] + 1;
                                    nxl[cnt[x] + 1] = cnt[x];
                                    nxr[cnt[x]] = cnt[x] + 1;
                            }
                    }
            } else {
                    if (cnt2[cnt[x] + 1] == 0) {
                            nxr[cnt[x] + 1] = nxr[cnt[x]];
                            nxl[nxr[cnt[x]]] = cnt[x] + 1;
                            nxl[cnt[x] + 1] = cnt[x];
                            nxr[cnt[x]] = cnt[x] + 1;
                    }
            }
            
            cnt[x] += 1;
            cnt2[cnt[x]] += 1;
    }

    while (l - 1 >= qr[i].fi.fi) {
            l--;

            int x = a[l];
            if (cnt[x] != 0) {
                    cnt2[cnt[x]] -= 1;
                    if (cnt2[cnt[x]] == 0) {
                            if (cnt2[cnt[x] + 1] == 0) {
                                    nxr[cnt[x] + 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] + 1;
                                    nxl[cnt[x] + 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] + 1;
                            } else {
                                    nxl[cnt[x] + 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] + 1;
                            }

                            nxr[cnt[x]] = 0;
                            nxl[cnt[x]] = 0;
                    } else {
                            if (cnt2[cnt[x] + 1] == 0) {
                                    nxr[cnt[x] + 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] + 1;
                                    nxl[cnt[x] + 1] = cnt[x];
                                    nxr[cnt[x]] = cnt[x] + 1;
                            }
                    }
            } else {
                    if (cnt2[cnt[x] + 1] == 0) {
                            nxr[cnt[x] + 1] = nxr[cnt[x]];
                            nxl[nxr[cnt[x]]] = cnt[x] + 1;
                            nxl[cnt[x] + 1] = cnt[x];
                            nxr[cnt[x]] = cnt[x] + 1;
                    }
            }
            
            cnt[x] += 1;
            cnt2[cnt[x]] += 1;
    }

    while (r-1 >= qr[i].fi.se) {
            int x = a[r];
            if (cnt[x] != 1) {
                    cnt2[cnt[x]] -= 1;
                    if (cnt2[cnt[x]] == 0) {
                            if (cnt2[cnt[x] - 1] == 0) {
                                    nxr[cnt[x] - 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] - 1;
                                    nxl[cnt[x] - 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] - 1;
                            } else {
                                    nxr[cnt[x] - 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] - 1;
                            }

                            nxr[cnt[x]] = 0;
                            nxl[cnt[x]] = 0;
                    } else {
                            if (cnt2[cnt[x] - 1] == 0) {
                                    nxr[cnt[x] - 1] = cnt[x];
                                    nxl[cnt[x] - 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] - 1;
                                    nxl[cnt[x]] = cnt[x] - 1;
                            }
                    }
            } else {
                    cnt2[cnt[x]] -= 1;
                    if (cnt2[cnt[x]] == 0) {
                            nxr[cnt[x] - 1] = nxr[cnt[x]];
                            nxl[nxr[cnt[x]]] = cnt[x] - 1;

                            nxr[cnt[x]] = 0;
                            nxl[cnt[x]] = 0;
                    }
            }

            cnt[x] -= 1;
            cnt2[cnt[x]] += 1;

            r--;
    }

    while (l + 1 <= qr[i].fi.fi) {
            int x = a[l];  
            if (cnt[x] != 1) {
                    cnt2[cnt[x]] -= 1;
                    if (cnt2[cnt[x]] == 0) {
                            if (cnt2[cnt[x] - 1] == 0) {
                                    nxr[cnt[x] - 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] - 1;
                                    nxl[cnt[x] - 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] - 1;
                            } else {
                                    nxr[cnt[x] - 1] = nxr[cnt[x]];
                                    nxl[nxr[cnt[x]]] = cnt[x] - 1;
                            }

                            nxr[cnt[x]] = 0;
                            nxl[cnt[x]] = 0;
                    } else {
                            if (cnt2[cnt[x] - 1] == 0) {
                                    nxr[cnt[x] - 1] = cnt[x];
                                    nxl[cnt[x] - 1] = nxl[cnt[x]];
                                    nxr[nxl[cnt[x]]] = cnt[x] - 1;
                                    nxl[cnt[x]] = cnt[x] - 1;
                            }
                    }
            } else {
                    cnt2[cnt[x]] -= 1;
                    if (cnt2[cnt[x]] == 0) {
                            nxr[cnt[x] - 1] = nxr[cnt[x]];
                            nxl[nxr[cnt[x]]] = cnt[x] - 1;

                            nxr[cnt[x]] = 0;
                            nxl[cnt[x]] = 0;
                    }
            }

            cnt[x] -= 1;
            cnt2[cnt[x]] += 1;

            l++;
    }

        // cout << "donemove\n";

        vector <int> vec; 
        int in = nxr[0];
        while(in!=n+1){
            // cout << in << ' ' << dif[in] << " dif\n";
            vec.pb(in); // value, len
            in = nxr[in];
        }
        reverse(vec.begin(), vec.end());
        assert(vec.size() <= 600);

        ll res = 0, f = 0, num = 0, cur = 0, pref = 0, suf = 0;
        for(auto x : vec) {
            int c = dif[x];

            ll le = (c + (f ^ 1))>>1, ri = (c + f)>>1;

            auto calc = [&](ll y) {
                ll res = (((x * y) * (x * y + 1)) >>1) + 
                    x * x * ((y - 1) * y * (2 * (y - 1) + 1) / 6 + 
                        (((y - 1) * y / 2))>>1);
                return res;
            };

            // le to le and ri to ri
            res += calc(le);
            res += calc(ri);
            // cout << "a: " << res << '\n';

            // le to ri
            res += cur * le * x * ri * x + ((le * (le + 1)>>1) * x) * 
                (ri * x) + (le * x) * (((ri * (ri + 1))>>1) * x);
            // cout << "b: " << res << '\n';

            // le to md
            res += (le * x) * pref;
            res += (((le * (le + 1))>>1) * x) * num;
            // cout << "c: " << res << '\n';

            // md to ri
            res += suf * (ri * x);
            res += num * (((ri * (ri + 1))>>1) * x);
            // cout << "d: " << res << '\n';

            pref = (((le * (le + 1))>>1) * x) + (pref + num * le) + 
                (((ri * (ri + 1))>>1) * x + (ri * x) * (le + cur));
            suf = (((ri * (ri + 1))>>1) * x) + (suf + num * ri) + 
                (((le * (le + 1))>>1) * x + (le * x) * (ri + cur));
            
            num += c * x;
            cur += c;
            f ^= (c & 1);
        }
        ans[idx] = res;
    }
    for(int i=1; i<=q; i++) cout << ans[i] << '\n';
}

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

diversity.cpp: In function 'int main()':
diversity.cpp:46:25: error: 'qr' was not declared in this scope; did you mean 'r'?
   46 |         while (r + 1 <= qr[i].fi.se) {
      |                         ^~
      |                         r
diversity.cpp:46:28: error: 'i' was not declared in this scope; did you mean 'ri'?
   46 |         while (r + 1 <= qr[i].fi.se) {
      |                            ^
      |                            ri
diversity.cpp:51:21: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
   51 |                     cnt2[cnt[x]] -= 1;
      |                     ^~~~
      |                     cnt
diversity.cpp:74:25: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
   74 |                     if (cnt2[cnt[x] + 1] == 0) {
      |                         ^~~~
      |                         cnt
diversity.cpp:83:13: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
   83 |             cnt2[cnt[x]] += 1;
      |             ^~~~
      |             cnt
diversity.cpp:86:21: error: 'qr' was not declared in this scope; did you mean 'r'?
   86 |     while (l - 1 >= qr[i].fi.fi) {
      |                     ^~
      |                     r
diversity.cpp:86:24: error: 'i' was not declared in this scope; did you mean 'ri'?
   86 |     while (l - 1 >= qr[i].fi.fi) {
      |                        ^
      |                        ri
diversity.cpp:91:21: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
   91 |                     cnt2[cnt[x]] -= 1;
      |                     ^~~~
      |                     cnt
diversity.cpp:114:25: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  114 |                     if (cnt2[cnt[x] + 1] == 0) {
      |                         ^~~~
      |                         cnt
diversity.cpp:123:13: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  123 |             cnt2[cnt[x]] += 1;
      |             ^~~~
      |             cnt
diversity.cpp:126:19: error: 'qr' was not declared in this scope; did you mean 'r'?
  126 |     while (r-1 >= qr[i].fi.se) {
      |                   ^~
      |                   r
diversity.cpp:126:22: error: 'i' was not declared in this scope; did you mean 'ri'?
  126 |     while (r-1 >= qr[i].fi.se) {
      |                      ^
      |                      ri
diversity.cpp:129:21: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  129 |                     cnt2[cnt[x]] -= 1;
      |                     ^~~~
      |                     cnt
diversity.cpp:152:21: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  152 |                     cnt2[cnt[x]] -= 1;
      |                     ^~~~
      |                     cnt
diversity.cpp:163:13: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  163 |             cnt2[cnt[x]] += 1;
      |             ^~~~
      |             cnt
diversity.cpp:168:21: error: 'qr' was not declared in this scope; did you mean 'r'?
  168 |     while (l + 1 <= qr[i].fi.fi) {
      |                     ^~
      |                     r
diversity.cpp:168:24: error: 'i' was not declared in this scope; did you mean 'ri'?
  168 |     while (l + 1 <= qr[i].fi.fi) {
      |                        ^
      |                        ri
diversity.cpp:171:21: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  171 |                     cnt2[cnt[x]] -= 1;
      |                     ^~~~
      |                     cnt
diversity.cpp:194:21: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  194 |                     cnt2[cnt[x]] -= 1;
      |                     ^~~~
      |                     cnt
diversity.cpp:205:13: error: 'cnt2' was not declared in this scope; did you mean 'cnt'?
  205 |             cnt2[cnt[x]] += 1;
      |             ^~~~
      |             cnt