#307394 (C++14) No.382 シャイな人たち (2)

提出ソース

結果

問題	No.382 シャイな人たち (2)
ユーザー	kopricky
提出日時	2018-12-31 07:33:27
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	1,870 ms / 8,000 ms
コード長	11,551 bytes
コンパイル時間	2,261 ms
コンパイル使用メモリ	183,072 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-10-11 06:11:52
合計ジャッジ時間	36,406 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1,870 ms 6,816 KB
testcase_01	AC	1,800 ms 6,816 KB
testcase_02	AC	1,563 ms 6,820 KB
testcase_03	AC	1,617 ms 6,820 KB
testcase_04	AC	1,557 ms 6,820 KB
testcase_05	AC	1,656 ms 6,816 KB
testcase_06	AC	1,399 ms 6,816 KB
testcase_07	AC	1,665 ms 6,816 KB
testcase_08	AC	1,436 ms 6,820 KB
testcase_09	AC	1,537 ms 6,816 KB
testcase_10	AC	1,487 ms 6,816 KB
testcase_11	AC	1,386 ms 6,816 KB
testcase_12	AC	1,552 ms 6,816 KB
testcase_13	AC	1,526 ms 6,816 KB
testcase_14	AC	1,696 ms 6,816 KB
testcase_15	AC	1,627 ms 6,816 KB
testcase_16	AC	1,491 ms 6,820 KB
testcase_17	AC	1,279 ms 6,820 KB
testcase_18	AC	1,583 ms 6,820 KB
testcase_19	AC	1,459 ms 6,820 KB
testcase_20	AC	2 ms 6,820 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#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 srep(i,s,t) for(int i=(int)(s);i<(int)(t);++i)
#define each(a,b) for(auto& (a): (b))
#define all(v) (v).begin(),(v).end()
#define len(v) (int)(v).size()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define cmx(x,y) x=max(x,y)
#define cmn(x,y) x=min(x,y)
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define spair(p) cout<<#p<<": "<<p.fi<<" "<<p.se<<endl
#define sar(a,n) cout<<#a<<":";rep(pachico,n)cout<<" "<<a[pachico];cout<<endl
#define svec(v) cout<<#v<<":";rep(pachico,v.size())cout<<" "<<v[pachico];cout<<endl
#define svecp(v) cout<<#v<<":";each(pachico,v)cout<<" {"<<pachico.first<<":"<<pachico.second<<"}";cout<<endl
#define sset(s) cout<<#s<<":";each(pachico,s)cout<<" "<<pachico;cout<<endl
#define smap(m) cout<<#m<<":";each(pachico,m)cout<<" {"<<pachico.first<<":"<<pachico.second<<"}";cout<<endl

using namespace std;

typedef pair<int,int> P;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<double> vd;
typedef vector<P> vp;
typedef vector<string> vs;

const int MAX_N = 100005;

template<typename T> class ListIterator;

// 常に不変なイテレーターが欲しい場合に別のコンテナにこれをかぶせる形で使うとよい
template<typename T>
class List {
public:
    using iterator = ListIterator<T>;

private:
    int N, sz;
    vector<int> prev, next;
	vector<T> container;
	friend ListIterator<T>;

public:
    // 添字 next_index の要素の前に添字 now_index の要素の iterator を挿入
    iterator insert(int next_index, int now_index){
        sz++;
        prev[now_index] = prev[next_index], next[now_index] = next_index;
        next[prev[next_index]] = now_index, prev[next_index] = now_index;
        return iterator(this, now_index);
    }
    // 添字 index の値を削除 → 次の要素の iterator を返す
    iterator erase(int index){
        sz--;
        next[prev[index]] = next[index], prev[next[index]] = prev[index];
        return iterator(this, next[index]);
    }

public:
	friend ostream& operator<< (ostream& os, List& ls){
		for(auto& val : ls){
			os << val << ' ';
		}
        return os;
    }
	const T& operator[](size_t index) const {
		return container[index];
	}
	T& operator[](size_t index){
		return container[index];
	}
	size_t size(){
		return sz;
	}
	bool empty(){
		return size() == 0;
	}
    T& front(){
        return container[next[N]];
    }
    T& back(){
        return container[prev[N]];
    }
	iterator begin(){
		return iterator(this, next[N]);
	}
	iterator end(){
		return iterator(this, N);
	}
	iterator insert(const iterator& itr, int index){
        return insert(itr.index, index);
    }
    iterator erase(const iterator& itr){
        return erase(itr.index);
    }
	iterator push_back(){
		return insert(N, prev[N]+1);
	}
	void pop_back(){
		erase(prev[N]);
	}
	void clear(){
        N = sz = 0;
		prev = next = {0};
        container.clear();
	}

public:
	List() : N(0), sz(0), prev({0}), next({0}){}
	List(int _N) : container(_N){
		build();
	}
	List(int _N, T val) : container(_N, val){
		build();
	}
	List(const List& ls):
		N(ls.N), sz(ls.sz), prev(ls.prev), next(ls.next), container(ls.container){}
    List(const vector<T>& vec) : container(vec){
        build();
    }
    List& operator=(const List& ls){
        container = ls.container;
        build();
        return *this;
    }
    List& operator=(const vector<T>& vec){
        container = vec;
        build();
        return *this;
    }
    List& operator=(const List&& ls) noexcept {
        container = ls.container;
        build();
        return *this;
    }
	List(initializer_list<T> vals) : container(vals.begin(), vals.end()){
		build();
	}
	iterator push_back(T val){
        N++, sz++;
        prev.push_back(N-1), prev[0] = N;
        next.back() = N, next.push_back(0);
		container.push_back(val);
        return iterator(this, N-1);
	}
	void build(){
		N = sz = (int)container.size();
		prev.resize(N+1), next.resize(N+1);
		iota(prev.begin(), prev.end(), -1), iota(next.begin(), next.end(), 1);
        prev[0] = N, next[N] = 0;
	}
};

