#210609 (C++17) No.284 門松と魔法(2)

提出ソース

結果

問題	No.284 門松と魔法(2)
ユーザー	mai
提出日時	2017-10-19 21:51:36
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	RE
実行時間	-
コード長	9,606 bytes
コンパイル時間	3,224 ms
コンパイル使用メモリ	217,828 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-11-21 07:42:45
合計ジャッジ時間	8,628 ms
ジャッジサーバーID （参考情報）	judge2 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	RE	-
testcase_01	RE	-
testcase_02	RE	-
testcase_03	RE	-
testcase_04	RE	-
testcase_05	RE	-
testcase_06	RE	-
testcase_07	RE	-
testcase_08	RE	-
testcase_09	RE	-
testcase_10	RE	-
testcase_11	RE	-
testcase_12	RE	-
testcase_13	RE	-
testcase_14	RE	-
testcase_15	RE	-
testcase_16	RE	-
testcase_17	RE	-
testcase_18	RE	-
testcase_19	RE	-
testcase_20	RE	-
testcase_21	RE	-
testcase_22	RE	-
testcase_23	RE	-
testcase_24	RE	-
testcase_25	RE	-
testcase_26	RE	-
testcase_27	RE	-
testcase_28	RE	-
testcase_29	RE	-
testcase_30	RE	-
testcase_31	RE	-
testcase_32	RE	-
testcase_33	RE	-
testcase_34	RE	-
testcase_35	RE	-
testcase_36	RE	-
testcase_37	RE	-
testcase_38	RE	-
testcase_39	RE	-

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

ソースコード

raw source code

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#include "bits/stdc++.h" // define macro "/D__MAI"

using namespace std;
typedef long long int ll;

#define xprintf(fmt,...) fprintf(stderr,fmt,__VA_ARGS__)
#define debugv(v) {printf("L%d %s > ",__LINE__,#v);for(auto e:v){cout<<e<<" ";}cout<<endl;}
#define debuga(m,w) {printf("L%d %s > ",__LINE__,#m);for(int x=0;x<(w);x++){cout<<(m)[x]<<" ";}cout<<endl;}
#define debugaa(m,h,w) {printf("L%d %s >\n",__LINE__,#m);for(int y=0;y<(h);y++){for(int x=0;x<(w);x++){cout<<(m)[y][x]<<" ";}cout<<endl;}}
#define ALL(v) (v).begin(),(v).end()
#define repeat(cnt,l) for(auto cnt=0ll;(cnt)<(l);++(cnt))
#define iterate(cnt,b,e) for(auto cnt=(b);(cnt)!=(e);++(cnt))
#define MD 1000000007ll
#define PI 3.1415926535897932384626433832795
#define EPS 1e-12
template<typename T1, typename T2> ostream& operator <<(ostream &o, const pair<T1, T2> p) { o << "(" << p.first << ":" << p.second << ")"; return o; }
template<typename iterator> inline size_t argmin(iterator begin, iterator end) { return distance(begin, min_element(begin, end)); }
template<typename iterator> inline size_t argmax(iterator begin, iterator end) { return distance(begin, max_element(begin, end)); }
template<typename T> T& maxset(T& to, const T& val) { return to = max(to, val); }
template<typename T> T& minset(T& to, const T& val) { return to = min(to, val); }

mt19937_64 randdev(8901016);
inline ll rand_range(ll l, ll h) {
    return uniform_int_distribution<ll>(l, h)(randdev);
}

