全wa,只能过样例,求大佬看看
查看原帖
全wa,只能过样例,求大佬看看
934810
1737923221dayang楼主2023/9/11 14:19
#include<iostream>
using namespace std;
int n,num[5001],mod=1000000007;
long long rel;
int main(){
cin>>n;
for(int i=1;i<=n;i++){
    int tmp=0;
    cin>>tmp;
    num[tmp]++;
}


for(int i=2;i<=5000;i++){
    if(num[i]>1){


    int a=(num[i]*(num[i]-1)/2)%mod;

    int b=0;
    for(int j=1;j<i;j++){
        if(j==(i-j)&&num[j]>=2) b+=(num[j]*(num[j]-1)/2)%mod;
        if(j!=(i-j)&&num[j]>0&&num[i-j]>0) b+=(num[j]*num[i-j])%mod;
    }
    rel+=(a*b)%mod;
    }
}
cout<<rel;
return 0;

}
2023/9/11 14:19
加载中...