[蓝桥杯2018初赛]递增三元组-双指针,枚举,排序,前缀和
生活随笔
收集整理的這篇文章主要介紹了
[蓝桥杯2018初赛]递增三元组-双指针,枚举,排序,前缀和
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目描述
給定三個整數數組
A = [A1, A2, … AN],
B = [B1, B2, … BN],
C = [C1, C2, … CN],
請你統計有多少個三元組(i, j, k) 滿足:
輸入
第一行包含一個整數N。
第二行包含N個整數A1, A2, … AN。
第三行包含N個整數B1, B2, … BN。
第四行包含N個整數C1, C2, … CN。
1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000
輸出
一個整數表示答案
樣例輸入
3
1 1 1
2 2 2
3 3 3
樣例輸出
27
法一:
前綴和寫法代碼如下:
#include <iostream> #include <cstring> using namespace std; const int N = 100010; typedef long long LL; int a[N], b[N], c[N], cnt[N], s[N], as[N], cs[N];int main() {int n;cin >> n;for (int i = 0; i < n; i++)cin >> a[i], a[i]++;for (int i = 0; i < n; i++)cin >> b[i], b[i]++;for (int i = 0; i < n; i++)cin >> c[i], c[i]++;for (int i = 0; i < n; i++)cnt[a[i]]++;for (int i = 1; i < N; i++)s[i] = s[i - 1] + cnt[i];for (int i = 0; i < n; i++)as[i] = s[b[i] - 1];memset(s, 0, sizeof(s));memset(cnt, 0, sizeof(cnt));for (int i = 0; i < n; i++)cnt[c[i]]++;for (int i = 1; i < N; i++)s[i] = s[i - 1] + cnt[i];for (int i = 0; i < n; i++)cs[i] = s[N - 1] - s[b[i]];LL res = 0;for (int i = 0; i < n; i++) {res += (LL)as[i] * cs[i];}cout << res << endl;return 0;}法二:
#include <iostream> #include <algorithm> using namespace std; const int N = 100010; int a[N], b[N], c[N];int main() {int n;cin >> n;for (int i = 0; i < n; i++)cin >> a[i];for (int i = 0; i < n; i++)cin >> b[i];for (int i = 0; i < n; i++)cin >> c[i];sort(a, a + n);sort(b, b + n);sort(c, c + n);int p = 0, q = 0;long long res = 0;for (int i = 0; i < n; i++) {while (p < n && a[p] < b[i])p++;while (q < n && c[q] <= b[i])q++;res += p * (long long )(n - q);}cout << res << endl;return 0; } 創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的[蓝桥杯2018初赛]递增三元组-双指针,枚举,排序,前缀和的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: uplay平台设置中文的具体方法
- 下一篇: [蓝桥杯2018初赛]星期一-日期计算