关键在于怎么读数据
#include <iostream> #include <sstream> #include <string> #include <vector> #include <cstring> using namespace std; int n, m; int cost[10000]; vector<int> adj[10000]; int allCost; bool isVis[10000]; void getCost(int m) { for(vector<int>::iterator iter = adj[m].begin(); iter != adj[m].end(); iter++) { if(!isVis[*iter]) { isVis[*iter] = true; allCost += cost[*iter]; getCost(*iter); } } } int main() { string data; int num; while(cin >> n && n != 0) { cin >> m; memset(isVis, false, sizeof(isVis)); for(int i = 0; i < n; i++) adj[i].clear(); getline(cin, data); for(int i = 0; i < n; i++) { getline(cin, data); istringstream stream(data); stream >> num; cost[i] = num; while(stream >> num) adj[i].push_back(num - 1); } isVis[m - 1] = true; allCost = cost[m - 1]; getCost(m - 1); cout << allCost << endl; } return 0; }
