三元組順序對矩陣轉置
步驟:
- 構造一個三元組,i,j,data
- 輸入一個二維矩陣
- 將二維矩陣轉成三元組
- 一個num數組,存每一列的非0元素的個數
- 一個cpot數組,存每一列第一個每0元素在new_triple中的位置
- 用一個for循環,利用cpot找到轉置後的的位置,然後行列交換就ok了
- 打印原二維數組、轉置前的三元組、轉置後的三元組
package 數據結構實驗;
import java.util.Arrays;
import java.util.Scanner;
public class MyTriple {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
int[][] arr = new int[n][m];
int i, j, k = 0;
Triple[] pre_Triple = new Triple[100];
Triple[] new_Triple = new Triple[100];
//2. 輸入一個二維矩陣
for(i = 0; i < n; i++) {
for(j = 0; j < m; j++) {
arr[i][j] = sc.nextInt();
}
}
//3. 將二維矩陣轉成三元組
for(i = 0; i < n; i++) {
for(j = 0; j < m; j++) {
if(arr[i][j] != 0) {
pre_Triple[k++] = new Triple(i,j,arr[i][j]);
}
}
}
int[] num = new int[k];
int[] cpot = new int[k];
//4. 一個num數組,存每一列的非0元素的個數
for(i = 0; i < k; i++) {
for(j = 0; j < k; j++) {
if(pre_Triple[j].j == i) {
num[i]++;
}
}
}
//5. 一個cpot數組,存每一列第一個每0元素在new_triple中的位置
//cpot[0] = 0;
for(i = 1; i < k; i++) {
cpot[i] = cpot[i-1] + num[i-1];
}
//6. 用一個for循環,利用cpot找到轉置後的的位置,然後行列交換就ok了
int col = 0;
int q = 0;
for( i = 0; i < k; i++) {
//每一行的元素都根據num數組,安排了足夠的空間
col = pre_Triple[i].j;
q = cpot[col];
new_Triple[q] = new Triple(pre_Triple[i].j,pre_Triple[i].i, pre_Triple[i].data);
cpot[col]++;
}
//7. 打印原二維數組、轉置前的三元組、轉置後的三元組
System.out.println("二維矩陣");
for(int p[] : arr) {
System.out.println(Arrays.toString(p));
}
System.out.println("轉置前的三元組");
for (i = 0; i < k; i++) {
System.out.println(pre_Triple[i].i + " " + pre_Triple[i].j + " " + pre_Triple[i].data);
}
System.out.println("轉置後的三元組");
for (i = 0; i < k; i++) {
System.out.println(new_Triple[i].i + " " + new_Triple[i].j + " " + new_Triple[i].data);
}
}
}
//1. 構造一個三元組,i,j,data
class Triple {
int i;//行
int j;//列
int data;//值
public Triple(int i, int j, int data) {
this.i = i;
this.j = j;
this.data = data;
}
}