#ifdef __MAI
#define getchar_unlocked getchar
#define putchar_unlocked putchar
#endif
#ifdef __VSCC
#define getchar_unlocked _getchar_nolock
#define putchar_unlocked _putchar_nolock
#endif
namespace {
#define isvisiblechar(c) (0x21<=(c)&&(c)<=0x7E)
    class MaiScanner {
    public:
        template<typename T> void input_integer(T& var) {
            var = 0;
            T sign = 1;
            int cc = getchar_unlocked();
            for (; cc<'0' || '9'<cc; cc = getchar_unlocked())
                if (cc == '-') sign = -1;
            for (; '0' <= cc&&cc <= '9'; cc = getchar_unlocked())
                var = (var << 3) + (var << 1) + cc - '0';
            var = var*sign;
        }
        inline int c() { return getchar_unlocked(); }
        inline MaiScanner& operator>>(int& var) {
            input_integer<int>(var);
            return *this;
        }
        inline MaiScanner& operator>>(long long& var) {
            input_integer<long long>(var);
            return *this;
        }
        inline MaiScanner& operator>>(string& var) {
            int cc = getchar_unlocked();
            for (; !isvisiblechar(cc); cc = getchar_unlocked());
            for (; isvisiblechar(cc); cc = getchar_unlocked())
                var.push_back(cc);
            return *this;
        }
        template<typename IT> void in(IT begin, IT end) {
            for (auto it = begin; it != end; ++it) *this >> *it;
        }
    };
}
MaiScanner scanner;


template<typename T>
//typedef int T;
class segtree {
    int size;
    vector<T> valdata;
    vector<int> addtree;
    vector<int> maxtree;

    int _geti(const vector<int>& v, int idx) const {
        if (size - 1 <= idx) return (idx - size + 1);
        // if (idx < 0) return idx;
        return v[idx];
    }
public:

    segtree(int n) {
        size = 8;
        while (size < n) size <<= 1;
        maxtree.resize(size);
        valdata.resize(size);
        addtree.resize(size);
        flat();
    }

    // tree初期化
    void flat() {
        for (int i = size - 2; 0 <= i; --i) {
            int c = 2 * i + 1;
            if (size - 1 <= c) {
                maxtree[i] = c - size + 1;
            }
            else {
                maxtree[i] = maxtree[c];
            }
        }
    }

    // TODO:verify
    // addtreeを0にして，valdataに加算する．
    void apply(T sum = 0, int ptr = 0) {
        if (size - 1 <= ptr) {
            valdata[ptr - size + 1] += sum;
        }
        else {
            sum += addtree[ptr];
            apply(sum, ptr * 2 + 1);
            apply(sum, ptr * 2 + 2);
            addtree[ptr] = 0;
        }
    }

    void fill(T val) {
        std::fill(valdata.begin(), valdata.end(), val);
        std::fill(addtree.begin(), addtree.end(), 0);
        flat();
    }

    // T operator[](int index) const {
    //     
    // }

    T getval(int index) const {
        T sum = valdata[index];
        index += size - 1;
        while (0 < index) {
            index = (index - 1) / 2; // 親へ移動
            sum += addtree[index];
        }
        return sum;
    }

    void setval(int index, const T e) {
        valdata[index] = e;

        for (int i = index + size - 1; 0 < i;) {
            i = (i - 1) / 2; // 親へ移動
            valdata[index] -= addtree[i];
        }
        index += size - 1;
        while (0 < index) {
            index = (index - 1) / 2; // 親へ移動
            int c1 = index * 2 + 1;
            int c2 = index * 2 + 2;
            int i1, i2;
            i1 = _geti(maxtree, c1);
            i2 = _geti(maxtree, c2);
            maxtree[index] = getval(i1) > getval(i2) ? i1 : i2;
        }
    }

    void addvalrange(int begin, int end, const T e, int ptr = 0, int rangebegin = 0, int rangeend = 0) {
        if (rangeend <= rangebegin) { rangeend = size; }

        if (rangeend <= begin || end <= rangebegin) return;
        if (begin <= rangebegin && rangeend <= end) {
            if (size - 1 <= ptr)
                valdata[ptr - size + 1] += e;
            else
                addtree[ptr] += e;
            return;
        }

        int rangemid = (rangebegin + rangeend) / 2;
        addvalrange(begin, end, e, ptr * 2 + 1, rangebegin, rangemid);
        addvalrange(begin, end, e, ptr * 2 + 2, rangemid, rangeend);

        int i1, i2;
        i1 = _geti(maxtree, ptr * 2 + 1);
        i2 = _geti(maxtree, ptr * 2 + 2);
        maxtree[ptr] = getval(i1) > getval(i2) ? i1 : i2;

    }

