B Ugly Pairs C Match Points 给定序列 $a$,求最多能匹配多少对差值绝对值大于 $z$,一个位置不能匹配多次。
显然,排序。我们不能对于每个数贪心的选最小的匹配,但是我们注意到答案最大是 $[{ n\over 2}]$,从前后一半选显然最优。
对于前一半,在后一半匹配一个最小的即可。
D 0-1-Tree 给定一棵带 $01$ 权值的无根树,求多少条形如 $111\dots 111$, $000 \dots 000$ 和 $000\dots000111\dots111$ 的路径。
并查集维护 $1$ 边连接和 $0$ 边连接的点集,枚举每个点作为第三种路径分界的情况,即 $1$ 和 $0$ 的联通块大小减一的乘积。
E Special Segments of Permutation 给定一个 $1$ 到 $n$ 的排列,求有多少对 $(l,r)$,满足 $a[l]+a[r]=\max_{i=l}^r a[i]$。
考虑枚举最大值位置,左右两边选择一对能够凑成最大值,只需要枚举短的一边,判断配对的另外一个数是否在另外一边。
复杂度证明类似于启发式合并,$O(n\log n)$。
代码 A 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 #include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <vector> #include <utility> #define ms(a,b) memset(a,b,sizeof(a)) using namespace std;typedef long long ll;typedef pair<int ,int > PII;const int mod = 998244353 ;const int inf = 1 << 30 ;const int maxn = 100000 + 5 ;int n, a[maxn];int main () { scanf ("%d" , &n); int ans = 0 , flag = 0 ; for (int i = 1 ; i <= n; i++) { scanf ("%d" , a + i); if (i > 1 ) { if (a[i] == 1 ) { if (a[i - 1 ] == 2 ) ans += 3 ; else if (a[i - 1 ] == 3 ) ans += 4 ; } else if (a[i] == 2 ) { if (a[i - 1 ] == 1 ) { if (i > 2 ) { if (a[i - 2 ] == 3 ) ans += 2 ; else ans += 3 ; } else { ans += 3 ; } } else if (a[i - 1 ] == 3 ) flag = 1 ; } else if (a[i] == 3 ) { if (a[i - 1 ] == 1 ) ans += 4 ; else if (a[i - 1 ] == 2 ) flag = 1 ; } } } if (flag) return puts ("Infinite" ), 0 ; puts ("Finite" ); cout << ans; return 0 ; }
B 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 #include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <vector> #include <utility> #include <string> #define ms(a,b) memset(a,b,sizeof(a)) using namespace std;typedef long long ll;typedef pair<int ,int > PII;const int mod = 998244353 ;const int inf = 1 << 30 ;const int maxn = 100000 + 5 ;char s[maxn];int cnt[maxn];int main () { int T; scanf ("%d" , &T); while (T--) { scanf ("%s" , s); ms (cnt, 0 ); int n = strlen (s); for (int i = 0 ; i < n; i++) cnt[s[i] - 'a' ]++; string ans; for (int i = 0 ; i < 26 ; i += 2 ) { int x = cnt[i]; while (x--) ans += i + 'a' ; } for (int i = 1 ; i < 26 ; i += 2 ) { int x = cnt[i]; while (x--) ans += i + 'a' ; } int flag = 0 ; for (int i = 1 ; i < n; i++) if (abs (ans[i] - ans[i - 1 ]) == 1 ) { flag = 1 ; break ; } if (flag) { ans = "" ; for (int i = 1 ; i < 26 ; i += 2 ) { int x = cnt[i]; while (x--) ans += i + 'a' ; } for (int i = 0 ; i < 26 ; i += 2 ) { int x = cnt[i]; while (x--) ans += i + 'a' ; } int flag = 0 ; for (int i = 1 ; i < n; i++) if (abs (ans[i] - ans[i - 1 ]) == 1 ) { flag = 1 ; break ; } if (flag) puts ("No answer" ); else puts (ans.c_str ()); } else puts (ans.c_str ()); } return 0 ; }
C 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 #include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <vector> #include <utility> #include <set> #include <queue> #include <functional> #define ms(a,b) memset(a,b,sizeof(a)) using namespace std;typedef long long ll;typedef pair<int ,int > PII;const int mod = 998244353 ;const int inf = 1 << 30 ;const int maxn = 200000 + 5 ;int n, z, a[maxn];multiset<int > st; struct node { int val, tag; bool operator <(const node& b) const { return val > b.val; } }; int main () { scanf ("%d%d" , &n, &z); for (int i = 1 ; i <= n; i++) scanf ("%d" , a + i); sort (a + 1 , a + 1 + n); int ans = 0 ; for (int i = n / 2 + 1 ; i <= n; i++) st.insert (a[i]); for (int i = 1 ; i <= n / 2 ; i++) { auto it = st.lower_bound (a[i] + z); if (it == st.end ()) continue ; st.erase (it); ans++; } cout << ans << endl; return 0 ; }
D 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 #include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <vector> #include <utility> #include <set> #ifdef XLor #define dbg(args...) do {cout << #args << " -> "; err(args);} while (0) #else #define dbg(...) #endif void err() {std::cout << std::endl;} template<typename T, typename...Args> void err(T a, Args...args){std::cout << a << ' '; err(args...);} #define ms(a,b) memset(a,b,sizeof(a)) using namespace std; typedef long long ll; typedef pair<int,int> PII; const int mod = 998244353; const int inf = 1 << 30; const int maxn = 200000 + 5; int n; vector<PII> edge[maxn]; ll ans; struct DSU { int pre[maxn], siz[maxn]; int find(int x) { return x == pre[x] ? x : pre[x] = find(pre[x]); } void join(int x, int y) { x = find(x); y = find(y); if (x == y) return ; pre[x] = y; siz[y] += siz[x]; } int check(int x, int y) { return find(x) == find(y); } void init(int n) { for (int i = 1; i <= n; i++) pre[i] = i, siz[i] = 1; } ll cal() { ll ans = 0; for (int i = 1; i <= n; i++) if (i == pre[i]) { ans += 1ll * siz[i] * (siz[i] - 1); } return ans; } int size(int u) { return siz[find(u)]; } } f[2]; set<int> nb[maxn]; int uu[maxn], vv[maxn], ww[maxn]; int main() { scanf(" %d", &n); f[0].init(n); f[1].init(n); for (int i = 2, u, v, w; i <= n; i++) { scanf(" %d%d%d", &u, &v, &w); uu[i] = u; vv[i] = v; ww[i] = w; f[w].join(u, v); } ans = f[0].cal() + f[1].cal(); dbg(ans); for (int i = 1; i <= n; i++) { if (f[0].size(i) > 1 && f[1].size(i) > 1) { ans += 1ll * (f[0].size(i) - 1) * (f[1].size(i) - 1); dbg(i, f[0].size(i) - 1, f[1].size(i) - 1); } } cout << ans << endl; return 0; }
E 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 #include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <vector> #include <utility> #ifdef XLor #define dbg(args...) do {cout << #args << " -> "; err(args);} while (0) #else #define dbg(...) #endif void err() {std::cout << std::endl;} template<typename T, typename...Args> void err(T a, Args...args){std::cout << a << ' '; err(args...);} #define ms(a,b) memset(a,b,sizeof(a)) using namespace std; typedef long long ll; typedef pair<int,int> PII; const int mod = 998244353; const int inf = 1 << 30; const int maxn = 200000 + 5; int n, a[maxn], pos[maxn]; int dp[maxn][20]; void init() { for (int j = 1; j < 20; j++) for (int i = 1; i + (1 << j) <= n + 1; i++) dp[i][j] = max(dp[i][j - 1], dp[i + (1 << (j - 1))][j - 1]); } int qmax(int l, int r) { int k = 0; while ((1 << (k + 1)) <= r - l + 1) k++; return max(dp[l][k], dp[r - (1 << k) + 1][k]); } ll solve(int l, int r) { if (r - l <= 1) return 0; dbg(l, r); int mx = qmax(l, r), mid = pos[mx]; ll ans = 0; if (mid - l < r - mid) { for (int i = l; i < mid; i++) if (pos[mx - a[i]] > mid && pos[mx - a[i]] <= r) ans++; } else { for (int i = mid + 1; i <= r; i++) if (pos[mx - a[i]] < mid && pos[mx - a[i]] >= l) ans++; } return ans + solve(l, mid - 1) + solve(mid + 1, r); } int main() { scanf(" %d", &n); for (int i = 1; i <= n; i++) scanf(" %d", a + i), dp[i][0] = a[i], pos[a[i]] = i; init(); cout << solve(1, n) << endl; return 0; }