Submission #787129

#TimeUsernameProblemLanguageResultExecution timeMemory
787129Sam_a17Digital Circuit (IOI22_circuit)C++17
100 / 100
925 ms46680 KiB
#define _CRT_SECURE_NO_WARNINGS #include <bits/stdc++.h> //#include "temp.cpp" #include <cstdio> using namespace std; #ifndef ONLINE_JUDGE #define dbg(x) cerr << #x <<" "; print(x); cerr << endl; #else #define dbg(x) #endif #define sz(x) (int)x.size() #define len(x) (int)x.length() #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define clr(x) (x).clear() #define uniq(x) x.resize(unique(all(x)) - x.begin()); #define blt __builtin_popcount #define pb push_back #define popf pop_front #define popb pop_back #define ld long double #define ll long long void print(long long t) {cerr << t;} void print(int t) {cerr << t;} void print(string t) {cerr << t;} void print(char t) {cerr << t;} void print(double t) {cerr << t;} void print(long double t) {cerr << t;} void print(unsigned long long t) {cerr << t;} #include <ext/pb_ds/assoc_container.hpp> using namespace __gnu_pbds; #define nl '\n' // Indexed Set template <class T> using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>; template <class T, class V> void print(pair <T, V> p); template <class T> void print(vector <T> v); template <class T> void print(set <T> v); template <class T, class V> void print(map <T, V> v); template <class T> void print(multiset <T> v); template <class T, class V> void print(T v[],V n) {cerr << "["; for(int i = 0; i < n; i++) {print(v[i]); cerr << " "; } cerr << "]";} template <class T, class V> void print(pair <T, V> p) {cerr << "{"; print(p.first); cerr << ","; print(p.second); cerr << "}";} template <class T> void print(vector <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} // template <class T> void print(vector <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(set <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(multiset <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(Tree <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T, class V> void print(map <T, V> v) {cerr << "[ "; for (auto i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(deque <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} // for random generations mt19937 myrand(chrono::steady_clock::now().time_since_epoch().count()); // mt19937 myrand(131); // for grid problems int dx[8] = {-1,0,1,0,1,-1,1,-1}; int dy[8] = {0,1,0,-1,1,1,-1,-1}; // lowest / (1 << 17) >= 1e5 / (1 << 18) >= 2e5 / (1 << 21) >= 1e6 void fastIO() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr); } // file in/out // void setIO(string str = "") { // fastIO(); // // if(str == "input") { // // freopen("input.txt", "r", stdin); // // freopen("output.txt", "w", stdout); // // } else if(str != "") { // // freopen((str + ".in").c_str(), "r", stdin); // // freopen((str + ".out").c_str(), "w", stdout); // // } // } const long long mod = 1000002022; const int N = 3e5 + 10; vector<int> adj[N]; long long dp[N][2]; int n, m, a[N], par[N]; int cnt = 0; long long add(long long a, long long b) { return (a + b) % mod; } long long mult(long long a, long long b) { return (a * b) % mod; } long long sub(long long a, long long b) { return (a - b + 2 * mod) % mod; } struct segTree { struct node { long long a, b; long long lazy; }; vector<node> tree; int size = 1; void init(int s) { while(size < s) { size *= 2; } tree.assign(2 * size - 1, {0, 0, 0}); } node merge(node a, node b) { a.a = add(a.a, b.a); a.b = add(a.b, b.b); a.lazy = 0; return a; } void propogate(int x, int lx, int rx) { if(rx - lx == 1 || !tree[x].lazy) { tree[x].lazy = 0; return; } tree[2 * x + 1].lazy += tree[x].lazy; tree[2 * x + 1].lazy %= 2; swap(tree[2 * x + 1].a, tree[2 * x + 1].b); tree[2 * x + 2].lazy += tree[x].lazy; tree[2 * x + 2].lazy %= 2; swap(tree[2 * x + 2].a, tree[2 * x + 2].b); tree[x].lazy = 0; } void upd(int u, int is_on, long long val, int x, int lx, int rx) { if(rx - lx == 1) { if(is_on) { tree[x].a = val; tree[x].b = 0; } else { tree[x].a = 0; tree[x].b = val; } return; } int mid = (lx + rx) / 2; if(u < mid) { upd(u, is_on, val, 2 * x + 1, lx, mid); } else { upd(u, is_on, val, 2 * x + 2, mid, rx); } tree[x] = merge(tree[2 * x + 1], tree[2 * x + 2]); } void upd(int u, int is_on, long long val) { upd(u, is_on, val, 0, 0, size); } void change(int l, int r, int x, int lx, int rx) { propogate(x, lx, rx); if(lx >= r || rx <= l) { return; } if(lx >= l && rx <= r) { assert(tree[x].lazy == 0); swap(tree[x].a, tree[x].b); tree[x].lazy++; propogate(x, lx, rx); return; } int mid = (lx + rx) / 2; change(l, r, 2 * x + 1, lx, mid); change(l, r, 2 * x + 2, mid, rx); tree[x] = merge(tree[2 * x + 1], tree[2 * x + 2]); } void change(int l, int r) { change(l, r, 0, 0, size); } }; segTree seg; long long s[N]; void dfs1(int node) { if(sz(adj[node]) == 0) { s[node] = 1; } else { s[node] = sz(adj[node]); for(auto i: adj[node]) { dfs1(i); s[node] = mult(s[node], s[i]); } } } void dfs2(int node, long long coef) { if(sz(adj[node]) == 0) { dp[node][a[node]] = coef; dp[node][1 - a[node]] = 0; } else if(sz(adj[node]) == 1) { int cur = adj[node][0]; par[cur] = node; dfs2(cur, coef); } else { int first = adj[node][0], second = adj[node][1]; par[first] = par[second] = node; vector<long long> pref(sz(adj[node]) + 1, 1), suf(sz(adj[node]) + 1, 1); for(int i = 0; i < sz(adj[node]); i++) { pref[i + 1] = mult(pref[i], s[adj[node][i]]); } for(int i = sz(adj[node]); i >= 1; i--) { suf[i - 1] = mult(suf[i], s[adj[node][i - 1]]); } for(int i = 0; i < sz(adj[node]); i++) { long long mulfCof = mult(pref[i], suf[i + 1]); dfs2(adj[node][i], mult(coef, mulfCof)); } } } void init(int maxN, int M, vector<int> P, vector<int> A) { n = maxN, m = M; for(int i = 1; i < sz(P); i++) { adj[P[i]].push_back(i); } par[0] = -1; for(int i = 0; i < m; i++) { a[n + i] = A[i]; cnt += a[n + i]; } dfs1(0); dfs2(0, 1); seg.init(n + m + 5); for(int i = 0; i < m; i++) { int cur = n + i; seg.upd(cur, a[cur], dp[cur][a[cur]]); } } int count_ways(int L, int R) { seg.change(L, R + 1); seg.propogate(0, 0, seg.size); return seg.tree[0].a; }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...