[백준 C++] 13504 XOR 합

이성훈·2022년 11월 6일
0

백준(Baekjoon online judge)

목록 보기
147/177

문제

N개의 수로 이루어진 수열 A가 주어진다.

수열 A에서 연속된 부분 수열을 고르려고 한다. 부분 수열의 XOR 합이란, 부분 수열에 들어있는 모든 원소를 XOR한 값을 의미한다.

수열 A가 주어졌을 때, XOR 합이 가장 큰 부분 수열을 찾는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. (1 ≤ T ≤ 10)

각 테스트 케이스의 첫째 줄에는 배열의 크기 N (1 ≤ N ≤ 100,000), 둘째 줄에는 수열 A에 들어있는 수가 주어진다. 수열 A에 들어있는 수는 32비트 부호있는 정수 범위 안에 들어가는 음이 아닌 정수이다.

출력

각각의 테스트 케이스마다 수열 A의 연속된 부분 수열 중에서 XOR 합이 가장 큰 부분 수열의 XOR 합을 출력한다.

https://www.acmicpc.net/problem/13504

풀이

V[i]를 0부터 인덱스i를 맨오른쪽 인덱스로한 부분 수열내의 모든 원소의 XOR합이라고 할때
V[0], V[1], V[2],...V[n]을 모두 역순으로 Trie구조에 넣은뒤 두 인덱스 i, j의 V[i] XOR V[j]의 최댓값을 찾아 출력하면된다.

예로들어 주어지는 수열이 {A, B, C}면
0 XOR A, A XOR B, A XOR B XOR C를 구하고,
이값들을 V[0], V[1], V[2]로 중복없이 Trie구조에 2진수로 변환하여 역순으로 넣는다.
그리고 max를 0 XOR V[i] 의 최댓값으로 만든뒤
모든 V[i]와 XOR결과가 최대인 2진수를 Trie구조에서 찾아서 이것을 정수로변환, max보다 크면 max를 갱신한다.

2진수로변환해서 역순으로 넣는과정과, XOR결과가 최댓값을 찾아가는 과정은 https://velog.io/@cldhfleks2/13505
이곳에 13505번문제를 풀며 자세히 기록해놓았다.

