20190814-華為筆試--找到用戶i的所有n度好友 C++

1.題目描述：

　　某軟件用戶共m個人，編號為0~m-1，用r[i][j]表示用戶i和j的好友關系，r[i][j]=0說明不是好友，r[i][j]=1~9數值越大關系越好。其次，r[i][j] = r[j][i]。找出i的n度好友。若不存在n度好友則輸出-1.

　　1度好友代表直接好友，2度好友代表好友的好友，，。按照推薦值降序輸出好友列表。即r[i][j] = 6,r[j][k]=4,則i的2度好友推薦值為r[i][j]+r[j][k]=10.

　　輸入：第一行T表示T組測試數據；

　　　　　第二行首先是整數k，接着3*k個整數，用空格隔開，形成<x,y,r[x][y]>的關系對。

　　　　示例：2

　　　　　　　10 5 2

　　　　　　　13 0 3 5 0 4 9 0 6 8 0 7 5 1 2 6 1 6 3 2 9 7 3 4 3 3 5 3 3 8 3 3 9 3 5 8 9 7 8 9　

　　　　　　　10 0 2

　　　　　　　13 0 3 5 0 4 9 0 6 8 0 7 5 1 2 6 1 6 3 2 9 7 3 4 3 3 5 3 3 8 3 3 9 3 5 8 9 7 8 9

　　輸出：輸出T行，每行對應每組測試數據 i的n度好友的降序輸出。

　　　　示例：7 0 4 9 

　　　　　　　1 5 8 9

2.理解：

　　使用dp[i][j]記錄用戶i到達用戶i的j+1度好友推薦值。已知用戶i的直接好友，依次找到它的n度好友。

　　初始化：利用給定的r[i][j]的值初始化dp數組，初始化1度好友值。

　　迭代遍歷r二維數組，當r中存在i的t度好友的直接好友時，該好友為i的t+1度好友。但該用戶要滿足：未曾出現在用戶i的好友列表中，且不為用戶i本身。

3.注意事項：

　　由於r[i][j] = r[j][i]，所以遍歷r數組，應遍歷vr[0]->vr[1]及vr[1]->vr[0]的雙向關系。

　　解決 新好友未曾出現在用戶i的好友列表中：利用set集合保存用戶i的好友列表。

　　dp[i][j]保存的好友推薦值，而排序時會改變好友編號的位置，需要用數組保存與dp對應的用戶編號，最終降序輸出用戶編號。

4.代碼：

#include "pch.h"
#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;

vector<int> getFriend(int m,int i,int n,vector<vector<int>> r) {
    vector<vector<int>> dp(m,vector<int>(n,0)); // n度好友好感值
    vector<int> res;
    vector<int> t;
    set<int> s;
    t.resize(m);
    for (int j = 0;j < m;j++) {
        t[j] = j;
    }

    for (auto rv : r) {
        if (rv[0] == i)
        {
            dp[rv[1]][0] = rv[2];    // 1度好友
            s.insert(rv[1]);
        }
    }
    /*cout << "初始化：" << endl;
    for (int j = 0;j < m;++j) {
        cout << j << ":" << dp[j][0] << endl;
    }*/

    for (int t = 1;t < n;++t) {
        for (auto rv : r) {
            //cout << rv[0] << ":" << dp[rv[0]][t - 1] << endl;
            if (dp[rv[0]][t-1] != 0 && rv[1] != i && s.find(rv[1])==s.end()) {
                
                dp[rv[1]][t] = dp[rv[0]][t-1] + rv[2];
                s.insert(rv[1]);
                //cout << rv[1] << ":" << dp[rv[1]][t] << endl;
            }
        }
    }
    for (int j = 0;j < m;j++) {
        for (int q = j + 1;q < m;++q) {
            if (dp[j][n - 1] < dp[q][n - 1]) {
                int tmp = dp[j][n - 1];
                dp[j][n - 1] = dp[q][n - 1];
                dp[q][n - 1] = tmp;
                tmp = t[j];
                t[j] = t[q];
                t[q] = tmp;
            }
        }
        
    }
    for (int j = 0;j < m;j++) {
        if (dp[j][n - 1] > 0) {
            res.push_back(t[j]);
        }
    }
    return res;
}

int main() {
    int T=1;
    while (T) {
        int m, i, n;
        m = 10, i = 0, n = 2;
        vector<vector<int>> r = { { 0, 3, 5} ,{ 0 ,4 ,9} ,{ 0, 6 ,8}, { 0, 7 ,5}, { 1 ,2, 6} ,{ 1, 6, 3}, { 2 ,9, 7}, { 3 ,4, 3}, { 3, 5, 3}, { 3 ,8, 3} ,{ 3, 9 ,3} ,{ 5, 8, 9} ,{ 7, 8 ,9 } };
        vector<vector<int>> rFri;
        for (auto rv : r) {
            vector<int> tmp;
            tmp.push_back(rv[1]);
            tmp.push_back(rv[0]);
            tmp.push_back(rv[2]);
            rFri.push_back(rv);
            rFri.push_back(tmp);
        }
        vector<int> res = getFriend( m, i, n, rFri);
        if (res.size() == 0)
            cout << "-1" << endl;
        else {
            for (auto v : res) {
                cout << v << " ";
            }
            cout << endl;
        }
        T--;
    }
}