#791542 (C++17) No.33 アメーバがたくさん

提出ソース

結果

問題	No.33 アメーバがたくさん
ユーザー	raven7959
提出日時	2022-08-22 13:36:27
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2 ms / 5,000 ms
コード長	6,998 bytes
コンパイル時間	2,773 ms
コンパイル使用メモリ	213,664 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-04-18 13:23:56
合計ジャッジ時間	3,568 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	2 ms 6,940 KB
testcase_10	AC	2 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,fma,abm,mmx,avx,avx2")
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rrep(i, n) for (int i = (int)(n - 1); i >= 0; i--)
#define all(x) (x).begin(), (x).end()
#define sz(x) int(x.size())
#define yn(joken) cout<<((joken) ? "Yes" : "No")<<"\n"
#define YN(joken) cout<<((joken) ? "YES" : "NO")<<"\n"
using namespace std;
using ll = long long;
using vi = vector<int>;
using vl = vector<ll>;
using vs = vector<string>;
using vc = vector<char>;
using vd = vector<double>;
using vld = vector<long double>;
using vvi = vector<vector<int>>;
using vvl = vector<vector<ll>>;
using vvs = vector<vector<string>>;
using vvc = vector<vector<char>>;
using vvd = vector<vector<double>>;
using vvld = vector<vector<long double>>;
using vvvi = vector<vector<vector<int>>>;
using vvvl = vector<vector<vector<ll>>>;
using vvvvi = vector<vector<vector<vector<int>>>>;
using vvvvl = vector<vector<vector<vector<ll>>>>;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
const int INF = 1e9;
const ll LINF = 2e18;
template <class T>
bool chmax(T& a, const T& b) {
    if (a < b) {
        a = b;
        return 1;
    }
    return 0;
}
template <class T>
bool chmin(T& a, const T& b) {
    if (b < a) {
        a = b;
        return 1;
    }
    return 0;
}
bool ispow2(int i) { return i && (i & -i) == i; }
bool ispow2(ll i) { return i && (i & -i) == i; }
template <class T>
vector<T> make_vec(size_t a) {
    return vector<T>(a);
}
template <class T, class... Ts>
auto make_vec(size_t a, Ts... ts) {
    return vector<decltype(make_vec<T>(ts...))>(a, make_vec<T>(ts...));
}
template <typename T>
istream& operator>>(istream& is, vector<T>& v) {
    for (int i = 0; i < int(v.size()); i++) {
        is >> v[i];
    }
    return is;
}
template <typename T>
ostream& operator<<(ostream& os, const vector<T>& v) {
    for (int i = 0; i < int(v.size()); i++) {
        os << v[i];
        if (i < int(v.size()) - 1) os << ' ';
    }
    return os;
}

static uint32_t RandXor(){
    static uint32_t x=123456789;
    static uint32_t y=362436069;
    static uint32_t z=521288629;
    static uint32_t w=88675123;
    uint32_t t;
 
    t=x^(x<<11);
    x=y; y=z; z=w;
    return w=(w^(w>>19))^(t^(t>>8));
}

static double Rand01(){
    return (RandXor()+0.5)*(1.0/UINT_MAX);
}

// 区間は閉区間で管理
// bool covered(T x): 点xをカバーされているかどうかを返す
// bool covered(T l,T r): 閉区間[l,r]がカバーされているかどうかを返す
// pair<T,T> covered_by(T x): 点xをカバーしている区間を返す,存在しない場合は[-TINF,-TINF]
// pair<T,T> covered_by(T l,T r): 上のやつの閉区間[l,r]バージョン
// T insert(T x): 点xを追加する,区間に含まれる点の数の増加量を返す
// T insert(T l,T r): 閉区間[l,r]を追加する,区間に含まれる点の数の増加量を返す
// T erase(T x): 点xを削除する,区間に含まれる点の数の減少量を返す
// T erase(T l,T r): 閉区間[l,r]を削除する,区間に含まれる点の数の減少量を返す
// int size(): 区間の数を返す
// T mex(T x=0): x以上の範囲でのmexを返す
// T len(): 区間の長さの合計を返す
// T points(): 区間に含まれる点の数の合計を返す

