글 읽기 - 시간초과입니다.

gusdnr9875 5년 전 0

ㄴ나름 합리적으로 접근한거 같은데 시간초과가 되었습니다. 어떤 방식을 접근해야 좋을까요?

#include <iostream>
using namespace std;

int arr[100001];  //트리의 관계를 나타내는 배열 
int arr2[100001] = { 0 };  // 점접의 가중치값을 저장할 배열 

int x = 0; 

int carculation_1(int i) {
	if (arr[i] == x&& i!=x) return 1;
	else if (arr[i] == i) return 0;
	else  return carculation_1(arr[i]);
}

void unioned(int a, int b) {
	if (a < b) arr[b] = a;
	else arr[a] = b;
}

int main() {
	int n = 0, m = 0, a = 0, b = 0;
	int input = 0, y = 0, result = 0;

	cin >> n >> m;

	for (int i = 1; i <= n; i++)     //처음에는 모두 자기 자신을 꼭대기니까 
		arr[i] = i;

	for (int i = 1; i <= n - 1; i++) { //접점들을 연결시킴 
		cin >> a >> b;
		unioned(a, b);
	}
	for (int i = 1; i <= n; i++)  //접점의 가중치를 저장함 
		cin >> arr2[i];

	while (m--) {
		cin >> input;
		
		if (input == 1) {
			cin >> x;
			result = arr2[x];
			for (int i = 1; i <= n; i++) 
				if (carculation_1(i)) 
					result = result^arr2[i];
			cout << result << endl;
		}
		else if (input == 2) {
			cin >> x >> y;
			arr2[x] =arr2[x]^y;
			for (int i = 1; i <= n; i++)
				if (carculation_1(i))
					arr2[i] =arr2[i]^y;
		}

	}

}