CodeForces 52B Right Triangles 矩阵上的计数

来源:互联网 发布:中国协同软件排名 编辑:程序博客网 时间:2024/06/05 13:34

题目链接:点击打开链接

题意:

问有多少个与矩阵边平行的直角三角形,且三角形的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 1005ll 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;}


0 0
原创粉丝点击