寶石比較重量

寶石比較重量

標籤（空格分隔）： 算法

---

題目描述：

小明陪小紅去看鑽石，他們從一堆鑽石中隨機抽取兩顆並比較她們的重量。這些鑽石的重量各不相同。在他們們比較了一段時間後，它們看中了兩顆鑽石g1和g2。現在請你根據之前比較的信息判斷這兩顆鑽石的哪顆更重。
給定兩顆鑽石的編號g1,g2，編號從1開始，同時給定關係數組vector,其中元素爲一些二元組，第一個元素爲一次比較中較重的鑽石的編號，第二個元素爲較輕的鑽石的編號。最後給定之前的比較次數n。請返回這兩顆鑽石的關係，若g1更重返回1，g2更重返回-1，無法判斷返回0。輸入數據保證合法，不會有矛盾情況出現。
測試樣例：
2,3,[[1,2],[2,4],[1,3],[4,3]],4
返回: 1

思路：
兩個元素之間有很明確的關係。如果i > j,則一定有j < i，所以可以看做是一個對稱的圖。
構建這個圖:

• 若a[i][j] = 1,表示i比j大；
• 若a[i][j] = -1,表示i比j小；
• 若a[i][j] = 0,表示i與j無法比較；

代碼爲：

import java.util.*;
public class Cmp {
public int cmp(int g1, int g2, int[][] records, int n) {
    // write code here
    int[][] graph = new int[100][100];
    for(int i=0; i<n; i++) {
        int a = records[i][0] - 1;
        int b = records[i][1] - 1;
        graph[a][b] = 1;
        graph[b][a] = -1;

        for(int j=0; j<100; j++) {
            if(graph[j][a] == 1) { //若j比a大，則j一定比b大
                graph[j][b] = 1;
                graph[b][j] = -1;
            }
            if(graph[b][j] == 1) { //若b比j大, 則j一定比a小
                graph[j][a] = -1;
                graph[a][j] = 1;
            }
        }

    }
    return graph[g1-1][g2-1];

  }
}