Problem:
vector를 이용하여 아래 Circle 클래스의 객체를 삽입하고 삭제하는 프로그램을 작성하라.
삭제 시에는 이름이 같은 모든 원을 삭제한다.
|
|
| 1 2 3 4 5 6 7 8 9 10 |
class Circle { string name; // 이름 int radius; // 반지름 public: Circle(int radius, string name) { this->radius = radius; this->name = name; } double getArea() { return 3.14*radius*radius; } string getName() { return name; } }; |
Execution Result:
Objective & Hints:
객체 포인터를 vector에 삽입, 삭제 연습
Code:
|
|
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
#include #include using namespace std; class Circle { string name; // 이름 int radius; // 반지름 public: Circle(int radius, string name) { this->radius = radius; this->name = name; } double getArea() { return 3.14*radius*radius; } string getName() { return name; } }; int main() { cout << "원을 삽입하고 삭제하는 프로그램입니다.\n"; vector<Circle*> v; while(true){ int num; int radius; string name; cout << "삽입:1, 삭제:2, 모두보기:3, 종료:4 >> "; cin >> num; switch (num){ case 1: cout << "생성하고자 하는 원의 반지름과 이름은 >> "; cin >> radius >> name; v.push_back(new Circle(radius, name)); break; case 2:{ cout << "삭제하고자 하는 원의 이름은 >> "; cin >> name; vector<Circle*>::iterator it = v.begin(); while(it != v.end()){ Circle *p = *it; if(p->getName() == name){ it = v.erase(it); delete p; } else it++; } break; } case 3:{ vector<Circle*>::iterator it; for(it=v.begin(); it!=v.end(); it++){ Circle *p = *it; cout << p->getName() << endl; } cout << endl; break; } case 4: return 0; } } } |
Explanation:
vector<Circle*> v; 형식으로 벡터를 생성합니다.
v.erase(it)는 벡터 v에서 it가 가리키는 원소를 삭제하고 난 다음, 벡터 내에 삭제된 다음 지점의 첫 원소에 대한 포인터를 리턴합니다.
그러므로 원소를 삭제한 후 다음 원소의 주소를 잡으려면 간단히 다음과 같이 하면 됩니다.