글 읽기 - 원소를 매개로, vector에서 지우고 싶습니다!

whazzulookinat 5년 전 0

제가 주석을 단 부분에서, (i,j)라는 pair을 vec에 삽입을 했습니다. 그리고 sorting과 여러 기타 프로세스 후, vec에서 (i,j)를 제거하고싶습니다. 그런데 sorting을 해버려서 주소는 찾기 힘들다고 생각해서, i와 j라는 값으로만 이들을 추적해서 지우는것 까지 가능할까요..? c++을 공부중이라서, 검색을 해보고 알아봤지만 시원한 답을 찾지못해 이곳에 남기게 되었습니다ㅠ

#include<iostream>
#include<utility>
#include<vector>
#include<algorithm>

using namespace std;


typedef struct node {
	int a;
	int b;
	struct node*next;
}Node;

Node*hashtable[9]; vector<pair<int, int>> vec; int n, m, h, a, b; pair<int, int> p; int cnt = 0;
int map[30][9] = { 0 }; int check = 0;

void hashh(int a, int b); void search(int k2); int add(int n, int k);
bool comp(pair<int, int>a, pair<int, int>b) {
	if (a.first > b.first) {
		return 1;
	}
	else {
		return 0;
	}
}

int main(void) {
	
	cin >> n >> m >> h;
	for (int i = 0; i < m; i++) {
		cin >> a >> b;
		map[a-1][b-1] = 1;
		vec.push_back(pair<int, int>(a, b));
	}
	//for (int i = 0; i < h; i++) {
	//	for (int j = 0; j < n - 1; j++) {
	//		cout << map[i][j];
	//	}
	//	cout << endl;
	//}
	//for (int i = 0; i <= 3; i++) {
	//	//cout << add(0, i);
	//	if (add(0, i) != -1) {
	//		cout << i;
	//		return 0;
	//	}
	//}
	//cout << -1;

	/*for (int i = 0; i <= 3; i++) {
		check = 0;
		if (add(0, i) != -1) {
			cout << add(0, i);
			return 0;
		}
	}
	cout << -1;*/
	//cout << add(0, 3);
	/*for (int i = 0; i < h; i++) {
		for (int j = 0; j < n - 1; j++) {
			cout << map[i][j] << ' ';
		}
		cout << endl;
	}*/
	//for (int i = 0; i < n; i++) {
	//	//cout << '?';
	//	search(i+1);
	//	cout << i + 1 << "==" << p.second << endl;
	//}
	add(0, 3);
	return 0;
}

int add(int s, int k) {
	if (s == k) {
		
		sort(vec.begin(), vec.end(), comp);
		for (int i = 0; i < m + k; i++) {
			hashh(vec[i].first, vec[i].second);
		}
		for (int i = 1; i < n; i++) {
			Node*ptrrr = hashtable[i];
			while (ptrrr != NULL) {
				cout << ptrrr->a << ' ' << ptrrr->b << endl;
				ptrrr = ptrrr->next;
			}
			cout << endl;
		}
		int answer = 0;
		
		for (int i = 0; i < n; i++) {
			//cout << '?';
			search(i+1);
			cout << endl;
			//cout << i+1<<"=="<<p.second << endl;
			if (i + 1 == p.second) {
				answer++;
			}
		}
		cout << endl;
		//cout <<n<<"==="<< answer << endl;
		if (answer == n) {
			check = 1;
			return k;
		}
		else {
			return -1;
		}
	}
	else {
		for (int i = 1; i <= h; i++) {
			for (int j = 1; j <= n - 1; j++) {
				if (map[i - 1][j - 1] == 0 && check==0){
					//cout << i-1 << j-1;
					if (((j - 2 >= 0 && map[i - 1][j - 2] == 0) || j - 2 < 0) && ((j < n - 1 && map[i - 1][j] == 0) || j >= n - 1)) {
						vec.push_back(make_pair(i, j)); //새로운 값을 삽입합니다.
						//cout << i - 1 << j - 1 << endl;
						map[i - 1][j - 1] = 1;
						for (int i = 0; i < h; i++) {
							for (int j = 0; j < n - 1; j++) {
								cout << map[i][j]<<' ';
							}
							cout << endl;
						}
						add(s + 1, k);
						map[i - 1][j - 1] = 0;
						vector<pair<int, int>>::iterator iter = find(vec.begin(), vec.end(), pair<int,int>(i,j)); //위에서 삽입한 값을 여기서 제거를 하고싶습니다!!!
						vec.erase(iter);
						//vec.pop_back();
					}
				}
			}
		}
		if (check == 1) {
			return k;
		}
		else {
			return -1;
		}

	}
}

void hashh(int a, int b) {
	Node*tmp = (Node*)malloc(sizeof(Node));
	tmp->a = a;
	tmp->b = b;
	if (hashtable[b] == NULL) {
		tmp->next = NULL;
		hashtable[b] = tmp;
	}
	else {
		tmp->next = hashtable[b];
		hashtable[b] = tmp;
	}
}

void search(int k2) {
	p = make_pair(0, k2);
	Node*ptr; Node*ptr2;
	while (p.first != h + 1) {
		//cout << p.first << ' ' << p.second << endl;
		if (p.second == 1) {
			ptr = hashtable[p.second];
			//cout << ptr->a;
			while (ptr != NULL) {
				if (ptr->a >= p.first) {
					p.first = ptr->a + 1;
					p.second = ptr->b + 1;
					break;
				}
				ptr = ptr->next;
			}
			if (ptr == NULL) {
				p.first = h + 1;
			}
		}
		else if (p.second == n) {
			ptr2 = hashtable[p.second - 1];
			while (ptr2 != NULL) {
				if (ptr2->a >= p.first) {
					p.first = ptr2->a + 1;
					p.second = ptr2->b;
					break;
				}
				ptr2 = ptr2->next;
			}
			if (ptr2 == NULL) {
				p.first = h + 1;
			}
		}
		else if (p.second > 1 && p.second < n) {
			ptr = hashtable[p.second];
			ptr2 = hashtable[p.second - 1];
			while (ptr != NULL) {
				if (ptr->a >= p.first) {
					break;
				}
				ptr = ptr->next;
			}
			while (ptr2 != NULL) {
				if (ptr2->a >= p.first) {
					break;
				}
				ptr2 = ptr2->next;
			}
			if (ptr == NULL && ptr2 == NULL) {
				p.first = h + 1;
			}
			else if (ptr == NULL && ptr2 != NULL) {
				p.first = ptr2->a + 1;
				p.second = ptr2->b;
			}
			else if (ptr != NULL && ptr2 == NULL) {
				p.first = ptr->a + 1;
				p.second = ptr->b + 1;
			}
			else if (ptr != NULL && ptr2 != NULL) {
				if (ptr->a < ptr2->a) {
					p.first = ptr->a + 1;
					p.second = ptr->b + 1;
				}
				else {
					p.first = ptr2->a + 1;
					p.second = ptr2->b;
				}
			}
		}
	}
	//cout << p.first << ' ' << p.second << endl;
}

yukariko 5년 전 0

vector안의 element를 찾아서 삭제는 가능합니다.

vector에 erase 멤버함수를 검색해보세요.

whazzulookinat 5년 전 0

답변감사합니다. 혹시 찾는 것은 find로 하면 될까요??

yukariko 5년 전 0

네 find 로 가능합니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

원소를 매개로, vector에서 지우고 싶습니다!

whazzulookinat 5년 전 0

yukariko 5년 전 0

whazzulookinat 5년 전 0

yukariko 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말