template<typename T>
class ListIterator : public std::iterator<bidirectional_iterator_tag, T, ptrdiff_t, T*, T&>{
private:
	friend List<T>;
	List<T>* ls_ptr;
    int index;

private:
	ListIterator(List<T>* ls, int _index) : ls_ptr(ls), index(_index){}

public:
    ListIterator(){}
	ListIterator(const ListIterator& itr){
		ls_ptr = itr.ls_ptr, index = itr.index;
	}
    operator int() const noexcept { return index; }
    ListIterator& operator=(const ListIterator& itr) & {
        ls_ptr = itr.ls_ptr, index = itr.index;
        return *this;
    }
    ListIterator& operator=(ListIterator&& itr) & noexcept {
        ls_ptr = itr.ls_ptr, index = itr.index;
        return *this;
    }
	T& operator*() const {
		return ls_ptr->container[index];
	}
	T* operator->() const {
		return &ls_ptr->container[index];
	}
	ListIterator& operator++(){
		index = ls_ptr->next[index];
		return *this;
	}
	ListIterator operator++(int){
		ListIterator res = *this;
		index = ls_ptr->next[index];
		return res;
	}
	ListIterator& operator--(){
		index = ls_ptr->prev[index];
		return *this;
	};
	ListIterator operator--(int){
		ListIterator res = *this;
		index = ls_ptr->prev[index];
		return res;
	};
	bool operator==(const ListIterator& itr) const {
		return !(*this != itr);
	};
	bool operator!=(const ListIterator& itr) const {
		return this->ls_ptr != itr.ls_ptr || this->index != itr.index;
	}
	friend void swap(const ListIterator<T>& itr1, const ListIterator<T>& itr2){
		ListIterator<T> tmp = itr1;
		itr1 = itr2, itr2 = itr1;
	};
};

// 多重辺はなし
class MIS {
private:
    struct edge {
        int to, rev;
    };

    using LL = unsigned __int128;

    int V;
    LL* G;
    List<int> rem_ver;
    vector<int> cand, deg, is_rem;
    vector<vector<int> > edge_info;
    LL not_use_ver;

    int get_first_element(LL num){
        int x = __builtin_ffsll(num & (((LL)1 << 64) - 1));
        int y = __builtin_ffsll((num >> 64) & (((LL)1 << 64) - 1));
        return x ? (x - 1) : (y + 63);
    }

    // int get_size(LL num){
    //     return __builtin_popcountll(num & (((LL)1 << 64) - 1)) + __builtin_popcountll((num >> 64) & (((LL)1 << 64) - 1));
    // }

