diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..daba299 --- /dev/null +++ b/.gitignore @@ -0,0 +1,4 @@ +src/agent.py +src/agent_runner.py +.checkpoints +.observation diff --git a/main.js b/main.js index 01f873f..31c1802 100644 --- a/main.js +++ b/main.js @@ -5,20 +5,201 @@ export class ZebraPuzzle { constructor() { - throw new Error('Remove this line and implement the function'); + this.houses = [1, 2, 3, 4, 5]; + this.nationalities = ['Englishman', 'Spaniard', 'Ukrainian', 'Norwegian', 'Japanese']; + this.colors = ['Red', 'Green', 'Ivory', 'Yellow', 'Blue']; + this.pets = ['Dog', 'Snails', 'Fox', 'Horse', 'Zebra']; + this.drinks = ['Coffee', 'Tea', 'Milk', 'OrangeJuice', 'Water']; + this.hobbies = ['Dancing', 'Painting', 'Reading', 'Football', 'Chess']; + + // Solve the puzzle during construction + this.solution = this.solve(); } /** * @return {string} */ waterDrinker() { - throw new Error('Remove this line and implement the function'); + return this.solution.waterDrinker; } /** * @return {string} */ zebraOwner() { - throw new Error('Remove this line and implement the function'); + return this.solution.zebraOwner; + } + + // Generate all permutations of an array + permute(arr) { + if (arr.length <= 1) return [arr]; + + const result = []; + for (let i = 0; i < arr.length; i++) { + const rest = this.permute(arr.slice(0, i).concat(arr.slice(i + 1))); + for (let j = 0; j < rest.length; j++) { + result.push([arr[i]].concat(rest[j])); + } + } + return result; + } + + // Check if two houses are next to each other + areNextTo(pos1, pos2) { + return Math.abs(pos1 - pos2) === 1; + } + + // Solve the zebra puzzle using constraint satisfaction + solve() { + // Generate all possible permutations for each category + const colorPerms = this.permute(this.colors); + const nationalityPerms = this.permute(this.nationalities); + const petPerms = this.permute(this.pets); + const drinkPerms = this.permute(this.drinks); + const hobbyPerms = this.permute(this.hobbies); + + // Try all combinations of permutations + for (const colors of colorPerms) { + // Constraint 6: The green house is immediately to the right of the ivory house + let ivoryPos = -1; + let greenPos = -1; + for (let i = 0; i < 5; i++) { + if (colors[i] === 'Ivory') ivoryPos = i; + if (colors[i] === 'Green') greenPos = i; + } + if (ivoryPos === -1 || greenPos === -1 || greenPos !== ivoryPos + 1) continue; + + // Constraint 15: The Norwegian lives next to the blue house + // We'll check this later when we have nationalities + + for (const nationalities of nationalityPerms) { + // Constraint 10: The Norwegian lives in the first house + if (nationalities[0] !== 'Norwegian') continue; + + // Constraint 15: The Norwegian lives next to the blue house + let bluePos = -1; + for (let i = 0; i < 5; i++) { + if (colors[i] === 'Blue') bluePos = i; + } + if (bluePos === -1 || !this.areNextTo(0, bluePos)) continue; + + for (const pets of petPerms) { + // Constraint 3: The Spaniard owns the dog + let spaniardPos = -1; + let dogPos = -1; + for (let i = 0; i < 5; i++) { + if (nationalities[i] === 'Spaniard') spaniardPos = i; + if (pets[i] === 'Dog') dogPos = i; + } + if (spaniardPos === -1 || dogPos === -1 || spaniardPos !== dogPos) continue; + + for (const drinks of drinkPerms) { + // Constraint 9: The person in the middle house drinks milk + if (drinks[2] !== 'Milk') continue; + + // Constraint 4: The person in the green house drinks coffee + if (drinks[greenPos] !== 'Coffee') continue; + + // Constraint 5: The Ukrainian drinks tea + let ukrainianPos = -1; + let teaPos = -1; + for (let i = 0; i < 5; i++) { + if (nationalities[i] === 'Ukrainian') ukrainianPos = i; + if (drinks[i] === 'Tea') teaPos = i; + } + if (ukrainianPos === -1 || teaPos === -1 || ukrainianPos !== teaPos) continue; + + // Constraint 13: The person who plays football drinks orange juice + let footballPos = -1; + let orangeJuicePos = -1; + for (let i = 0; i < 5; i++) { + if (drinks[i] === 'OrangeJuice') orangeJuicePos = i; + } + // We'll find football position later + + for (const hobbies of hobbyPerms) { + // Find football position + for (let i = 0; i < 5; i++) { + if (hobbies[i] === 'Football') footballPos = i; + } + if (footballPos === -1 || orangeJuicePos === -1 || footballPos !== orangeJuicePos) continue; + + // Constraint 7: The snail owner likes to go dancing + let snailPos = -1; + let dancingPos = -1; + for (let i = 0; i < 5; i++) { + if (pets[i] === 'Snails') snailPos = i; + if (hobbies[i] === 'Dancing') dancingPos = i; + } + if (snailPos === -1 || dancingPos === -1 || snailPos !== dancingPos) continue; + + // Constraint 8: The person in the yellow house is a painter + let yellowPos = -1; + let paintingPos = -1; + for (let i = 0; i < 5; i++) { + if (colors[i] === 'Yellow') yellowPos = i; + if (hobbies[i] === 'Painting') paintingPos = i; + } + if (yellowPos === -1 || paintingPos === -1 || yellowPos !== paintingPos) continue; + + // Constraint 14: The Japanese person plays chess + let japanesePos = -1; + let chessPos = -1; + for (let i = 0; i < 5; i++) { + if (nationalities[i] === 'Japanese') japanesePos = i; + if (hobbies[i] === 'Chess') chessPos = i; + } + if (japanesePos === -1 || chessPos === -1 || japanesePos !== chessPos) continue; + + // Constraint 2: The Englishman lives in the red house + let englishmanPos = -1; + let redPos = -1; + for (let i = 0; i < 5; i++) { + if (nationalities[i] === 'Englishman') englishmanPos = i; + if (colors[i] === 'Red') redPos = i; + } + if (englishmanPos === -1 || redPos === -1 || englishmanPos !== redPos) continue; + + // Constraint 11: The person who enjoys reading lives in the house next to the person with the fox + let readingPos = -1; + let foxPos = -1; + for (let i = 0; i < 5; i++) { + if (hobbies[i] === 'Reading') readingPos = i; + if (pets[i] === 'Fox') foxPos = i; + } + if (readingPos === -1 || foxPos === -1 || !this.areNextTo(readingPos, foxPos)) continue; + + // Constraint 12: The painter's house is next to the house with the horse + let horsePos = -1; + for (let i = 0; i < 5; i++) { + if (pets[i] === 'Horse') horsePos = i; + } + if (horsePos === -1 || !this.areNextTo(paintingPos, horsePos)) continue; + + // If we get here, we have a valid solution + let waterDrinker = ''; + let zebraOwner = ''; + + for (let i = 0; i < 5; i++) { + if (drinks[i] === 'Water') { + waterDrinker = nationalities[i]; + } + if (pets[i] === 'Zebra') { + zebraOwner = nationalities[i]; + } + } + + return { + waterDrinker: waterDrinker, + zebraOwner: zebraOwner + }; + } + } + } + } + } + + // Edge Case: No valid solution found + throw new Error('No valid solution found for the Zebra Puzzle'); } } diff --git a/test.js b/test.js new file mode 100644 index 0000000..27af4be --- /dev/null +++ b/test.js @@ -0,0 +1,13 @@ +import { test } from 'node:test'; +import assert from 'node:assert/strict'; +import { ZebraPuzzle } from './main.js'; + +test('ZebraPuzzle should find the water drinker and zebra owner', () => { + const puzzle = new ZebraPuzzle(); + + // Test that the water drinker is the Norwegian + assert.strictEqual(puzzle.waterDrinker(), 'Norwegian'); + + // Test that the zebra owner is the Japanese + assert.strictEqual(puzzle.zebraOwner(), 'Japanese'); +});