形状工厂-lintcode

C++代码：

/** * Your object will be instantiated and called as such: * ShapeFactory* sf = new ShapeFactory(); * Shape* shape = sf->getShape(shapeType); * shape->draw(); */class Shape {public:    virtual void draw() const = 0;};class Rectangle: public Shape {public:        void draw() const {        cout<<" ----"<<endl;        cout<<"|    |"<<endl;        cout<<" ----"<<endl;    }};class Square: public Shape {public:    void draw() const{        cout << " ---- " << endl;        cout<<"|    |"<<endl;        cout<<"|    |"<<endl;        cout<<" ---- "<<endl;    }};class Triangle: public Shape {    void draw() const {        cout<<"  /\\  "<<endl;        cout<<" /  \\ "<<endl;        cout<<"/____\\"<<endl;    }};class ShapeFactory {public:    /**     * @param shapeType a string     * @return Get object of type Shape     */    Shape* getShape(string& shapeType) {        if (shapeType == "Square") {            return new Square();          }else if (shapeType == "Triangle") {            return new Triangle();        }else if (shapeType == "Rectangle") {            return new Rectangle();        }    }};

python代码：

"""Your object will be instantiated and called as such:sf = ShapeFactory()shape = sf.getShape(shapeType)shape.draw()"""class Shape:    def draw(self):        raise NotImplementedError('This method should have implemented.')class Triangle(Shape):    def draw(self):        print '  /\\  '        print ' /  \\ '        print '/____\\'class Rectangle(Shape):    def draw(self):        print ' ---- '        print r'|    |'        print r' ---- 'class Square(Shape):    def draw(self):        print ' ---- '        print '|    |'        print '|    |'        print ' ---- 'class ShapeFactory:    # @param {string} shapeType a string    # @return {Shape} Get object of type Shape    def getShape(self, shapeType):        if shapeType == 'Triangle':            return Triangle()        if shapeType == 'Rectangle':            return Rectangle()        if shapeType == 'Square':            return Square()