CodeForces 52 B.Right Triangles(水~)

Description
给出一个n*m的矩阵,问其中以三个*为顶点且直角边水平竖直的直角三角形个数
Input
第一行两个整数n和m表示矩阵行列数,之后输入一个n*m字符矩阵(1<=n,m<=1000)
Output
输出满足条件的直角三角形个数
Sample Input
这里写图片描述
Sample Output
9
Solution
每一个星作为直角顶点对答案的贡献即其所处行的星的个数乘上其所处列的星的个数(除去自己),所以预处理每行每列星的个数然后枚举直角顶点累加答案即可
Code

#include
#include
using namespace std;
typedef long long ll;
const int maxn=1005;
char s[maxn][maxn];
int n,m,row[maxn],col[maxn];
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)scanf("%s",s[i]+1);
        memset(row,0,sizeof(row));
        memset(col,0,sizeof(col));
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                if(s[i][j]=='*')row[i]++,col[j]++;
        ll ans=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                if(s[i][j]=='*')ans+=(ll)(row[i]-1)*(col[j]-1);
        printf("%I64d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(Code,Forces,水题)