遞增三元組

給定三個整數數組
A = [A1, A2, ... AN], 
B = [B1, B2, ... BN], 
C = [C1, C2, ... CN],
請你統計有多少個三元組(i, j, k) 滿足:
1. 1 <= i, j, k <= N  
2. Ai < Bj < Ck  

【輸入格式】 
第一行包含一個整數N。
第二行包含N個整數A1, A2, ... AN。
第三行包含N個整數B1, B2, ... BN。
第四行包含N個整數C1, C2, ... CN。

對於30%的數據,1 <= N <= 100  
對於60%的數據,1 <= N <= 1000 
對於100%的數據,1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000 

【輸出格式】
一個整數表示答案

【樣例輸入】
3
1 1 1
2 2 2
3 3 3

【樣例輸出】
27 


資源約定:
峯值內存消耗(含虛擬機) < 256M
CPU消耗  < 1000ms

分析:對於任意一個b[i],a中小於b[i]的個數*c中大於b[i]的個數就是當前b[i]所能構造的所有的遞增三元組。

那麼快速求出比b[i]小的a元素個數,比b[i]大的c元素個數即可。

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int a[MAXN], b[MAXN], c[MAXN];
int n, sum;
int main()
{
	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);
	sum = 0;
	for (int i = 0; i < n; i++) {
		int x = (lower_bound(a, a + n, b[i]) - a);//a中有多少個比b[i]小的
		int y = (n - (upper_bound(c, c + n, b[i]) - c));//c中有多少個比b[i]大的
		sum += x * y;
	}
	cout << sum;
	return 0;
}

 

發佈了106 篇原創文章 · 獲贊 136 · 訪問量 7萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章