zoukankan      html  css  js  c++  java
  • 联合省选 2020 (A卷) 题解

    [题解]

    (A)

    题意等价于求一个单调不增函数和单调不降函数的第一个交点。

    线段树上二分即可。

    时间复杂度 : (O(NlogN))

    (B)

    (f(k) = sum_{i=0}^mb_iinom{k}{i})

    (sum_{k=0}^nf(k) imes x^k imes inom{n}{k}=sum_{k=0}^nsum_{i=0}^{min(m,k)}b_iinom{k}{i}x^kinom{n}{k}\=sum_{i=0}^mb_iinom{n}{i}sum_{k=i}^{n}x^kinom{n-i}{k-i}\=sum_{i=0}^mb_iinom{n}{i}x^{i}(x+1)^{n-i})

    时间复杂度 : (O(MlogN))

    (C)

    保序回归问题简要记录 (详见集训队论文) :

    定义 :

    对于一个正整数 (p) , 给定一个偏序关系 (有向无环图) , 图中每个点有权值 ((w_{i} , y_{i})) , 需要给每个点分配一个权值 , 使得回归代价 (sum_{iin U} w_i|f_i-y_i|^p) 最小 , 且一条路径上的 (f_{i}) 递增。

    我们将上述问题称为 (L_{p}) 问题。

    (L_{p}) 均值的定义 :

    我们定义 (L_{p}) 均值是使 (sumlimits_{iin U}w_i|k-y_i|^p) 最小的 (k) , 也就是所有 (f_{i}) 相等时的最优取值。对目标式求导 , 解出的零点即为 (L_{p}) , 特别地 , 当 (p = 1) 时 ,(L)(w_{i}) 的带权中位数 (left(sumlimits_{iin U}w_iy_i ight)left/left(sumlimits_{iin U}w_i ight) ight.) 而当 (p > 1) 时 , (L_{p}) 均值是唯一的 , 且一定存在一组最优解 , 使得对于每个 (f_{i}) 都能找到某个点集使其带权中位数为 (f_{i})

    整体二分 :

    运用整体二分 , 可以求得这个问题的近似解。

    定义 (solve(U , l , r)) 表示点集为 (U) , (f_{i}) 取值在 ([l , r]) 间的 (L_{p}) 问题。 那么我们希望通过某种方式求出哪些点的 (f_{i} leq mid) , 哪些 (> mid)。 这样就可以构造两个规模减半的子问题。

    我们尝试构造另一个问题 , 设开区间 ((a , b))([l , r]) 包含 , 且 (U) 的任意非空子集的 (L_{p}) 均值均不在 ((a , b)) 间 , 这个问题保证了对于任意的子问题都满足一组最优解使得其取值都不在 ((a , b)) 间。

    引理 :

    (solve(U , a , b)) 的一组最优解为 ( ilde z) , 且满足 (forall iin U, ilde z_{i} otin (a,b)) , 则存在 (solve(U , l , r)) 的一组最优解 (z) , 使得 (z_{i} leq a) 当且仅当 ( ilde z_{i} = a)

    类似地有 (z_igeq bLeftrightarrow ilde z_i=b)

    由于 (U) 的任意非空子集的并集是有限集 , 所以一定存在足够小的 (epsilon) 使得 ((mid , mid + epsilon)) 满足条件。那么就可以由 (solve(mid , mid + epsilon)) 得到 , 当 (epsilon) 无限趋近于 (0) 时 , 节点 (i) 的权值是 (w_i[(mid+epsilon-y_i)^p-(mid-y_i)^p]) , 把所有点权除以 (epsilon) , 权值变为 (w_{i}(mid - y_{i})^p)((mid - y_{i})) 处的导数 , 当 (p = 2) 时为 (2w_{i}(mid - y_{i}))。 偏序关系可以看做求最小权闭合子图。 至此 , (L_{p}) 问题得到了完美的解决。

    回到本题 , 题目相当于给定了 (N)(01) 极大线性无关组 , 从某组向量中拿去一个 , 加入新向量 ,这两者之间构成偏序关系 , 最后要求代价最小。 这本质上就是一个 (L_{2}) 问题。 唯一需要注意的是 , 本题要求的解是整数 , 因此在 (l + 1 = r) 时跑最小权闭合子图选择每个点即可。

    (D)

    考虑如何拿到 (60) 分 , 不妨用 (dp_{S}) 表示目前已经确定了集合 (S) 的最小代价 , 枚举下一个该放谁即可。 需要用到一些"费用提前计算" 的思想。 时间复杂度 : (O(2 ^ NN ^ 2))

    考虑优化 , 不妨记 (in_{S , i}) 表示 (S) 集合内连向点 (i) 的边数 , (out_{S , i}) 表示 (i) 连向 (S) 集合内的边数。 由于本题空间限制较紧 , 需要将 (in) 拆成 (in')(in'') , (out) 拆成 (out')(out'') , 分别算前 (frac{m}{2}) 位和后 (frac{m}{2}) 位。我们还要计算 (g_{S}) 表示集合 (S) 连向集合外的边数 , 这可以通过容斥原理得到。

    至此 , 我们得到了一个时间复杂度为 (O(2 ^ M * M)) , 空间复杂度为 (O(2 ^ {frac{M}{2}} * M)) 的优秀做法。具体细节见 [代码]

    (E)

    考虑从根开始 (DFS) , 每次将节点权值加入一个数据结构中。

    我们要求这个数据结构能支持 :

    (1.) 整体加一

    (2.) 查询异或和

    (3.) 合并两个数据结构

    看到 , "异或和" , 不难想到使用字典树。 但在这题中从高位到低位建字典树是不行的 , 不妨从低位到高位建树。 这样 , 每次加一相当于翻转了一条链上的左右子树。

    通过分析后得到时间复杂度为 (O(NlogN))

    (F)

    首先我们有 (sum_{d|x}varphi(d)=x)

    答案可以表示为 :

    (sum_{T}left(sum_{i=1}^{n-1}w_{e_i} ight) imesgcd(w_{e_1},w_{e_2},dots,w_{e_{n-1}}))

    (=sum_{T}left(sum_{i=1}^{n-1}w_{e_i} ight) imesleft(sum_{d|gcd(w_{e_1},w_{e_2},dots,w_{e_{n-1}})}varphi(d) ight))

    (=sum_{d=1}^{W}varphi(d) imesleft(sum_{substack{T\d|w_{e_1},dots ,d|w_{e_{n-1}}}}sum_{i=1}^{n-1}w_{e_i} ight))

    其中 (W) 表示最大边权。

    后面那部分显然是所有 (w_{e_{i}})(d_{i}) 倍数的边组成的子图的生成树边权和。

    普通的矩阵树定理 , 我们求得的是生成树的边权乘积之积。 而对于求和 , 不妨令每条边为一个多项式 ((wx + 1)) , 运行矩阵树定理 , 最终要求的是答案多项式中 ((N - 1)) 项的系数。

    时间复杂度 (O(frac{sumsigma_0(w_{e_i})}{n-1}cdot n^3)=O(n^2sumsigma_0(w_{e_i})))

    [代码]

    (A)

    #ifdef _MSC_VER
    #define _CRT_SECURE_NO_WARNINGS
    #endif
    
    #include<bits/stdc++.h>
    
    using namespace std;
    
    #ifndef LOCAL
    	  #define eprintf(...) fprintf(stderr , _VA_ARGS)
    #else
    	  #define eprintf(...) 42
    #endif
    
    // define evenbao
    #define PII pair<int , int>
    #define FI first
    #define SE second
    #define MP make_pair
    
    typedef long long LL;
    
    template <typename T> inline void chkmax(T &x , T y) { x = max(x , y); }
    template <typename T> inline void chkmin(T &x , T y) { x = min(x , y); }
    template <typename T> inline void read(T &x) {
        T f = 1; x = 0;
        char c = getchar();
        for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
        for (; isdigit(c); c = getchar()) x = (x << 3) + (x << 1) + c - '0';
        x *= f;
    }
    
    const int N = 2e6 + 10;
    
    int n , m , ans , ansp;
    int opt[N] , type[N] , b[N] , x[N] , y[N] , k[N];
    
    struct Segment_Tree {
    		int mn[N * 4] , tag[N * 4] , mx[N * 4] , mxp[N * 4];
    		inline void pushup(int now) {
    				mn[now] = min(mn[now << 1] , mn[now << 1 | 1]);
    				mx[now] = max(mx[now << 1] , mx[now << 1 | 1]);
    				if (mx[now << 1 | 1] >= mx[now << 1]) mxp[now] = mxp[now << 1 | 1];
    				else mxp[now] = mxp[now << 1];
    		}
    		inline void pushdown(int now , int l , int r) {
    				int mid = (l + r) >> 1;
    				tag[now << 1] += tag[now]; tag[now << 1 | 1] += tag[now];
    				mn[now << 1] += tag[now]; mn[now << 1 | 1] += tag[now];
    				mx[now << 1] += tag[now]; mx[now << 1 | 1] += tag[now];
    				tag[now] = 0;
    				return;
    		} 
    		inline void build(int now , int l , int r) {
    				if (l == r) { mn[now] = mx[now] = 0; mxp[now] = l; return; }
    				int mid = (l + r) >> 1;
    				build(now << 1 , l , mid); build(now << 1 | 1 , mid + 1 , r);
    				pushup(now);
    		}
    		inline void modify(int now , int l , int r , int ql , int qr , int val) {
    				if (l == ql && r == qr) {
    						mx[now] += val; mn[now] += val; tag[now] += val;
    						return;
    				}
    				pushdown(now , l , r);
    				int mid = (l + r) >> 1;
    				if (mid >= qr) modify(now << 1 , l , mid , ql , qr , val);
    				else if (mid + 1 <= ql) modify(now << 1 | 1 , mid + 1 , r , ql , qr , val);
    				else modify(now << 1 , l , mid , ql , mid , val) , modify(now << 1 | 1 , mid + 1 , r , mid + 1 , qr , val);
    				pushup(now);
    		}
    } ls , rs;
    
    inline void upd(int y , int x) {
    		if (y > ans) ans = y , ansp = x;
    		else if (y == ans) chkmax(ansp , x);
    } 
    inline void query(int now , int l , int r) {
    		if (ls.mx[now] <= rs.mn[now]) {
    				upd(ls.mx[now] , ls.mxp[now]);
    				return;
    		}
    		if (rs.mx[now] <= ls.mn[now]) {
    				upd(rs.mx[now] , rs.mxp[now]);
    				return;
    		}
    		ls.pushdown(now , l , r); rs.pushdown(now , l , r);
    		int mid = (l + r) >> 1;
    		query(now << 1 , l , mid); query(now << 1 | 1 , mid + 1 , r);
    }
    
    inline void solve() {
    		ans = ansp = 0;
    		query(1 , 1 , n);
    		if (!ans) printf("Peace
    ");
    		else printf("%d %d
    " , b[ansp] , ans * 2);
    }
    int main() {
    		
    		read(n);
    		for (int i = 1; i <= n; ++i) {
    				read(type[i]);
    				if (type[i] == 1) {
    						read(opt[i]);
    						read(x[i]);
    						read(y[i]);
    						b[++m] = x[i];
    				} else read(k[i]);
    		}
    		sort(b + 1 , b + m + 1);
    		m = unique(b + 1 , b + m + 1) - b - 1;
    		for (int i = 1; i <= n; ++i)
    				if (type[i] == 1) x[i] = lower_bound(b + 1 , b + m + 1 , x[i]) - b;
    		ls.build(1 , 1 , m); rs.build(1 , 1 , m);
    		for (int i = 1; i <= n; ++i) {
    				if (type[i] == 1) {
    						if (opt[i] == 0) ls.modify(1 , 1 , m , x[i] , m , y[i]);
    						else rs.modify(1 , 1 , m , 1 , x[i] , y[i]);
    						solve();
    				} else {
    						if (opt[k[i]] == 0) ls.modify(1 , 1 , m , x[k[i]] , m , -y[k[i]]);
    						else rs.modify(1 , 1 , m , 1 , x[k[i]] , -y[k[i]]);
    						solve();
    				}
    		}
    	  return 0;
    }
    

    (B)

    #include<bits/stdc++.h>
    using namespace std;
    
    typedef long long LL;
    
    const int M = 2000 + 10;
    
    int n , x , m , mod;
    int S[M][M] , a[M];
    
    template <typename T> inline void chkmax(T &x , T y) { x = max(x , y); }
    template <typename T> inline void chkmin(T &x , T y) { x = min(x , y); }
    template <typename T> inline void read(T &x) {
    	T f = 1; x = 0;
    	char c = getchar();
    	for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
    	for (; isdigit(c); c = getchar()) x = (x << 3) + (x << 1) + c - '0';
    	x *= f;
    }
    
    inline void inc(int &x , int y) { x = x + y < mod ? x + y : x + y - mod; }
    inline void dec(int &x , int y) { x = x - y >= 0 ? x - y : x - y + mod; } 
    inline int quickpow(int a , int n) {
    	int b = a , res = 1;
    	for (; n; n >>= 1 , a = (LL) a * a % mod)
    		if (n & 1) res = (LL) res * a % mod;
    	return res;
    }
    
    int main() {
    	
    	read(n); read(x); read(mod); read(m);
    	for (int i = 0; i <= m; ++i) read(a[i]);
    	S[0][0] = 1;
    	for (int i = 1; i <= m; ++i)
    	for (int j = 1; j <= i; ++j)
    		S[i][j] = (1LL * S[i - 1][j] * j % mod + S[i - 1][j - 1]) % mod;
    	int ans = 0;
    	for (int i = 0; i <= m; ++i) {
    		int coef = 1 , res = 0;
    		for (int j = 0; j <= i; ++j) {
    			if (j) coef = (LL) coef * (n - j + 1) % mod;
    			inc(res , (LL) S[i][j] * coef % mod * quickpow(x , j) % mod * quickpow(x + 1 , n - j) % mod);
    		}
    		res = (LL) res * a[i] % mod;
    		inc(ans , res);
    	}
    	printf("%d
    " , ans);
    	return 0;
    }
    

    (C)

    // XYX
    #include<bits/stdc++.h>
    using namespace std;
    const int MAXN = 1e3 + 5;
    const int Maxb = 64;
    typedef long long ll;
    template <typename T> void chkmax(T &x, T y) {x = max(x, y); }
    template <typename T> void chkmin(T &x, T y) {x = min(x, y); } 
    template <typename T> void read(T &x) {
    	x = 0; int f = 1;
    	char c = getchar();
    	for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
    	for (; isdigit(c); c = getchar()) x = x * 10 + c - '0';
    	x *= f;
    }
    struct LinearBasis {
    	ll a[Maxb], b[Maxb];
    	void init() {
    		memset(a, 0, sizeof(a));
    		memset(b, 0, sizeof(b));
    	}
    	void ins(ll x, ll y) {
    		for (int i = Maxb - 1; i >= 0; i--)
    			if (x & (1ll << i)) {
    				if (a[i] == 0) {
    					a[i] = x;
    					b[i] = y;
    					return;
    				} else {
    					x ^= a[i];
    					y ^= b[i];
    				}
    			}
    	}
    	ll query(ll x) {
    		ll ans = 0;
    		for (int i = Maxb - 1; i >= 0; i--)
    			if (x & (1ll << i)) {
    				x ^= a[i];
    				ans ^= b[i];
    			}
    		return ans;
    	}
    } LB;
    namespace NetworkFlow {
    	const ll INF = 1e18;
    	const int MAXP = 1e5 + 5;
    	struct edge {
    		int dest; ll flow;
    		unsigned pos;
    	}; vector <edge> a[MAXP];
    	int tot, s, t, dist[MAXP];
    	unsigned curr[MAXP];
    	void init(vector <int> u, int n) {
    		for (auto x : u) a[x].clear();
    		a[s = n + 1].clear();
    		a[t = n + 2].clear();
    	}
    	void addedge(int x, int y, ll z) {
    		a[x].push_back((edge) {y, z, a[y].size()});
    		a[y].push_back((edge) {x, 0, a[x].size() - 1});
    	}
    	ll dinic(int pos, ll limit) {
    		if (pos == t) return limit;
    		ll used = 0, tmp;
    		for (unsigned &i = curr[pos]; i < a[pos].size(); i++)
    			if (a[pos][i].flow != 0 && dist[pos] + 1 == dist[a[pos][i].dest] && (tmp = dinic(a[pos][i].dest, min(limit - used, a[pos][i].flow)))) {
    				used += tmp;
    				a[pos][i].flow -= tmp;
    				a[a[pos][i].dest][a[pos][i].pos].flow += tmp;
    				if (used == limit) return used;
    			}
    		return used;
    	}
    	bool bfs(vector <int> u) {
    		static int q[MAXP];
    		int l = 0, r = 0;
    		for (auto x : u) dist[x] = 0;
    		dist[t] = 0, dist[s] = 1, q[0] = s;
    		while (l <= r) {
    			int tmp = q[l];
    			for (unsigned i = 0; i < a[tmp].size(); i++)
    				if (dist[a[tmp][i].dest] == 0 && a[tmp][i].flow != 0) {
    					q[++r] = a[tmp][i].dest;
    					dist[q[r]] = dist[tmp] + 1;
    				}
    			l++;
    		}
    		return dist[t] != 0;
    	}
    	pair <vector <int>, vector <int>> flow(vector <int> u) {
    		while (bfs(u)) {
    			for (auto x : u) curr[x] = 0;
    			curr[s] = curr[t] = 0;
    			dinic(s, INF);
    		}
    		bfs(u);
    		vector <int> resx, resy;
    		for (auto x : u)
    			if (dist[x]) resx.push_back(x);
    			else resy.push_back(x);
    		return make_pair(resx, resy);
    	}
    }
    int n, m, res[MAXN]; vector <int> a[MAXN];
    ll c[MAXN]; int v[MAXN], x[MAXN], y[MAXN];
    void work(vector <int> u, int l, int r) {
    	if (l == r) {
    		for (auto x : u) res[x] = l;
    		return;
    	}
    	NetworkFlow :: init(u, n);
    	int mid = (l + r) / 2, s = n + 1, t = n + 2;
    	for (auto x : u) {
    		NetworkFlow :: addedge(x, t, 1ll * (v[x] - mid) * (v[x] - mid));
    		NetworkFlow :: addedge(s, x, 1ll * (v[x] - mid - 1) * (v[x] - mid - 1));
    		for (auto y : a[x])
    			NetworkFlow :: addedge(y, x, NetworkFlow :: INF);
    	}
    	pair <vector <int>, vector <int>> res = NetworkFlow :: flow(u);
    	work(res.first, l, mid), work(res.second, mid + 1, r);
    }
    int main() {
    	read(n), read(m);
    	for (int i = 1; i <= n; i++)
    		read(c[i]);
    	for (int i = 1; i <= n; i++)
    		read(v[i]);
    	LB.init();
    	for (int i = 1; i <= m; i++) {
    		read(x[i]);
    		LB.ins(c[x[i]], 1ll << (i - 1));
    	}
    	for (int i = 1; i <= n; i++) {
    		ll mask = LB.query(c[i]);
    		for (int j = 1; j <= m; j++)
    			if ((1ll << (j - 1)) & mask) a[x[j]].push_back(i);
    	}
    	LB.init();
    	for (int i = 1; i <= m; i++) {
    		read(y[i]);
    		LB.ins(c[y[i]], 1ll << (i - 1));
    	}
    	for (int i = 1; i <= n; i++) {
    		ll mask = LB.query(c[i]);
    		for (int j = 1; j <= m; j++)
    			if ((1ll << (j - 1)) & mask) a[i].push_back(y[j]);
    	}
    	int Max = 0; vector <int> u;
    	for (int i = 1; i <= n; i++) {
    		chkmax(Max, v[i]);
    		u.push_back(i);
    	}
    	work(u, 0, Max);
    	ll ans = 0;
    	for (int i = 1; i <= n; i++)
    		ans += 1ll * (res[i] - v[i]) * (res[i] - v[i]);
    	cout << ans << endl;
    	return 0;
    }
    

    (D)

    #include<bits/stdc++.h>
    
    using namespace std;
    
    typedef long long LL;
    
    #define rep(i , l , r) for (int i = (l); i < (r); ++i)
    #define popcnt(S) __builtin_popcount(S)
    #define ctz(S) __builtin_ctz(S)
    
    const int NM = 1e5 + 51, MM = 24, MK = 1 << MM, U0 = (1 << 12) - 1;
    
    int c0[MM][U0 + 1], c1[MM][U0 + 1], c2[MM][U0 + 1], c3[MM][U0 + 1];
    int s[NM], g[MK], rec[MM][MM], f[MK];
    int N , M , K , U;
    
    inline int cnt1(int x , int S) {
    	int hi = S >> 12 , lo = S & U0;
    	return c0[x][lo] + c1[x][hi]; 
    }
    inline int cnt2(int S , int x) {
    	int hi = S >> 12 , lo = S & U0;
    	return c2[x][lo] + c3[x][hi];
    }
    
    int main() {
    	 
    	 scanf("%d%d%d" , &N , &M , &K); U = (1 << M) - 1;
    	 for (int i = 1; i <= N; ++i) {
    	 	 scanf("%d" , s + i); --s[i];
    	 	 if (i > 1) ++rec[s[i - 1]][s[i]];
    	 }
    	 for (int i = 0; i < M; ++i)
    	 	 for (int S = 1; S <= U0; ++S) {
    	 	 	 int x0 = ctz(S) , x1 = x0 + 12;
    			 c0[i][S] = c0[i][S & (S - 1)] + rec[i][x0];
    			 c1[i][S] = c1[i][S & (S - 1)] + rec[i][x1];
    			 c2[i][S] = c2[i][S & (S - 1)] + rec[x0][i];
    			 c3[i][S] = c3[i][S & (S - 1)] + rec[x1][i];  	 
    		 }
    	 for (int S = 1; S <= U; ++S) {
    	 	 int S0 = S & (S - 1) , x = ctz(S);
    	 	 g[S] = g[S0] + cnt1(x , U ^ S) - cnt2(S0 , x);
    	 }
    	 memset(f , 0x3f , sizeof(f));
    	 f[0] = 0;
    	 for (int S = 0; S < U; ++S)
    	 	for (int i = 0; i < M; ++i)
    	 		if (!((S >> i) & 1))
    	 			f[S | (1 << i)] = min(f[S | (1 << i)] , f[S] + g[S] + K * (popcnt(S) + 1) * (cnt1(i, S) + cnt2(U ^ S ^ (1 << i), i)));
    	 printf("%d
    " , f[U]); 	
         return 0;
    }
    

    (E)

    #include<bits/stdc++.h>
    using namespace std;
    
    const int N = 6e5 + 10;
    
    struct Edge {
    	int to , nxt;
    } E[N << 1];
    
    typedef long long LL;
    
    int n , tot , size;
    int val[N] , ch[N * 25][2] , sum[N * 25] , sz[N * 25] , head[N] , fa[N];
    int top = 0 , rt[N] , stk[N];
    LL ans;
    
    inline void addedge(int u , int v) {
    	E[++tot] = (Edge) {v , head[u]};
    	head[u] = tot;
    }
    inline void pushup(int now) {
    	sum[now] = ((sum[ch[now][0]] ^ sum[ch[now][1]]) << 1) ^ sz[ch[now][1]];
    }
    inline int merge(int u , int v) {
    	if (!u || !v) return u + v;
    	ch[u][0] = merge(ch[u][0] , ch[v][0]);
    	ch[u][1] = merge(ch[u][1] , ch[v][1]);
    	sz[u] ^= sz[v]; sum[u] ^= sum[v];
    	return u;
    } 
    inline void upd(int now) {
    	top = 0;
    	for (int i = 0; i <= 22 && now; ++i) {
    		swap(ch[now][0] , ch[now][1]); stk[++top] = now;
    		now = ch[now][0]; 
    	}	
    	for (int i = top; i >= 1; --i)
    		pushup(stk[i]);
    	return;
    }
    inline void insert(int now , int x) {
    	top = 0;
    	for (int i = 0; i <= 22; ++i) {
    		sz[now] ^= 1; stk[++top] = now;	 
    		bool val = (x >> i) & 1; 
    		if (!ch[now][val]) ch[now][val] = ++size; 
    		now = ch[now][val]; 
    	}
    	for (int i = top; i >= 1; --i)
    		pushup(stk[i]);
    	return;
    }
    inline void solve(int u , int fa) {
    	for (int i = head[u]; i; i = E[i].nxt) {
    		int v = E[i].to; if (v == fa) continue;
    		solve(v , u); rt[u] = merge(rt[u] , rt[v]);
    	}
    	if (rt[u]) upd(rt[u]);
    	if (!rt[u]) rt[u] = ++size;
    	insert(rt[u] , val[u]); 
    	ans += (LL) sum[rt[u]];
    	return;
    }
    int main() {
    	
    	cin >> n;
    	for (int i = 1; i <= n; ++i) cin >> val[i];
    	for (int i = 2; i <= n; ++i) {
    		cin >> fa[i];
    		addedge(i , fa[i]); addedge(fa[i] , i);
    	}
    	solve(1 , 0);
    	cout << ans << "
    ";
    	return 0;
    }
    

    (F)

    #include<bits/stdc++.h>
    using namespace std;
    
    typedef long long LL;
    
    const int N = 55;
    const int mod = 998244353;
    const int M = 2e5 + 10;
    
    int tot , n , m;
    int u[M] , v[M] , w[M] , pr[M] , phi[M] , cntE[M] , f[M];
    
    template <typename T> inline void chkmax(T &x , T y) { x = max(x , y); }
    template <typename T> inline void chkmin(T &x , T y) { x = min(x , y); }
    template <typename T> inline void read(T &x) {
    	T f = 1; x = 0;
    	char c = getchar();
    	for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
    	for (; isdigit(c); c = getchar()) x = (x << 3) + (x << 1) + c - '0';
    	x *= f;
    }
    inline int quickpow(int a , int n) {
    	int res = 1;
    	for (; n; n >>= 1 , a = (LL) a * a % mod)
    		if (n & 1) res = (LL) res * a % mod;
    	return res;
    }
    struct Func { 
    	int a , b;
    } a[N][N];
    
    Func operator + (Func a , Func b) {
    	return (Func) {(a.a + b.a) % mod , (a.b + b.b) % mod};
    }
    Func operator - (Func a , Func b) {
    	return (Func) {(a.a - b.a + mod) % mod , (a.b - b.b + mod) % mod};
    }
    Func operator * (Func a , Func b) {
    	return (Func) {1LL * a.a * b.a % mod , (1LL * a.a * b.b % mod + 1LL * a.b * b.a % mod) % mod};	
    }
    Func operator / (Func a , Func b) {
    	int Inv = quickpow(b.a , mod - 2); 
    	return (Func) {1LL * a.a * Inv % mod , 1LL * (1LL * a.b * b.a % mod - 1LL * a.a * b.b % mod + mod) % mod * 1LL * Inv % mod * 1LL * Inv % mod};
    }
    
    inline void inc(int &x , int y) { x = x + y < mod ? x + y : x + y - mod; }
    inline void dec(int &x , int y) { x = x - y >= 0 ? x - y : x - y + mod; }
    inline void init(int n) {
    	phi[1] = 1;
    	for (int i = 2; i <= n; ++i) {
    		if (!f[i]) {
    			f[i] = i;
    			pr[++tot] = i;
    			phi[i] = (i - 1) % mod;
    		}
    		for (int j = 1; i * pr[j] <= n && j <= tot; ++j) {
    			int tmp = i * pr[j]; f[tmp] = pr[j];
    			if (pr[j] == f[i]) {
    				phi[i * pr[j]] = phi[i] * pr[j];
    				break;
    			} else phi[i * pr[j]] = phi[i] * phi[pr[j]];
    		}
    	}
    }
    inline void add(int u , int v , Func w) {
    	a[u][u] = a[u][u] + w; 
    	a[u][v] = a[u][v] - w;
    }
    inline int gauss() {
    	int sign = 1; Func ans = (Func) {1 , 0};
    	for (int i = 1; i <= n - 1; ++i) {
    		for (int j = i + 1; j <= n - 1; ++j) 
    			if (a[j][i].a) {
    				sign = mod - sign;
    				swap(a[i] , a[j]);
    				break;
    			}
    		ans = ans * a[i][i];
    		for (int j = i + 1; j <= n - 1; ++j) {
    			Func Inv = a[j][i] / a[i][i];
    			for (int k = i; k <= n - 1; ++k)
    				a[j][k] = a[j][k] - a[i][k] * Inv;
    		} 
    	}
    	return 1LL * sign * ans.b % mod;
    }
    inline int solve(int D) {
    	memset(a , 0 , sizeof(a));
    	for (int i = 1; i <= m; ++i) {
    		if (w[i] % D == 0) {
    			add(u[i] , v[i] , (Func) {1 , w[i]});
    			add(v[i] , u[i] , (Func) {1 , w[i]});
    		}
    	}
    	return gauss();
    }
    inline void add_factor(int x) {
    	for (int i = 1; i * i <= x; ++i) {
    		if (x % i == 0) {
    			++cntE[i];
    			if (i * i != x) ++cntE[x / i];
    		}
    	}
    }
    int main() {
    	
    	read(n); read(m); int limit = 0; 
    	for (int i = 1; i <= m; ++i) {
    		read(u[i]) , read(v[i]) , read(w[i]);
    		add_factor(w[i]);
    		chkmax(limit , w[i]);
    	}
    	init(limit);
    	int ans = 0;
    	for (int i = 1; i <= limit; ++i) {
    		if (cntE[i] < n - 1) continue;
    		inc(ans , 1LL * phi[i] * solve(i) % mod);
    	}
    	printf("%d
    " , ans);
    	return 0;
    }
    
  • 相关阅读:
    sql server 的变量
    psycopg2 (python与postgresql)
    sublime text3 设置快速生成代码
    关于 Form 表单的 enctype 属性
    根据二进制流判断文件类型
    URL编码和Base64编码 (转)
    GZip 压缩及解压缩
    HttpWebRequest 请求 Api 及 异常处理
    c# BinaryWriter 和 BinaryReader
    JQ 上传文件(单个,多个,分片)
  • 原文地址:https://www.cnblogs.com/evenbao/p/14453334.html
Copyright © 2011-2022 走看看