预处理答案+求和输出,瓶颈在于求和输出部分复杂度为 O(n2),预处理部分只跑 2s 问题不大
#include<bits/stdc++.h>
using namespace std;
int n;
int a[200002];
int ans[20000002];
void init()
{
ans[1]=1;
for(int i=2;i<=20000000;i++)
{
int l=1,r=log(i)/log(2),fa=0;
while(l<=r)
{
int mid=(l+r)/2,num=(1<<mid);
if(i%num==0)
{
fa=mid;
l=mid+1;
}else{
r=mid-1;
}
}
ans[i]=i/(1<<fa);
}
// for(int i=1;i<=100;i++)
// {
// cout<<ans[i]<<" ";
// }
}
int main()
{
init();
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
long long tot=0;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
tot+=ans[a[i]+a[j]];
}
}
printf("%lld",tot);
return 0;
}
/*
2~2e7
*/