// main.cpp
#include "Oswietlenie.h"
#include "Pokoj.h"

int main()
{
    Oswietlenie lampka1;
    Oswietlenie lampka2;
    Oswietlenie lampka3;
    lampka1.wypisz_stan();
    lampka2.wypisz_stan();
    lampka3.wypisz_stan();
    lampka1.wlacz();
    lampka1.wypisz_stan();
    lampka2.wlacz();
    lampka2.wypisz_stan();
    Pokoj nowy_pokoj(&lampka1, &lampka2, &lampka3);
    nowy_pokoj.pokaz_stan();
    nowy_pokoj.wylacz_wszystkie();
    nowy_pokoj.pokaz_stan();
    lampka1.wypisz_stan();
    return 0;
}

// head.h
#ifndef HEAD_H_INCLUDED
#define HEAD_H_INCLUDED

class Pokoj;
class Oswietlenie;

#endif // HEAD_H_INCLUDED

// Pokoj.h
#ifndef POKOJ_H
#define POKOJ_H
#include "Oswietlenie.h"

class Pokoj
{
    friend class Oswietlenie;
    Oswietlenie* lampki[3];
public:
    Pokoj(Oswietlenie*, Oswietlenie*, Oswietlenie*);
    void wlacz_wszystkie();
    void wylacz_wszystkie();
    void pokaz_stan();
};

#endif // POKOJ_H

// Pokoj.cpp
#include "Pokoj.h"
#include <iostream>

using namespace std;

Pokoj::Pokoj(Oswietlenie* l1, Oswietlenie* l2, Oswietlenie* l3)
{
    lampki[0] = l1;
    lampki[1] = l2;
    lampki[2] = l3;
}

void Pokoj::wlacz_wszystkie()
{
    for(int i = 0; i < 3; ++i)
        lampki[i]->wlacz();
}

void Pokoj::wylacz_wszystkie()
{
    for(int i = 0; i < 3; ++i)
        lampki[i]->wylacz();
}

void Pokoj::pokaz_stan()
{
    if(lampki[0]->stan == true && lampki[1]->stan == true && lampki[2]->stan == true)
    {
        cout << "Wszystkie swiatla w pokoju sa wlaczone\n";
        return;
    }
    if(lampki[0]->stan == false && lampki[1]->stan == false && lampki[2]->stan == false)
    {
        cout << "Wszystkie swiatla w pokoju sa wylaczone\n";
        return;
    }
    cout << "W pokoju wlaczone sa swiatla nr ";
    for(int i = 0; i < 3; ++i)
        if(lampki[i]->stan == true)
            cout << i + 1 << ", ";
}

// Oswietlenie.h
#ifndef OSWIETLENIE_H
#define OSWIETLENIE_H
#include "../head.h"

class Oswietlenie
{
    friend class Pokoj;
    bool stan;
public:
    Oswietlenie() : stan(false) {}
    void wlacz() {stan = true;}
    void wylacz() {stan = false;}
    void wypisz_stan();
};

#endif // OSWIETLENIE_H

// Oswietlenie.cpp
#include "Oswietlenie.h"
#include <iostream>

using namespace std;

void Oswietlenie::wypisz_stan()
{
    if(stan)
        cout << "Lampka wlaczona\n";
    else
        cout << "Lampka wylaczona\n";
}