#define _CRT_SECURE_NO_WARNINGS 
#include <bits/stdc++.h>
#define mp std::make_pair 
#define mt std::make_tuple
#define dq std::deque
#define pq std::priority_queue
#define sw std::swap
#define ts(x) std::to_string(x)
#define tc() c_str()
#define sc(x, ...) scanf(x, ##__VA_ARGS__) 
#define pr(x, ...) printf(x, ##__VA_ARGS__) 
#define ins(x) insert(x)
#define pb(x) push_back(x)
#define pf(x) push_front(x)
#define PB() pop_back()
#define PF() pop_front()
#define ph(x) push(x)
#define TT() top()
#define PP() pop()
#define BB() back()
#define FF() front()
#define cls() clear()
#define emp() empty()
#define len(x) x.length()
#define sz(x) ((int)x.size()) //컨테이너에서 사용
#define ms(a) memset(a, 0, sizeof(a)) //0으로 초기화
#define rep(i, n) for(int i = 0; i < n ; i++)
#define rrep(i, r, n) for(int i = r; i < n ; i++)
#define rrrep(i, r, n) for(ll i = r; i < n ; i++)
#define _rrep(i, r, n) for(int i = r; i >= n; i--)
#define _rrrep(i, r, n) for(ll i = r; i >= n; i--)
#define each(x, a) for (auto& x: a)
#define all(x) x.begin(),x.end() //STL에서 전체 처리할때 사용
#define range(x, r, n) x.begin() + r, x.begin() + n //STL에서 구간설정
#define ct continue
#define br break
#define rt return
#define _TYF typedef //코드줄이기
#define _UG using
#define _TCE template <class T> inline
//#define MAX 
const int IMAX = INT32_MAX; const int IMIN = INT32_MIN;
const long long LMAX = LLONG_MAX; const long long LMIN = LLONG_MIN;
const long double PI = 3.141592653589793238462643383279502884197;
_UG std::vector; _UG std::stack; _UG std::queue; _UG std::tuple; _UG std::set; _UG std::list; _UG std::bitset; _UG std::string; _UG std::pair; _UG std::greater;
_UG std::tie; _UG std::sort; _UG std::max_element; _UG std::min_element; _UG std::fill; _UG std::stoi; _UG std::stod; _UG std::stof; _UG std::stol; _UG std::stold; _UG std::stoll; _UG std::stoul; _UG std::stoull; _UG std::to_string;
//_UG std::max; //_UG std::min; //_UG std::map;
_TYF long long ll; _TYF unsigned long long ull;
_TYF pair<int, int> pii; _TYF pair<double, int> pdi; _TYF pair<int, double> pid; _TYF pair<double, double> pdd; _TYF pair<int, ll> pil;
_TYF pair<ll, int> pli; _TYF pair<ll, ll> pll; _TYF pair<ull, ull> pullull; _TYF pair<int, char> pic; _TYF pair<char, int> pci;
_TYF pair<char, char> pcc; _TYF pair<long, char> plc; _TYF pair<char, long> pcl; _TYF pair<ll, char> pllc; _TYF pair<char, ll> pcll;
_TYF pair<ull, char> pullc; _TYF pair<char, ull> pcull; _TYF pair<int, string> pis; _TYF pair<string, int> psi; _TYF pair<long, string> pls;
_TYF pair<string, long> psl; _TYF pair<ll, string> plls; _TYF pair<string, ll> psll; _TYF pair<ull, string> pulls;
_TYF pair<string, ull> psull; _TYF pair<string, string> pss;
_TYF tuple<int, int, int> tiii; _TYF tuple<int, int, int, int> tiiii;
_TYF tuple<ll, ll, ll> tlll; _TYF tuple<ll, ll, ll, ll> tllll;
_TYF vector<string> vs; _TYF queue<string> qs; _TYF stack<string> ss; _TYF dq<string> dqs; _TYF pq<string> pqs; _TYF dq<string> dqs;
_TYF vector<char> vc; _TYF queue<char> qc; _TYF stack<char> sc; _TYF dq<char> dqc; _TYF pq<char> pqc; _TYF dq<char> dqc;
_TYF vector<int> vi; _TYF queue<int> qi; _TYF stack<int> si; _TYF dq<int> dqi; _TYF pq<int> pqi; _TYF dq<int> dqi;
_TYF vector<pii> vii; _TYF queue<pii> qii; _TYF stack<pii> sii; _TYF dq<pii> dqii; _TYF pq<pii> pqii; _TYF dq<pii> dqii;
_TYF vector<tiii> viii; _TYF queue<tiii> qiii; _TYF stack<tiii> siii; _TYF dq<tiii> dqiii; _TYF pq<tiii> pqiii; _TYF dq<tiii> dqiii;
_TYF vector<tiiii> viiii; _TYF queue<tiiii> qiiii; _TYF stack<tiiii> siiii; _TYF dq<tiiii> dqiiii; _TYF pq<tiiii> pqiiii; _TYF dq<tiiii> dqiiii;
_TYF vector<pll> vll; _TYF queue<pll> qll; _TYF stack<ll> sll; _TYF dq<pll> dqll; _TYF pq<pll> pqll; _TYF dq<pll> dqll;
_TYF vector<tlll> vlll; _TYF queue<tlll> qlll; _TYF stack<tlll> slll; _TYF dq<tlll> dqlll; _TYF pq<tlll> pqlll; _TYF dq<tlll> dqlll;
_TYF vector<tllll> vllll; _TYF queue<tllll> qllll; _TYF stack<tllll> sllll; _TYF dq<tllll> dqllll; _TYF pq<tllll> pqllll; _TYF dq<tllll> dqllll;
_TCE T sq(T num) { rt num* num; }//제곱함수
_TCE T GCD(T num1, T num2) { if (num2 == 0) rt num1; rt GCD(num2, num1 % num2); }
_TCE T LCM(T num1, T num2) { if (num1 == 0 || num2 == 0) rt num1 + num2; rt num1* (num2 / GCD(num1, num2)); }
//STL 전용 초기화 함수들 ( ms~~ )
_TCE void msq(T& a) { while (!a.empty()) a.PP(); }//queue clear
_TCE void msv(T& a) { a.cls(); }//vector clear
_TCE void msdq(T& a) { a.cls(); }//deque clear
_TCE void msm(T& a) { a.cls(); }//map clear
_TCE void mss(T& a) { while (!a.empty()) a.PP(); }//stack, set clear
_TCE void mspq(T& a) { while (!a.empty()) a.PP(); }//priority_queue clear
//pii operator - (pii a, pii b) { rt pii(a.first - b.first, a.second - b.second); }
//bool operator <= (pii a, pii b) { rt a.first <= b.first && a.second <= b.second; } 
//bool operator >= (pii a, pii b) { rt a.first >= b.first && a.second >= b.second; } 
//bool operator < (pii a, pii b) { if (a == b) return false; rt a <= b; } 
//bool operator > (pii a, pii b) { if (a == b) return false; rt a >= b; }

