首页 > 代码库 > CodeForces 52B Right Triangles 矩阵上的计数

CodeForces 52B Right Triangles 矩阵上的计数

题目链接:点击打开链接

题意:

问有多少个与矩阵边平行的直角三角形,且三角形的3个顶点都是*


对于 L形 或者_| 形的三角形,我们只需要知道在_ 上方有多少个*即可,下底边则任取2个

所以用l[i]表示 第i列的*的个数

然后扫完一行,再把这行的*更新到 l[] 里

从上到下扫一遍得到所有 L _| 的三角形

再从下到上扫一遍 得到 ~| 和 |~ 的。。


#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <queue>
#include <functional>
#include <iostream>
using namespace std;
#define ll long long
#define N 1005
ll l[N];
ll n, m;
vector<ll> x;
char s[N][N];
int main(){
    ll i, j;
    while(cin>>n>>m){
        for(i = 1; i <= n; i++)
            scanf("%s", s[i]+1);
        ll ans = 0;
        memset(l, 0, sizeof l);
        for(i = 1; i <= n; i++)
        {
            ll pre = 0;
            x.clear();
            for(j = 1; j <= m; j++)
            {
                if(s[i][j] == '*' && l[j])
                    x.push_back(l[j]);
                if(s[i][j] == '*')
                    pre++;
            }
            for(j = 0; j < x.size(); j++)
                ans += x[j] * (pre - 1);
        //    cout<<ans<<endl;
            for(j = 1; j <= m; j++)
                if(s[i][j] == '*')l[j]++;
        }
        memset(l, 0, sizeof l);
        for(i = n; i ; i--)
        {
            ll pre = 0;
            x.clear();
            for(j = 1; j <= m; j++)
            {
                if(s[i][j] == '*' && l[j])
                    x.push_back(l[j]);
                if(s[i][j] == '*')
                    pre++;
            }
            for(j = 0; j < x.size(); j++)
                ans += x[j] * (pre - 1);
         //   for(j = 0; j < x.size(); j++)cout<<x[j]<<" "; puts("");            cout<<"    "<<ans<<endl;
            for(j = 1; j <= m; j++)
                if(s[i][j] == '*')l[j]++;
        }
        cout<<ans<<endl;
    }
    return 0;
}