AISing Programming Contest 2019C（DFS,BFS）

最新推荐文章于 2022-06-24 21:33:41 发布

转载最新推荐文章于 2022-06-24 21:33:41 发布 · 89 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/ldudxy/p/10457041.html

本文介绍了一种使用深度优先搜索(DFS)算法来解决矩阵中不同元素计数的问题。通过遍历矩阵，对每个未访问过的元素进行DFS，统计不同类型的元素数量并计算它们的组合数，从而得出所有可能的不同元素组合总数。

#include<bits/stdc++.h>
using namespace std;
int n,m;
long long a=0,b=0,ans=0;
char s[407][407];
int vis[407][407];
const int dir[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
void dfs(int x,int y){
    vis[x][y]=1;
    if(s[x][y]=='#')
        b++;
    else if(s[x][y]=='.')
        a++;
    for(int i=0;i<4;i++)
        if(x+dir[i][0]>0&&x+dir[i][0]<=n&&y+dir[i][1]<=m&&y+dir[i][1]>0&&!vis[x+dir[i][0]][y+dir[i][1]]&&s[x+dir[i][0]][y+dir[i][1]]!=s[x][y])
        //不同的点就继续dfs
            dfs(x+dir[i][0],y+dir[i][1]);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%s",s[i]+1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            if(!vis[i][j])
                a=b=0,dfs(i,j),ans+=a*b;//.和#的数目相乘即为这一条块的答案，原因相当于枚举起点终点情况为乘积和
    printf("%lld",ans);
    return 0;
}