C++算法代碼——花生與巧克力

又是一道水題無人刷……

題目來自：http://218.5.5.242:9018/JudgeOnline/problem.php?id=2411

題目描述

花生喜歡吃巧克力，他在桌子上擺了 n 堆巧克力，第 i 堆有 ai 個巧克力。

花生喜歡加法，因此若是對於一堆巧克力，它的巧克力個數剛好爲桌上另外不一樣兩堆的巧克力個數之和，花生就稱這堆巧克力是好的。

如今花生想知道桌上有多少個好的巧克力堆，他將這個問題交給了你。

 

輸入

共兩行。

第一行是整數 n ,表示巧克力的堆數。

第二行有 n 正整數，第 i 個正整數爲 ai ，表示第 i 堆巧克力有 ai 個巧克力。

 

輸出

共一行，輸出答案。

 

樣例輸入

5
2 2 3 1 1

樣例輸出

3

提示

對於 100% 的數據， 3 <= n <= 100,1 <= ai <= 10000 。

做者分析：數據水到家了，枚舉都能AC。

#include <iostream>
#include <cstring>
using namespace std;

int main(){
    int n;
    bool t = 0;
    cin >> n;
    int a[n+1],ans = 0,b[n+1];
    memset(b,0,sizeof(b));
    for (int i = 1;i <= n;i++){
        cin >> a[i];
    }
    for (int i = 1;i <= n;i++){
        for (int j = 1;j <= n;j++){
            for (int k = 1;k <= n;k++){
                if (k != i && j != i && k != j && a[j] + a[k] == a[i]){
                    b[i] = 1;
                }
            }
        }
    }
    for (int i = 1;i <= n;i++){
        if (b[i]) ans++;
    }
    cout << ans;
}