题目链接:点我啊╭(╯^╰)╮
D 0-1-Tree
题目大意:
一棵树,边为 0 0 0 或 1 1 1,要求经过 1 1 1 的边后不能走 0 0 0 边,问有多少条路?
解题思路:
将边为
0
0
0 的点放到一起,
1
1
1 的边放到一起,数量初始为
1
1
1
从该点出发的路径答案就是 该点
0
0
0 集的数量
∗
*
∗ 该点
1
1
1 集的数量
−
1
- 1
−1
2020 / 6 / 10 更新:
上面的题解是错的
考虑只经过
0
0
0 或
1
1
1 的边,从该点出发的路径数就是
0
0
0 或
1
1
1 的连通块大小
考虑走完
0
0
0 再走
1
1
1 的情况,那么找出路径上
0
0
0 和
1
1
1 的临界点
用 该点
0
0
0 集的数量
∗
*
∗ 该点
1
1
1 集的数量
−
1
- 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
1
1 到
n
n
n 的序列
p
p
p
计算有多少个区间满足:
p
l
+
p
r
=
max
i
=
l
r
p
i
p_l + p_r = \max \limits_{i = l}^{r} p_i
pl+pr=i=lmaxrpi
解题思路:
因为不重复,所以一定是峰才能满足
那么
O
n
On
On 去找峰,处理一下峰两边就行了
时间复杂度:
O
n
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);
}