洛谷[USACO21DEC] Convoluted Intervals S
题目大意
有 n n n个区间,第 i i i个区间为 [ a i , b i ] [a_i,b_i] [ai,bi],都在 [ 0 , m ] [0,m] [0,m]上。对于每一个 k ∈ [ 0 , 2 m ] k\in [0,2m] k∈[0,2m],求满足 a i + a j ≤ k ≤ b i + b j a_i+a_j\leq k\leq b_i+b_j ai+aj≤k≤bi+bj的有序对 ( i , j ) (i,j) (i,j)的个数。
1 ≤ n ≤ 2 × 1 0 5 , 1 ≤ m ≤ 5000 1\leq n\leq 2\times 10^5,1\leq m\leq 5000 1≤n≤2×105,1≤m≤5000
题解
由差分的思想可知每一个 k k k的答案为满足小于等于 k k k的 a i × a j a_i\times a_j ai×aj的 ( i , j ) (i,j) (i,j)的有序对的数量减去满足小于 k k k的 b i × b j b_i\times b_j bi×bj的 ( i , j ) (i,j) (i,j)的有序对的数量。
设 z 1 j z1_j z1j表示满足 a i = j a_i=j ai=j的 i i i的数量, z 2 j z2_j z2j表示满足 b i = j b_i=j bi=j的 i i i的数量。左边加一,右边减一,则
- c n t i + j + = z 1 i × z 1 j cnt_{i+j}+=z1_i\times z1_j cnti+j+=z1i×z1j
- c n t i + j + 1 + = z 2 i × z 2 j cnt_{i+j+1}+=z2_i\times z2_j cnti+j+1+=z2i×z2j
最后,求 c n t cnt cnt的前缀和即可得到答案。
时间复杂度为 O ( m 2 ) O(m^2) O(m2)。
code
#include<bits/stdc++.h>
using namespace std;
int n,m,a[200005],b[200005];
long long ans=0,z1[5005],z2[5005],cnt[10005];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d",&a[i],&b[i]);
++z1[a[i]];++z2[b[i]];
}
for(int i=0;i<=m;i++){
for(int j=0;j<=m;j++){
cnt[i+j]+=z1[i]*z1[j];
cnt[i+j+1]-=z2[i]*z2[j];
}
}
for(int i=0;i<=2*m;i++){
ans+=cnt[i];
printf("%lld\n",ans);
}
return 0;
}