#941828 (C++17) No.2588 Increasing Record

提出ソース

結果

問題	No.2588 Increasing Record
ユーザー	i_am_noob
提出日時	2024-01-04 14:42:36
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	269 ms / 3,000 ms
コード長	6,634 bytes
コンパイル時間	4,505 ms
コンパイル使用メモリ	273,484 KB
実行使用メモリ	59,440 KB
最終ジャッジ日時	2024-09-27 18:45:58
合計ジャッジ時間	14,182 ms
ジャッジサーバーID （参考情報）	judge3 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	3 ms 6,816 KB
testcase_02	AC	3 ms 6,940 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	3 ms 6,944 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	2 ms 6,944 KB
testcase_10	AC	3 ms 6,940 KB
testcase_11	AC	3 ms 6,940 KB
testcase_12	AC	46 ms 6,944 KB
testcase_13	AC	49 ms 6,940 KB
testcase_14	AC	50 ms 6,944 KB
testcase_15	AC	55 ms 6,940 KB
testcase_16	AC	111 ms 17,024 KB
testcase_17	AC	161 ms 28,928 KB
testcase_18	AC	217 ms 39,696 KB
testcase_19	AC	261 ms 47,720 KB
testcase_20	AC	253 ms 48,860 KB
testcase_21	AC	254 ms 48,964 KB
testcase_22	AC	269 ms 48,660 KB
testcase_23	AC	247 ms 48,664 KB
testcase_24	AC	251 ms 48,840 KB
testcase_25	AC	138 ms 25,856 KB
testcase_26	AC	175 ms 36,476 KB
testcase_27	AC	212 ms 47,144 KB
testcase_28	AC	221 ms 46,680 KB
testcase_29	AC	213 ms 47,156 KB
testcase_30	AC	122 ms 31,448 KB
testcase_31	AC	162 ms 45,416 KB
testcase_32	AC	198 ms 56,796 KB
testcase_33	AC	204 ms 59,212 KB
testcase_34	AC	211 ms 59,412 KB
testcase_35	AC	199 ms 59,440 KB
testcase_36	AC	198 ms 59,436 KB
testcase_37	AC	245 ms 49,168 KB
testcase_38	AC	164 ms 27,008 KB
testcase_39	AC	157 ms 42,616 KB
testcase_40	AC	154 ms 42,724 KB
testcase_41	AC	151 ms 43,164 KB
testcase_42	AC	156 ms 43,288 KB
testcase_43	AC	158 ms 43,848 KB
testcase_44	AC	115 ms 26,664 KB
testcase_45	AC	131 ms 34,752 KB
testcase_46	AC	154 ms 44,764 KB

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

ソースコード

raw source code

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

#define all(a) a.begin(),a.end()
#define pb push_back
#define sz(a) ((int)a.size())

using ll=long long;
using u32=unsigned int;
using u64=unsigned long long;
using i128=__int128;
using u128=unsigned __int128;
using f128=__float128;

using pii=pair<int,int>;
using pll=pair<ll,ll>;

template<typename T> using vc=vector<T>;
template<typename T> using vvc=vc<vc<T>>;
template<typename T> using vvvc=vc<vvc<T>>;

using vi=vc<int>;
using vll=vc<ll>;
using vvi=vc<vi>;
using vvll=vc<vll>;

#define vv(type,name,n,...) \
    vector<vector<type>> name(n,vector<type>(__VA_ARGS__))
#define vvv(type,name,n,m,...) \
    vector<vector<vector<type>>> name(n,vector<vector<type>>(m,vector<type>(__VA_ARGS__)))

template<typename T> using min_heap=priority_queue<T,vector<T>,greater<T>>;
template<typename T> using max_heap=priority_queue<T>;

// https://trap.jp/post/1224/
#define rep1(n) for(ll i=0; i<(ll)(n); ++i)
#define rep2(i,n) for(ll i=0; i<(ll)(n); ++i)
#define rep3(i,a,b) for(ll i=(ll)(a); i<(ll)(b); ++i)
#define rep4(i,a,b,c) for(ll i=(ll)(a); i<(ll)(b); i+=(c))
#define cut4(a,b,c,d,e,...) e
#define rep(...) cut4(__VA_ARGS__,rep4,rep3,rep2,rep1)(__VA_ARGS__)
#define per1(n) for(ll i=((ll)n)-1; i>=0; --i)
#define per2(i,n) for(ll i=((ll)n)-1; i>=0; --i)
#define per3(i,a,b) for(ll i=((ll)a)-1; i>=(ll)(b); --i)
#define per4(i,a,b,c) for(ll i=((ll)a)-1; i>=(ll)(b); i-=(c))
#define per(...) cut4(__VA_ARGS__,per4,per3,per2,per1)(__VA_ARGS__)
#define rep_subset(i,s) for(ll i=(s); i>=0; i=(i==0?-1:(i-1)&(s)))

template<typename T, typename S> constexpr T ifloor(const T a, const S b){return a/b-(a%b&&(a^b)<0);}
template<typename T, typename S> constexpr T iceil(const T a, const S b){return ifloor(a+b-1,b);}

template<typename T>
void sort_unique(vector<T> &vec){
    sort(vec.begin(),vec.end());
    vec.resize(unique(vec.begin(),vec.end())-vec.begin());
}

template<typename T, typename S> constexpr bool chmin(T &a, const S b){if(a>b) return a=b,true; return false;}
template<typename T, typename S> constexpr bool chmax(T &a, const S b){if(a<b) return a=b,true; return false;}

