PAT甲级 1034 Head of a Gang （用邻接矩阵表示图，DFS）

原题链接：https://pintia.cn/problem-sets/994805342720868352/exam/problems/type/7?problemSetProblemId=994805456881434624&page=0

 问题：

最开始数组只开到1010，有一个测试点始终显示段错误

后修改为2010，AC，分析原因可能是因为N行输入，每行有两个姓名，可能的人数会加倍，所以需要开两倍空间，防止数组越界

代码：（C++）

#include <iostream>
#include <map>
#include <vector>
#include <utility>
#include <algorithm>
using namespace std;

#define endl '\n'

int N, K;
map<string, int> m;//将名字转换为编号
map<int, string> m2;//将编号转换成名字
bool arr[2010][2010];//存储两人之间的通话时间
int total[2010];//存储这个人的总通话时间
bool vis[2010];
vector<pair<string, int> > v;//存储输出
int cnt;//每个帮派的成员数目
int head;//存储头目的编号

int index = 0;

int totalSum = 0;
void DFS(int x) {
	vis[x] = true;
	cnt++;
	for (int i = 0; i < index; i++) {
		if (!vis[i] && arr[x][i]) {
			if (total[i] > total[head]) {
				head = i;
			}
			totalSum += total[i];
			DFS(i);
		}
	}
	return;
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

	//input
	cin >> N >> K;
	for (int i = 0; i < N; i++) {
		string n1, n2;
		int t;
		cin >> n1 >> n2 >> t;
		if (m.find(n1) == m.end()) {
			m[n1] = index;
			m2[index] = n1;
			index++;
		}
		if (m.find(n2) == m.end()) {
			m[n2] = index;
			m2[index] = n2;
			index++;
		}
		int i1 = m[n1], i2 = m[n2];
		arr[i1][i2] = true;
		arr[i2][i1] = true;
		total[i1] += t;
		total[i2] += t;
	}

	for (int i = 0; i < index; i++) {
		if (!vis[i]) {
			cnt = 0;
			head = i;
			totalSum = total[i];
			DFS(i);
			if (cnt > 2 && totalSum > K * 2) {
				v.emplace_back(make_pair(m2[head], cnt));
			}
		}
	}

	//ouput
	cout << v.size() << endl;
	sort(v.begin(), v.end());
	for (int i = 0; i < v.size(); i++) {
		cout << v[i].first << " " << v[i].second << endl;
	}

	return 0;
}