基準時間限制：1 秒 空間限制：131072 KB 分值: 5 難度：1級演算法題

 收藏

 關注

有這樣一段程式，fun會對整數陣列A進行求值，其中Floor表示向下取整：

fun(A)

    sum = 0

    for i = 1 to A.length

        for j = i+1 to A.length

            sum = sum + Floor((A[i]+A[j])/(A[i]*A[j])) 

    return sum

給出陣列A，由你來計算fun(A)的結果。例如：A = {1, 4, 1}，fun(A) = [5/4] + [2/1] + [5/4] = 1 + 2 + 1 = 4。

Input

第1行：1個數N，表示陣列A的長度(1 <= N <= 100000)。
第2 - N + 1行：每行1個數A[i]（1 <= A[i] <= 10^9)。

Output

輸出fun(A)的計算結果。

Input示例

3
1 4 1

Output示例

4

按照題目的意思來模擬的話，程式碼應該是如下所示：

#pragma GCC optimize(2)
#include <bits/stdc++.h>
using namespace std;
#define clr(a) memset(a,0,sizeof(a))
#define line cout<<"-----------------"<<endl;

typedef long long ll;
const int maxn = 1e5+10;
const int MAXN = 1e6+10;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9+7;
const int N = 1010;

ll n;
ll a[maxn];
ll fun(ll n){
	ll sum = 0;
	for(int i = 1; i <= n; i++){
		for(int j = i + 1; j <= n; j++){
			sum = sum + floor((a[i] + a[j]) / (a[i] * a[j]));
		}
	}
	return sum;
}

int main(){
	scanf("%lld", &n);
	for(int i = 1; i <= n; i++){
		scanf("%lld", &a[i]);
	}
	printf("%lld\n", fun(n));
	return 0;
}
 

毫無疑問，肯定超時。我們再來重新思考一下這道題目。

對於兩個數 x , y.   floor((x + y) / (x * y)) 只可能有三種情況。0 、 1 、 2 

當  

x == 1 y == 1 的時候結果為2 

x == 1 y == R 的時候結果為1

x == 2 y == 2  的時候結果為2

分別統計1 和 2 出現的次數 統計其對結果的貢獻

#pragma GCC optimize(2)
#include <bits/stdc++.h>
using namespace std;
#define clr(a) memset(a,0,sizeof(a))
#define line cout<<"-----------------"<<endl;

typedef long long ll;
const int maxn = 1e5+10;
const int MAXN = 1e6+10;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9+7;
const int N = 1010;

int n;
int a[maxn];

int main(){
	scanf("%d", &n);
	int num1 = 0, num2 = 0;
	for(int i = 1; i <= n; i++){
		scanf("%d", &a[i]); 
		if(a[i] == 1) num1 ++;
		else if(a[i] == 2) num2 ++;
	}
	ll ans = 0;
	ans += num2 * (num2 - 1) / 2;
	ans += num1 * (n - 1);
	cout << ans << endl;
	return 0;
}