Question

Based on the Inheritance question, modify your program (class A particularly) to where Polymorphism can properly occur. Ensure that your overridden function(s) is(are) virtual. In main, Create two pointer objects of type A. The first one should create a new A, the second should create a new B object. The parameter values may be arbitrary.

Call the print function for each of the objects in main.

Answer 1

#include <iostream>

using namespace std;

class A {
    int a;
    int b;
public:
    A(int a, int b) {
        setA(a);
        setB(b);
    }

    void setA(int a) { this->a = a; }

    void setB(int b) { this->b = b; }

    int getA() { return a; }

    int getB() { return b; }

    virtual int calcValue() { return a * b / 10; }

    virtual void print() { cout << "Result is " << calcValue() << endl; }
};

class B : public A {
public:
    B(int a, int b) : A(a, b) {}

    int calcValue() {
        return (getA() / getB()) * 10;
    }

    void print() {
        cout << "Result from B is " << calcValue() << endl;
    }
};

int main() {
    A *a = new A(2, 7);
    A *b = new B(4, 9);
    a->print();
    b->print();
    return 0;
}