template<typename T>
struct RangeSet{
    set<pair<T,T>> st;
    T TINF,num;
 
    RangeSet(): num(0) {
        TINF=numeric_limits<T>::max()/2;
        st.emplace(TINF,TINF);
        st.emplace(-TINF,-TINF);
    }
    
    bool covered(T l,T r){
        assert(l<=r);
        auto itr=prev(st.lower_bound({l+1,l+1}));
        return itr->first<=l and r<=itr->second;
    }
    bool covered(T x){
        return covered(x,x);
    }

    pair<T,T> covered_by(T l,T r){
        assert(l<=r);
        auto itr=prev(st.lower_bound({l+1,l+1}));
        if(itr->first<=l and r<=itr->second) return *itr;
        return make_pair(-TINF,-TINF);
    }
    pair<T,T> covered_by(T x){
        return covered_by(x,x);
    }

    T insert(T l,T r){
        assert(l<=r);
        auto itr=prev(st.lower_bound({l+1,l+1}));
        if(itr->first<=l and r<=itr->second) return T(0);
        T sum_erased=T(0);
        // itr->first<=l and l<=itr->second+1
        if(l<=itr->second+1){
            l=itr->first;
            sum_erased+=itr->second-itr->first+1;
            itr=st.erase(itr);
        }
        else itr=next(itr);
        while(r>itr->second){
            sum_erased+=itr->second-itr->first+1;
            itr=st.erase(itr);
        }
        if(itr->first-1<=r and r<=itr->second){
            sum_erased+=itr->second-itr->first+1;
            r=itr->second;
            st.erase(itr);
        }
        st.emplace(l,r);
        num+=r-l+1-sum_erased;
        return r-l+1-sum_erased;
    }
    T insert(T x){
        return insert(x,x);
    }

    T erase(T l,T r){
        assert(l<=r);
        auto itr=prev(st.lower_bound({l+1,l+1}));
        // itr->first<=l and r<=itr->second
        if(r<=itr->second){
            // 完全に1つの区間に包含されている
            if(itr->first<l)  st.emplace(itr->first,l-1);
            if(r<itr->second) st.emplace(r+1,itr->second);
            st.erase(itr);
            num-=r-l+1;
            return r-l+1;
        }
 
        T ret=T(0);
        if(itr->first<=l and l<=itr->second){
            ret+=itr->second-l+1;// 消えた
            if(itr->first<l) st.emplace(itr->first,l-1);
            itr=st.erase(itr);// 次へ
        }
        else itr=next(itr);
        while(itr->second<=r){
            ret+=itr->second-itr->first+1;
            itr=st.erase(itr);
        }
        // 右端が区間の間にあるか
        // itr->first<=r and r<=itr->second
        if(itr->first<=r){
            ret+=r-itr->first+1;
            if(r<itr->second) st.emplace(r+1,itr->second);
            st.erase(itr);
        }
        num-=ret;
        return ret;
    }
    T erase(T x){
        return erase(x,x);
    }

    int size(){
        return (int)st.size()-2;
    }

    T mex(T x=0){
        auto itr=prev(st.lower_bound({x+1,x+1}));
        if(itr->first<=x and x<=itr->second) return itr->second+1;
        else return x;
    }

    T len(){
        return num-size();
    }

    T points(){
        return num;
    }

    // for debug
    void output(){
        cout<<"RangeSet : ";
        for(auto &p:st){
            if(p.first==-TINF or p.second==TINF) continue;
            cout<<"["<<p.first<<", "<<p.second<<"] ";
        }
        cout<<"\n";
    }
};

void solve(){
    ll N,D,T;
    cin>>N>>D>>T;
    vl X(N);
    cin>>X;
    map<ll,RangeSet<ll>> mp;
    rep(i,N){
        ll R=X[i]%D;
        while(R<0) R+=D;
        mp[R].insert((X[i]-R)/D-T,(X[i]-R)/D+T);
    }
    ll ans=0;
    for(auto [k,v]:mp){
        ans+=v.points();
    }
    cout<<ans<<endl;
}

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    solve();
}

yukicoder

結果

テストケース

ソースコード