Educational CF Round 64 (Div. 2)___D DSU E 贪心

题目链接：点我啊╭(╯^╰)╮

D 0-1-Tree

题目大意：

     一棵树，边为 $0$ 或 $1$，要求经过 $1$ 的边后不能走 $0$ 边，问有多少条路？

解题思路：

    将边为 $0$ 的点放到一起，$1$ 的边放到一起，数量初始为 $1$
    从该点出发的路径答案就是 该点 $0$ 集的数量 $\ast$ 该点 $1$ 集的数量 $-1$

---

2020 / 6 / 10 更新：
上面的题解是错的
考虑只经过 $0$ 或 $1$ 的边，从该点出发的路径数就是 $0$ 或 $1$ 的连通块大小
考虑走完 $0$ 再走 $1$ 的情况，那么找出路径上 $0$ 和 $1$ 的临界点
用 该点 $0$ 集的数量 $\ast$ 该点 $1$ 集的数量 $-1$，就表示以该点为临界点的所有情况
然后两种情况可以都可以用上面的公式来计算

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair <int,int> pii;
const ll mod = 1e9 + 7;
const int maxn = 3e5 + 10;
int n, ans;
int f0[maxn], f1[maxn];
int si0[maxn], si1[maxn];

int find0(int v) {
	return f0[v] == v ? v : f0[v] = find0(f0[v]);
}
int find1(int v) {
	return f1[v] == v ? v : f1[v] = find1(f1[v]);
}

int main(){
	scanf("%d", &n);
	for(int i=1; i<=n; i++){
		f0[i] = f1[i] = i;
		si0[i] = si1[i] = 1;
	}
	int u, v, c;
	for(int i=1; i<n; i++){
		scanf("%d%d%d", &u, &v, &c);
		if(c){
			u = find1(u), v =find1(v);
			f1[u] = v;
			si1[v] += si1[u];
		}
		else {
			u = find0(u), v =find0(v);
			f0[u] = v;
			si0[v] += si0[u];
		}
	}
	
	ll ans = 0;
	for(int i=1; i<=n; i++)
		ans += 1ll * si0[find0(i)] * si1[find1(i)] - 1;
	printf("%lld\n", ans);
}

---

E Special Segments of Permutation

题目大意：

     给一个 $1$ 到 $n$ 的序列 $p$
     计算有多少个区间满足：
     $p_l+p_r=\underset{i=l}{\overset{r}{\max }}{p}_i$

解题思路：

    因为不重复，所以一定是峰才能满足
    那么 $On$ 去找峰，处理一下峰两边就行了
    时间复杂度：$On$

核心：思维很重要啊

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair <int,int> pii;
const ll mod = 1e9 + 7;
const int maxn = 3e5 + 10;
int n, ans, p[maxn], to[maxn];

int main() {
	scanf("%d", &n);
	for(int i=0; i<n; i++) scanf("%d", p+i);
	for(int i=1; i<n-1; i++) {
		if(p[i]>p[i-1] && p[i]>p[i+1]) {
			int cur = p[i];
			for(int j=i-1; j>=0 && p[j]<cur; j--) to[p[j]] = cur;
			for(int j=i+1; j<n && p[j]<cur; j++)
				if(to[cur-p[j]]==cur)
					ans++;
		}
	}

	printf("%d\n", ans);
}