最大队伍士兵数java,拼多多笔试：最大士兵问题

题目：输入一个N*M的矩阵，元素为1表示有一个士兵，元素为0表示没有士兵，如果一个士兵的上下左右相邻的士兵可以归为一个队伍，同一个队伍的士兵越多，队伍越强大，现在可以随意移动一个士兵，求移动之后可以得到的最大队伍的士兵数：

比如输入:

4 4

1 0 0 1

1 1 0 1

0 0 0 1

1 1 1 1

输出10。

思路：和岛问题很像，我当时没想出来，看了别人的思路后面补了一下，基本是按照下面来的。

public static int res=0;

public static int tag=1;//岛标签

public static HashMap map;

public static int tmpsum=0;//每个岛的最大士兵数量

//对岛上的元素添加标签并且统计个数

public static void process(int[][] matrix,int row,int col){

for(int i=0;i

for(int j=0;j

if(matrix[i][j]==1)

{

tag++;//这个岛的标签

tmpsum=0;

infect(matrix,i,j,row,col);//尝试对周围进行感染

System.out.println("tag: "+tag+"sum: "+tmpsum);

map.put(tag, tmpsum);//每个岛的标签和士兵数量保存在map里面

}

}

}

}

public static void process2(int[][] matrix ,int row,int col){

HashSet set;

for(int i=0;i

for(int j=0;j

if(matrix[i][j]==0){

int tmp=0;

set=new HashSet<>();

int[] sizhou=new int[4];//四周

//如果超过边界或者是0，就不用统计

sizhou[0]=i==0||matrix[i-1][j]==0?-1:matrix[i-1][j];//上面的标签

sizhou[1]=i==row-1||matrix[i+1][j]==0?-1:matrix[i+1][j];//上下左右

sizhou[2]=j==0||matrix[i][j-1]==0?-1:matrix[i][j-1];

sizhou[3]=j==col-1||matrix[i][j+1]==0?-1:matrix[i][j+1];

for(int m=0;m<4;m++){//使用set去重

if(sizhou[m]!=-1){

if(!set.contains(sizhou[m])){

set.add(sizhou[m]);

tmp+=map.get(sizhou[m]);//累加

}

}

}

if(set.size()==map.size()){

int tmpres=tmp;

res=Math.max(res,tmpres);

}else{

int tmpres=tmp+1;

res=Math.max(res,tmpres);

}

}

}

}

}

public static void infect(int[][] matrix,int i,int j,int row,int col){

if(i<0||i>=row||j<0||j>=col||matrix[i][j]!=1)

return ;

matrix[i][j]=tag;

tmpsum++;

infect(matrix,i+1,j,row,col);

infect(matrix,i-1,j,row,col);

infect(matrix,i,j+1,row,col);

infect(matrix,i,j-1,row,col);

}

public static void main(String[] args) {

// TODO Auto-generated method stub

Scanner sc=new Scanner(System.in);

map=new HashMap<>();

while(sc.hasNext()){

int n=sc.nextInt();

int m =sc.nextInt();

sc.nextLine();

int[][] matrix=new int[n][m];

for(int i=0;i

for(int j=0;j

matrix[i][j]= sc.nextInt();

}

sc.nextLine();

}

process(matrix,n,m);

process2(matrix,n,m);

System.out.println(res);

}

}