template<typename T, typename S> istream& operator >> (istream& i, pair<T,S> &p){return i >> p.first >> p.second;}
template<typename T, typename S> ostream& operator << (ostream& o, const pair<T,S> &p){return o << p.first << ' ' << p.second;}

#ifdef i_am_noob
#define bug(...) cerr << "#" << __LINE__ << ' ' << #__VA_ARGS__ << "- ", _do(__VA_ARGS__)
template<typename T> void _do(vector<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
template<typename T> void _do(set<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
template<typename T> void _do(unordered_set<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
template<typename T> void _do(T && x) {cerr << x << endl;}
template<typename T, typename ...S> void _do(T && x, S&&...y) {cerr << x << ", "; _do(y...);}
#else
#define bug(...) 777771449
#endif

template<typename T> void print(vector<T> x){for(auto i: x) cout << i << ' ';cout << "\n";}
template<typename T> void print(set<T> x){for(auto i: x) cout << i << ' ';cout << "\n";}
template<typename T> void print(unordered_set<T> x){for(auto i: x) cout << i << ' ';cout << "\n";}
template<typename T> void print(T && x) {cout << x << "\n";}
template<typename T, typename... S> void print(T && x, S&&... y) {cout << x << ' ';print(y...);}

template<typename T> istream& operator >> (istream& i, vector<T> &vec){for(auto &x: vec) i >> x; return i;}

vvi read_graph(int n, int m, int base=1){
    vvi adj(n);
    for(int i=0,u,v; i<m; ++i){
        cin >> u >> v,u-=base,v-=base;
        adj[u].pb(v),adj[v].pb(u);
    }
    return adj;
}

vvi read_tree(int n, int base=1){return read_graph(n,n-1,base);}

template<typename T, typename S> pair<T,S> operator + (const pair<T,S> &a, const pair<T,S> &b){return {a.first+b.first,a.second+b.second};}

template<typename T> constexpr T inf=0;
template<> constexpr int inf<int> = 0x3f3f3f3f;
template<> constexpr ll inf<ll> = 0x3f3f3f3f3f3f3f3f;

template<typename T> vector<T> operator += (vector<T> &a, int val){for(auto &i: a) i+=val; return a;}

template<typename T> T isqrt(const T &x){T y=sqrt(x+2); while(y*y>x) y--; return y;}

#define ykh mt19937 rng(chrono::steady_clock::now().time_since_epoch().count())

#include<atcoder/all>
using namespace atcoder;

using mint=modint998244353;
//using mint=modint1000000007;

template<int N>
struct Dsu{
    int par[N],siz[N];
    void init(int n){for(int i=0; i<n; ++i) par[i]=i,siz[i]=1;}
    int Find(int x){return x==par[x]?x:par[x]=Find(par[x]);}
    bool Union(int x, int y){
        x=Find(x),y=Find(y);
        if(x==y) return 0;
        if(siz[x]<siz[y]) swap(x,y);
        siz[x]+=siz[y],par[y]=x;
        return 1;
    }
    bool is_head(int x){return par[x]==x;}
};

void ahcorz(){
    int n,m; cin >> n >> m;
    vvi adj(n);
    rep(m){
        int u,v; cin >> u >> v; u--,v--;
        if(u>v) swap(u,v);
        adj[v].pb(u);
    }
    Dsu<200005> d;
    d.init(n);
    vvi adj2(n);
    rep(n){
        for(int j: adj[i]) if(d.Find(j)!=i){
            bug(i,d.Find(j));
            adj2[i].pb(d.Find(j));
            d.par[d.Find(j)]=i;
        }
    }
    vi tl(n),tr(n);
    int _t=-1;
    auto dfs=[&](auto &self, int u) -> void{
        tl[u]=++_t;
        for(int v: adj2[u]) self(self,v);
        tr[u]=_t;
    };
    dfs(dfs,n-1);
    rep(n) bug(i,tl[i],tr[i]);
    vv(int,par,n,20,-1);
    rep(n) for(int j: adj2[i]) par[j][0]=i;
    per(n-1) rep(j,1,20){
        if(par[i][j-1]==-1) par[i][j]=-1;
        else par[i][j]=par[par[i][j-1]][j-1];
    }
    auto is_anc=[&](int u, int v){
        return tl[u]<=tl[v]&&tr[u]>=tr[v];
    };
    auto lca=[&](int u, int v){
        if(is_anc(u,v)) return u;
        per(20) if(par[u][i]!=-1&&!is_anc(par[u][i],v)) u=par[u][i];
        return par[u][0];
    };
    vc<mint> dp(n);
    fenwick_tree<mint> bit(n);
    rep(n){
        dp[i]++;
        vi stk;
        sort(all(adj[i]),[&](int u, int v){return tl[u]<tl[v];});
        stk.pb(i);
        for(int j: adj[i]){
            int lst=-1;
            while(!is_anc(stk.back(),j)){
                lst=stk.back(),stk.pop_back();
            }
            if(lst!=-1) stk.pb(lca(lst,j));
            dp[i]+=bit.sum(tl[stk.back()]+1,tl[j]+1);
            stk.pb(j);
        }
        bit.add(tl[i],dp[i]);
        if(tr[i]+1<n) bit.add(tr[i]+1,-dp[i]);
    }
    mint res;
    rep(n) res+=dp[i];
    print(res.val());
}

signed main(){
    ios_base::sync_with_stdio(0),cin.tie(0);
    cout << fixed << setprecision(20);
    int t=1;
    //cin >> t;
    while(t--) ahcorz();
}

yukicoder

結果

テストケース

ソースコード