码迷,mamicode.com
首页 > 其他好文 > 详细

华为笔试:矩阵相邻

时间:2019-09-03 20:46:21      阅读:256      评论:0      收藏:0      [点我收藏+]

标签:笔试   include   ==   数组   int   space   roo   color   代码   

题目描述:

一个矩阵,5*5,取相邻(二个成员有一个边是相同的)的6个,输入一个6个成员列表,判断是否满足?

矩阵成员如下:

[[1,2,3,4,5],

[11,12,13,14,15],

[21,22,23,24,25],

[31,32,33,34,35],

[41,42,43,44,45]].

 

输入描述:

包含6个矩阵成员数组,如:1,2,3,4,5,11以一个空格分隔,支持多行

1,2,3,4,5,11

1,2,11,14,25,15

 

输出描述:

满足输出1,否则输出0,每一行输入一个输出

1

0

 

备注:

输入没有合法判断,每个成员不重复。

 

思路分析:

利用并查集实现。设定一个长度为6的数组,其中每个成员单独成组,接下来判断两两是否相邻,若相邻则合并为一组。最后判断是否存在孤立的组,若有则输出0,否则输出1。

 

代码:

 1 #include<iostream>
 2 #include<vector>
 3 #include<algorithm>
 4 using namespace std;
 5 
 6 vector<int>Tree(6);
 7 int findRoot(int x){
 8     if(Tree[x]==-1) return x;
 9     int tmp=findRoot(Tree[x]);
10     Tree[x] = tmp;
11     return tmp;
12 }
13 
14 int main(){
15     int A[6];
16     while(cin>>A[0]){
17         for(int i=1;i<6;i++) cin>>A[i];
18         for(int i=0;i<6;i++) Tree[i]=-1;
19         for(int i=0;i<6;i++){
20             for(int j=i+1;j<6;j++){
21                 int x=max(A[i],A[j]);
22                 int y=min(A[i],A[j]);
23                 if(x-y==1 || x-y==10){
24                     int a=findRoot(i);
25                     int b=findRoot(j);
26                     if(a!=b) Tree[a]=b;
27                 }
28             }
29         }
30         int ans=count(Tree.begin(),Tree.end(),-1)==1?1:0;
31         cout<<ans<<endl;
32     }
33     return 0;
34 }

 

华为笔试:矩阵相邻

标签:笔试   include   ==   数组   int   space   roo   color   代码   

原文地址:https://www.cnblogs.com/LJ-LJ/p/11455415.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!