牛客练习赛77

传送门
小G的约数
思路：每一个小于等于n的$i$都存在倍数$i\ast j$对G(n)产生贡献，$i=j$时只产生$i$个贡献，$j>i$时产生$i+j$的贡献，$j<i$时在枚举$j$时已经被计算过，所以只用计算$j>i$时的贡献即可，所以$i$只用枚举到$\sqrt{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)=${\sum }_1^{n}i\ast \lfloor \frac{n}{i}\rfloor$
可以直接整除分块。

#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:少了一层循环，但哈希表复杂度增大，实测快了一倍。