PollPost.java

import java.util.HashMap;
import java.util.Map;

public class PollPost extends Post {
    private Map<String, Integer> options = new HashMap<>();

    public PollPost(String title, String optionsString, User user) {
        super(title, user);
        for (String option : optionsString.split(";")) {
            option = option.trim();
            options.put(option, 0);
        }
    }

    @Override
    public void display() {
        System.out.println("Post #" + getId());
        System.out.println("Created By: " + user.getFullName() + " (@" + user.getUsername() + ")");
        System.out.println("Title: " + title);
        for (Map.Entry<String, Integer> entry : options.entrySet()) {
            System.out.println(entry.getKey() + ": " + entry.getValue());
        }
        System.out.println();
    }

    public void vote(int optionIndex) {
        String[] optionKeys = options.keySet().toArray(new String[0]);
        if (optionIndex >= 0 && optionIndex < optionKeys.length) {
            String selectedOption = optionKeys[optionIndex];
            options.put(selectedOption, options.get(selectedOption) + 1);
        } else {
            System.out.println("Invalid option index.");
        }
    }

    public String[] getOptions() {
        return options.keySet().toArray(new String[0]);
    }
}