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--;
    }
}

 

转载于:https://www.cnblogs.com/lpomeloz/p/11359322.html