    int getmaxrangeIdx(int begin, int end, int ptr = 0, int rangebegin = 0, int rangeend = 0) const {
        if (rangeend <= rangebegin) { rangeend = size; }

        if (rangeend <= begin || end <= rangebegin) return begin; // note:範囲外なら適当な範囲内のindexを返す
        if (begin <= rangebegin && rangeend <= end) return _geti(maxtree, ptr);

        int rangemid = (rangebegin + rangeend) / 2;
        int l = getmaxrangeIdx(begin, end, ptr * 2 + 1, rangebegin, rangemid);
        int r = getmaxrangeIdx(begin, end, ptr * 2 + 2, rangemid, rangeend);

        if (getval(l) >= getval(r))
            return l;
        else
            return r;
    }

    T getmaxrange(int begin, int end) {
        return getval(getmaxrangeIdx(begin, end));
    }
};



ll m, n, kei;

// 順列が与えられるケースを考えると，下のLKSSの通りになる．☆3.5相当．
// 次のような重複を考慮する必要がある．
// [1,3,1,3,1] => 5 => [WA]

// 2つ前の値を保持しなければならない．
// セグ木の葉に複数のデータを載せる？

// pair<count, prevprev>

int main() {
    abort();

    scanner >> n;
    vector<int> aa;
    aa.reserve(n);

    map<int, int> zip;

    repeat(i, n) {
        int a;
        scanner >> a;
        aa.push_back(a);
        zip[a] = 0;
    }

    // 圧縮．
    {
        int c = 0;
        for (auto& p : zip)
            p.second = c++;
    }
    for (auto& a : aa)
        a = zip[a];

    segtree<int> upkado(n); // 門↑松↓列列
    segtree<int> dnkado(n); // 門↓松↑列列

    repeat(i, n) {
        int a = aa[i];
        int u = dnkado.getmaxrange(0, a) + 1;
        int d = upkado.getmaxrange(a + 1, n) + 1;
        if (upkado.getval(a) < u)
            upkado.setval(a, u);
        if (dnkado.getval(a) < d)
            dnkado.setval(a, d);
        // repeat(j, n)
        //     printf("%2d ", upkado.getval(j));
        // cout << endl;
        // repeat(j, n)
        //     printf("%2d ", dnkado.getval(j));
        // cout << endl;
    }

    int ans = max(upkado.getval(aa[n - 1]), dnkado.getval(aa[n - 1]));

    if (ans <= 2)
        cout << 0 << endl;
    else
        cout << ans << endl;


    return 0;
}

/*

LKSSという問題を考えたら丸かぶりだった．

- - - - - - -

LKSS

順列 x から好きな整数を好きなだけ取り除いて門松列列を作るとき，その数列の長さの最大値を求めてください．

ただし，門松列列が作れないときは0を出力してください．

- - - - - - -


数列 x について，新たな門松列列の定義を導入する．

門↑松↓列列
|x|=1 : x は門↑松↓列列
|x|=2 : x1 < x2 ならば 門↑松↓列列
|x|>=3 : x1 < x2 かつ 門松列列 ならば 門↑松↓列列

門↓松↑列列
|x|=1 : x は門↓松↑列列
|x|=2 : x1 > x2 ならば 門↓松↑列列
|x|>=3 : x1 > x2 かつ 門松列列 ならば 門↓松↑列列

今から考えたいのは，2つのdp配列

dp1[x] 要素xまでみたとき，最長の門↑松↓列列
dp2[x] 要素xまでみたとき，最長の門↓松↑列列

dp配列を更新するには，
dp1[x] = max(dp2[1..x-1]) + 1
dp2[x] = max(dp1[x+1..N]) + 1
とすれば良い．

dpの定義より，求めたい最長門松列列の長さは，max(dp1[x_N], dp[x_N])．
ただし，長さ2以下の門松列列は存在しないので，その場合は0と出力する．

配列のある区間の最大を求めるために，セグメントツリーを用いることで，計算量をO(logN)に抑えることができる．

- - - - - - -


*/

yukicoder

結果

テストケース

ソースコード