已知兩個已按升序排列的整數數組 f[] 和 g[],長度分別為 m 和 n(1 ≤ m, n ≤ 1000)。求 f[] 中每個元素比 g[] 中多少個元素大,將這些計數相加作為答案。
例如:
f = [1, 3, 5, 7, 9]
g = [2, 3, 4, 7, 8]
f[0]=1 比 g 全部小 → 0
f[1]=3 比 g[0]=2 大 → 1
f[2]=5 比 g[0..2] 大 → 3
f[3]=7 比 g[0..2] 大 → 3
f[4]=9 比 g 全部大 → 5
答案 = 0+1+3+3+5 = 12
第一行:m n
第二行:m 個整數(f[])
第三行:n 個整數(g[])
輸出支配值總和。
5 5 1 3 5 7 9 2 3 4 7 8
12
由於 m, n ≤ 1000,最多 106 次比較,完全在時限內。直接雙層循環統計即可。
利用兩數組已排序的性質,維護一個指針 j 指向 g[] 中第一個不小於 f[i] 的位置。因為 f[] 遞增,j 只會單調右移,總複雜度線性。
#include <stdio.h>
int main() {
int m, n;
scanf("%d %d", &m, &n);
int f[1000], g[1000];
for (int i = 0; i < m; i++) scanf("%d", &f[i]);
for (int j = 0; j < n; j++) scanf("%d", &g[j]);
long long ans = 0;
int j = 0;
for (int i = 0; i < m; i++) {
while (j < n && g[j] < f[i]) j++;
ans += j; // g[0..j-1] 都小於 f[i]
}
printf("%lld\n", ans);
return 0;
}
long long 存儲。