    bool finish(){
        return (int)cand.size() + (int)rem_ver.size() <= ans_size;
    }
    bool erase(int u, bool use, bool alr_use, int& add_size, vector<pair<int, int> >& erase_ver){
        erase_ver.emplace_back(u, rem_ver.erase(u));
        is_rem[u] = 0;
        if(use && !alr_use){
            add_size++, cand.push_back(u);
        }else{
            if(finish()) return false;
        }
        int v;
        for(LL st = G[u]; st != 0; st ^= (LL)1 << v){
            v = get_first_element(st);
            G[v] ^= (LL)1 << u, edge_info[u].emplace_back(v);
            deg[v]--;
            if(deg[v] == 0) erase_ver.emplace_back(v, rem_ver.erase(v)), is_rem[v] = 0;
            if(use){
                if(deg[v] > 0) not_use_ver |= ((LL)1 << v);
            }else if(((not_use_ver >> v) & (LL)1) == 0 && deg[v] == 0){
                add_size++, cand.push_back(v);
            }
        }
        return !finish();
    }
    bool erase_rec_set(bool flag, int& add_size, vector<pair<int, int> >& erase_ver){
        while(not_use_ver != 0){
            int v = get_first_element(not_use_ver);
            not_use_ver ^= ((LL)1 << v);
            if(deg[v] == 0) continue;
            if(!erase(v, flag, flag, add_size, erase_ver)) return false;
        }
        return true;
    }
    bool erase_rec(int u, bool use, int& add_size, vector<pair<int, int> >& erase_ver){
        if(!erase(u, use, false, add_size, erase_ver)){
            not_use_ver = 0;
            return false;
        }
        while(not_use_ver != 0){
            if(!erase_rec_set(false, add_size, erase_ver)){
                not_use_ver = 0;
                return false;
            }
        }
        return true;
    }
    bool preprocessing(int& add_size, vector<pair<int, int> >& erase_ver){
        for(int u : rem_ver){
            if(is_rem[u] == 1 && deg[u] <= 1){
                if(!erase_rec(u, true, add_size, erase_ver)) return false;
            }
        }
        return true;
    }
    void postprocessing(int& add_size, vector<pair<int, int> >& erase_ver){
        for(int i = 0; i < add_size; i++){
            cand.pop_back();
        }
        for(int i = (int)erase_ver.size() - 1; i >= 0; i--){
            auto& p = erase_ver[i];
            while(!edge_info[p.first].empty()){
                int q = edge_info[p.first].back();
                G[q] |= (LL)1 << p.first, deg[q]++;
                edge_info[p.first].pop_back();
            }
            rem_ver.insert(p.second, p.first);
            is_rem[p.first] = 1;
        }
    }
    bool check(){
        if(rem_ver.empty()){
            if((int)cand.size() > ans_size){
                ans_size = (int)cand.size(), ans_set = cand;
            }
            return false;
        }
        return true;
    }
    void transition(int next_ver, bool use){
        int pl_add_size = 0;
        vector<pair<int, int> > pl_erase_ver;
        if(!erase_rec(next_ver, use, pl_add_size, pl_erase_ver)){
            return postprocessing(pl_add_size, pl_erase_ver);
        }
        if(check()) judge();
        return postprocessing(pl_add_size, pl_erase_ver);
    }
    int find_max_degree_ver(){
        int max_deg = -1, next_ver = -1;
        for(int v : rem_ver){
            if(deg[v] > max_deg) max_deg = deg[v], next_ver = v;
        }
        return next_ver;
    }
    void judge(){
        int add_size = 0;
        vector<pair<int, int> > erase_ver;
        if(!preprocessing(add_size, erase_ver)) return postprocessing(add_size, erase_ver);
        if(!check()) return postprocessing(add_size, erase_ver);
        int next_ver = find_max_degree_ver();
        transition(next_ver, true);
        if(deg[next_ver] > 1 && (int)cand.size() + (int)rem_ver.size() > ans_size + 1){
            transition(next_ver, false);
        }
        return postprocessing(add_size, erase_ver);
    }
public:
    vector<int> ans_set;
    int ans_size;

    MIS(int node_size) :
        V(node_size), rem_ver(V), deg(V, 0), is_rem(V, 1), edge_info(V), not_use_ver(0), ans_size(0){
        for(int i = 0; i < V; i++) rem_ver[i] = i;
        G = new LL[V];
    }
    void add_edge(int u, int v){
        G[u] |= (LL)1 << v, G[v] |= (LL)1 << u;
        deg[u]++, deg[v]++;
    }
    int solve(){
        judge();
        return ans_size;
    }
};

int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
    ll S;
    cin >> S;
    S = S * 12345LL % 1000003LL;
    int n = (S % 120) + 2, m = 0;
    MIS mis(n);
    S = S * 12345LL % 1000003LL;
    ll P = S;
    rep(i,n){
        srep(j,i+1,n){
            S = S * 12345LL % 1000003LL;
            if(S >= P){
                mis.add_edge(i, j);
                m++;
            }
        }
    }
    if(m == 0){
        cout << "-1\n";
    }else{
        cout << mis.solve() + 1 << "\n";
        for(int i = 0; i < len(mis.ans_set)-1; i++){
            cout << mis.ans_set[i] + 1 << " ";
        }
        cout << mis.ans_set.back() + 1 << "\n";
    }
    return 0;
}

yukicoder

結果

テストケース

ソースコード