传送门
小G的约数
思路:每一个小于等于n的
i
i
i都存在倍数
i
∗
j
i*j
i∗j对G(n)产生贡献,
i
=
j
i=j
i=j时只产生
i
i
i个贡献,
j
>
i
j>i
j>i时产生
i
+
j
i+j
i+j的贡献,
j
<
i
j<i
j<i时在枚举
j
j
j时已经被计算过,所以只用计算
j
>
i
j>i
j>i时的贡献即可,所以
i
i
i只用枚举到
n
\sqrt{n}
n即可。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=50010;
int n,k;
ll sum[N],ans;
void init(int n=50000){
for(int i=1;i<=n;++i){
for(int j=i;j<=n;j+=i) sum[j]+=i;
}
for(int i=1;i<=n;++i){
sum[i]+=sum[i-1];
}
}
ll get(ll x,ll y){
return 1ll*(x+y)*(y-x+1)/2;
}
void solve(){
scanf("%d",&n);
init();
for(int i=1;i<=sum[n]/i;++i){
ans+=(sum[n]/i-i+1)*i;
ans+=get(i+1,sum[n]/i);
}
cout<<ans<<endl;
}
int main(){
solve();
}
将上面过程展开之后可以得到 G(n)=
∑
1
n
i
∗
⌊
n
i
⌋
\sum^{n}_{1}{i*\lfloor \frac{n}{i} \rfloor }
∑1ni∗⌊in⌋
可以直接整除分块。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=50010;
int n,k;
ll get(ll x,int y){
return 1ll*(x+y)*(y-x+1)/2;
}
ll G(ll n){
ll ans=0;
for(ll l=1,r;l<=n;l=r+1){
r=n/(n/l);
ans+=get(l,r)*(n/l);
}
return ans;
}
void solve(){
scanf("%d",&n);
cout<<G(G(n))<<endl;
}
int main(){
solve();
}
D a==a ^ b ^ b
所以 a[i]^b[j] = (1<<x) ^ ( 1<<y) 等价于 a[i] = b[j] ^ (1<<x) ^ (1<<y)
所以枚举所有的x和y查找是否存在 这样的a[i]即可,但是 unordered_map成功的T了,只能写哈希表了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300010;
int n,m;
int a[N],b[N],num[N];
ll ans;
int h[N];
// 如果x在哈希表中,返回x的下标;如果x不在哈希表中,返回x应该插入的位置
int find(int x)
{
int t = (x % N + N) % N;
while (h[t] != 0 && h[t] != x)
{
t ++ ;
if (t == N) t = 0;
}
return t;
}
void solve(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) {
scanf("%d",a+i);
int x=find(a[i]);
h[x]=a[i];
num[x]++;
}
for(int i=1;i<=m;++i){
scanf("%d",b+i);
for(int j=0;j<30;++j){
for(int k=j+1;k<30;++k) {
int x=b[i]^((1<<j)|(1<<k));//等于b[i] ^ (1<<j) ^ (1<<k)
x=find(x);
if(h[x]) ans+=num[x];
}
}
}
printf("%lld",ans);
}
int main(){
solve();
}
将上式进一步变形可得 a[i] ^ (1<<y) = b[j] ^ (1<<x)
所以 将a[i]^(1<<y)的可能值全部插入哈希表,直接查询 b[y] ^ (1<<x) 即可。
但是这样存在一个问题,x可能等于y,当x==y且上式成立时 a[i]一定等于b[j],所以 b[j] ^ (1<<x) 的 x 种情况都误加了,减掉即可,最后还要去重(除2)。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300010,M=9200010;//哈希表的范围
int n,m;
int a[N],b[N],num[M];
ll ans;
int h[M];
map<int,int> mp;
// 如果x在哈希表中,返回x的下标;如果x不在哈希表中,返回x应该插入的位置
int find(int x)
{
int t = (x % M + M) % M;
while (h[t] != -1 && h[t] != x)
{
t ++ ;
if (t == M) t = 0;
}
return t;
}
void solve(){
memset(h,-1,sizeof h);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) {
scanf("%d",a+i);
mp[a[i]]++;
for(int j=0;j<30;++j){
int x=find(a[i]^(1<<j));
h[x]=a[i]^(1<<j);
num[x]++;
}
}
for(int i=1;i<=m;++i){
scanf("%d",b+i);
for(int j=0;j<30;++j){
int x=find(b[i]^(1<<j));
if(h[x]!=-1) {
ans+=num[x];
}
}
}
for(int i=1;i<=m;++i) ans-=mp[b[i]]*30;
printf("%lld",ans>>1);
}
int main(){
solve();
}
ps:少了一层循环,但哈希表复杂度增大,实测快了一倍。