前言
该死的概率期望,一看就会,一做就废。
T1
暴力
注意此题不需要去重,不难发现行和列可以单独计算。
用r[]和c[]表示每一行和每一列的宝藏数,
宽度只有1的就解决了,那宽度不为1的呢?
维护2个前缀和,宽度大小就是r[i]-r[j-1],暴力枚举前缀和后放到桶里。利用乘法原理统计即可。
时间复杂度O(n^2+m^2+k)
代码实现:
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5,M=105;
int n,m,k,r[N],c[M],ans;
int qr[N],qc[M];
int f[N],g[N];
int main(){
scanf("%d%d%d",&n,&m,&k);
char s;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>s;
r[i]+=(s=='$');
c[j]+=(s=='$');
}
}
for(int i=1;i<=n;i++) qr[i]=qr[i-1]+r[i];
for(int i=1;i<=m;i++) qc[i]=qc[i-1]+c[i];
for(int len=1;len<=n;len++)
for(int i=1;i<=n-len+1;i++){
int j=i+len-1;
f[qr[j]-qr[i-1]]++;
}
for(int len=1;len<=m;len++)
for(int i=1;i<=m-len+1;i++){
int j=i+len-1;
g[qc[j]-qc[i-1]]++;
}
for(int i=0;i<=k;i++) ans+=f[i]*g[k-i];
cout<<ans;
return 0;
}
我们发现可以优化枚举的内容,O(m^2)枚举每一条,由于宝藏数必须为k,所以可以用双端队列O(n)扫一遍,统计即可,这里不做赘述。
正解
由于是矩阵,容易想到二维前缀和,同时我们观察到这两条线可以分成2个直角的部分,正好可以用前缀和维护,对于另一部分,就可以用后缀和维护,如图:
涂色部分的值等于
f ( x 2 , y 2 ) − f ( x 1 − 1 , y 1 − 1 ) + g ( x 1 , y 1 ) − g ( x 2 + 1 , y 2 + 1 ) f(x2,y2)-f(x1-1,y1-1)+g(x1,y1)-g(x2+1,y2+1) f(x2,y2)−f(x1−