Codeforces Round #626 Div. 2
比赛链接 https://codeforces.com/contest/1323
比赛记录 https://blog.csdn.net/cheng__yu_/article/details/105395197
B. Count Subrectangles(好题)
题意:给定长度为 n 的 a 数组和长度为 m 的 b 数组。数组元素由01组成。a数组b数组相乘构成矩阵C :
c
i
j
=
a
i
×
b
j
c_{ij}=a_i \times b_j
cij=ai×bj。问最后大小为 k 的全 1 矩阵有多少个。
(
1
≤
n
,
m
≤
40000
)
(1\le n,m\le 40000)
(1≤n,m≤40000)
思路:可以发现 a数组控制行,b数组控制列。a中连续 x 个 1 和b中连续 y 个1可以构成大小为xy的全 1 矩阵。因此枚举一下k的因子即可。
技巧:利用前缀和判断是否为数组中连续的1
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=4e4+10,mod=998244353;
int n,m;
int a[maxn],b[maxn];
ll k;
int main()
{
cin>>n>>m>>k;
vector<int> vec;
for(int i=1;i*i<=k;++i)
{
if(k%i==0)
{
vec.push_back(i);
if(i*i!=k)
vec.push_back(k/i);
}
}
for(int i=1;i<=n;++i)
cin>>a[i],a[i]+=a[i-1];
for(int i=1;i<=m;++i)
cin>>b[i],b[i]+=b[i-1];
ll ans=0;
for(auto d : vec)
{
int cnt1=0,cnt2=0;
for(int i=d;i<=n;++i)
if(a[i]-a[i-d]==d)
cnt1++;
int dd=k/d;
for(int i=dd;i<=m;++i)
if(b[i]-b[i-dd]==dd)
cnt2++;
ans+=cnt1*cnt2;
}
cout<<ans<<"\n";
return 0;
}
C. Unusual Competitions(替换)
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e6+10,mod=998244353;
int n,pref[maxn];
string s;
int main()
{
cin>>n>>s;
for(int i=1;i<=n;++i)
pref[i]=pref[i-1]+(s[i-1]=='('?1:-1);
if(pref[n])
{
puts("-1");
return 0;
}
int ans=0,last=0;
for(int i=1;i<=n;++i)
{
if(pref[i]==0&&pref[last+1]==-1)
ans+=i-last;
if(pref[i]==0)
last=i;
}
cout<<ans<<"\n";
return 0;
}
D. Present(二进制异或)
题意:给定长度为 n (4e5)的数组 a(1e7)。计算式子:
思路:考虑对答案的最终影响,从二进制来看,第 i 位为1,那么有奇数对
a
j
+
a
k
a_j+a_k
aj+ak在第 i 位为 1。高于 i 位的对答案无影响,所以对
2
i
+
1
2^{i+1}
2i+1取模,
a
j
+
a
k
a_j+a_k
aj+ak在
[
0
,
2
i
+
2
−
2
]
[0,2^{i+2}-2]
[0,2i+2−2]范围内。如果它们和第 i 位为 1 ,那么它们的范围在
[
2
i
,
2
i
+
1
−
1
]
[2^i,2^{i+1}-1 ]
[2i,2i+1−1]、
[
2
i
+
2
i
+
1
,
2
i
+
2
−
2
]
[2^i+2^{i+1},2^{i+2}-2]
[2i+2i+1,2i+2−2]之内
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=4e5+5,mod=1e9+7;
const int inf=0x3f3f3f3f;
int n,a[maxn],b[maxn];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;++i)
scanf("%d",&a[i]);
int ans=0;
for(int i=0;i<25;++i)
{
for(int j=1;j<=n;++j)
b[j]=a[j]%(1<<i+1);
sort(b+1,b+1+n);
int sum=0;
for(int j=1;j<=n;++j)
{
int l=max(0,(1<<i)-b[j]);
int r=(1<<i+1)-1-b[j];
sum+=upper_bound(b+1,b+1+n,r)-lower_bound(b+1,b+1+n,l);
l=max(0,(1<<i)+(1<<i+1)-b[j]);
r=(1<<i+2)-2-b[j];
sum+=upper_bound(b+1,b+1+n,r)-lower_bound(b+1,b+1+n,l);
if(b[j]+b[j]&(1<<i))
sum--;
}
if(sum/2&1)
ans|=1<<i;
}
printf("%d\n",ans);
return 0;
}