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