int T, N;
int binary[32]; //맨끝은 -1저장
struct xorTrie {
	xorTrie* node[2];
	xorTrie() : node() {}

	~xorTrie() {
		if (node[0]) node[0];
		if (node[1]) node[1];
	}

	void insert(int* binary) {
		if (*binary == -1) rt;

		if (node[*binary] == nullptr)
			node[*binary] = new xorTrie();
		node[*binary]->insert(binary + 1);
	}

	bool find(int* binary) {
		if (*binary == -1) //끝에도달시 존재하는것
			rt true;
		if (node[*binary] == nullptr) //중간에 끊기면 존재안하는것
			rt false;
		rt node[*binary]->find(binary + 1);
	}

	void maxXOR(int* binary) {
		if (*binary == -1)
			rt;
		int bit = *binary;
		if (node[bit ^ 1]) {
			*binary = 1;
			node[bit ^ 1]->maxXOR(binary + 1);
		}
		else {
			*binary = 0;
			node[bit]->maxXOR(binary + 1);
		}
	}
};

//프로그램 메인 로직
void func() {
	sc("%d", &T);
	while (T--) {
		xorTrie trie;
		sc("%d", &N);
		vi XOR(N); //0~i인덱스까지의 부분수열의 XOR결과를 저장

		int max = 0; //부분수열의 XOR값중 최댓값을 저장
		int prevXOR = 0;
		//0번째부터 N번째까지의 순차적인 부분XOR결과를 중복없이 trie에 저장
		rep(k, N) {
			int number;
			sc("%d", &number);
			XOR[k] = prevXOR ^ number;
			prevXOR = XOR[k];

			int curXOR = XOR[k];
			//지금까지의 부분XOR결과를 2진수binary에 역순으로 저장
			_rrep(i, 30, 0) binary[i] = 0;
			_rrep(i, 30, 0) {
				binary[i] = curXOR & 1;
				curXOR >>= 1;
			}
			binary[31] = -1;

			if (!trie.find(binary)) {
				trie.insert(binary);
				max = std::max(max, XOR[k]);
			}

		}

		//누적XOR값들과 XOR가 최대가되는 trie내의 2진수를 찾은뒤
		//이값들 간의 최댓값을 max(정수로변환해서)에 저장
		rep(k, N) {
			int cur = XOR[k];

			//정수cur을 2진수 binary로 변환
			_rrep(i, 30, 0) binary[i] = 0;
			_rrep(i, 30, 0) {
				binary[i] = cur & 1;
				cur >>= 1;
			}
			binary[31] = -1;

			//누적XOR값과 지금껏 trie에있는 값중 최대 XOR이되는 값을 찾음
			trie.maxXOR(binary);

			int num = 0;
			int digit = 1;
			_rrep(i, 30, 0) {
				if (binary[i] == 1) num |= digit;
				digit <<= 1;
			}

			max = std::max(max, num);
		}

		pr("%d\n", max);

		trie.~xorTrie(); //메모리 해제
	}
}

int main(void) {
	func();

	rt 0;
}
profile
I will be a socially developer

0개의 댓글