洛谷 P3799 妖梦拼木棒

题目链接：

妖梦拼木棒 - 洛谷https://www.luogu.com.cn/problem/P3799

 明明很简单的一道暴力题，我却wa了好几次 orz ，记录一下踩到的坑。

1.取模次数不够导致炸long long（下次要记住，最好每做一次运算都取一次模）

2.第二个外层for循环一开始写的5005，数组越界了，看了好半天才看出这里出的问题。（就算写5005，换成 < 也是ok的，可我写的 <= ，害）

#include <bits/stdc++.h>
#define C1(a) (a)
#define C2(a) ((a) * ((a)-1) / 2)
#define int long long
using namespace std;
const int mod = 1e9+7;
const int maxa = 5e3+5;
int n;
int cnt[maxa];
int ans = 0;

signed main(){
    memset(cnt, 0, sizeof(cnt));
    ios::sync_with_stdio(false);
    cin >> n;
    for(int i=1; i<=n; i++){
        int k; cin >> k;
        cnt[k]++;
    }
    for(int i=2; i<=5000; i++){ //这里一开始写了5005，数组越界了！！（被自己蠢到）
        if(cnt[i]<2) continue;
        int times = C2(cnt[i]) % mod; //取两根长度为i的木棒
        for(int j=1; j<=i/2; j++){
            int k=i-j;
            if(j==k && cnt[j]>=2) ans += times * (C2(cnt[j]) % mod) % mod;
            else if(j!=k && cnt[j]>=1 && cnt[k]>=1) ans += times * (C1(cnt[j]) % mod) * (C1(cnt[k]) % mod) % mod;
            ans %= mod;
        }
    }
    cout << ans % mod << endl;
    return 0;
}