solucao_projeto_3.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import copy"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "class Mapa:\n",
    "\n",
    "    def __init__(self, mapa):\n",
    "        # Importante usar o deepcopy para não alterar as variáveis passadas\n",
    "        self.mapa_original = copy.deepcopy(mapa) # Variável usada no método representativo\n",
    "        self.mapa = copy.deepcopy(mapa) # Variável usada na varredura que será modificada\n",
    "\n",
    "\n",
    "    def __conta_tamanho_rio__(self, x, y, tamanho = 0):\n",
    "        # Criei uma variável auxiliar pra não ter que escrever self toda hora\n",
    "        mapa = self.mapa\n",
    "\n",
    "        tamanho += 1\n",
    "            \n",
    "        # Visita o ponto acima\n",
    "        if y > 0 and mapa[y - 1][x] == 1:\n",
    "            mapa[y - 1][x] = 'Visitado'\n",
    "            tamanho = self.__conta_tamanho_rio__(x, y - 1, tamanho)\n",
    "\n",
    "\n",
    "        # Visita o ponto abaixo\n",
    "        if y + 1 < len(mapa) and mapa[y + 1][x] == 1:\n",
    "            mapa[y + 1][x] = 'Visitado'\n",
    "            tamanho = self.__conta_tamanho_rio__(x, y + 1, tamanho)\n",
    "\n",
    "\n",
    "        # Visita o ponto à direita\n",
    "        if x + 1 < len(mapa[0]) and mapa[y][x + 1] == 1:\n",
    "            mapa[y][x+1] = 'Visitado'\n",
    "            tamanho = self.__conta_tamanho_rio__(x + 1, y, tamanho)\n",
    "\n",
    "\n",
    "        # Visita o ponto à esquerda\n",
    "        if x > 0 and mapa[y][x - 1] == 1:\n",
    "            mapa[y][x-1] = 'Visitado'\n",
    "            tamanho = self.__conta_tamanho_rio__(x - 1, y, tamanho)\n",
    "\n",
    "        return tamanho\n",
    "        \n",
    "\n",
    "    def acha_rio(self):\n",
    "\n",
    "        tamanho_rios = []\n",
    "\n",
    "        for y in range(len(self.mapa)):\n",
    "            for x in range(len(self.mapa[0])):\n",
    "                if self.mapa[y][x] == 1:\n",
    "                    self.mapa[y][x] = 'Visitado'\n",
    "\n",
    "                    # Começa a varredura a partir do primeiro ponto = 1 encontrado\n",
    "                    tamanho_rios.append(self.__conta_tamanho_rio__(x, y))\n",
    "\n",
    "        self.tamanho_rios = sorted(tamanho_rios)\n",
    "\n",
    "    def __repr__(self):\n",
    "\n",
    "        texto = ''\n",
    "        texto += 'Contagem dos rios do mapa:\\n\\n'\n",
    "\n",
    "        for y in range(len(self.mapa_original)):\n",
    "            texto += '| '\n",
    "            for x in range(len(self.mapa_original[0])):\n",
    "                texto += str(self.mapa_original[y][x]) + ' '\n",
    "            texto += '|\\n'\n",
    "\n",
    "        if self.tamanho_rios == []:\n",
    "            texto += '\\nNão encontrei nenhum rio!'\n",
    "        else:\n",
    "            texto += f'\\nEncontrei {len(self.tamanho_rios)} rios: {self.tamanho_rios}\\n'\n",
    "\n",
    "        return texto\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [],
   "source": [
    "mapa_1 = [[0, 0, 0, 1, 0],\n",
    "          [1, 1, 0, 1, 1],\n",
    "          [1, 0, 0, 1, 0],\n",
    "          [0, 1, 0, 0, 1],\n",
    "          [0, 0, 0, 0, 0],\n",
    "          [1, 1, 1, 1, 1],\n",
    "          [1, 0, 1, 0, 1]]\n",
    "\n",
    "# Resposta esperada: [1, 1, 3, 4, 8]\n",
    "\n",
    "\n",
    "mapa_2 = [[1, 1, 0],\n",
    "          [1, 0, 0],\n",
    "          [1, 0, 0],\n",
    "          [1, 0, 1],\n",
    "          [0, 1, 0],\n",
    "          [1, 0, 0],\n",
    "          [1, 0, 1],\n",
    "          [1, 1, 1],\n",
    "          [1, 0, 0]]\n",
    "\n",
    "# Resposta esperada: [1, 1, 5, 7]\n",
    "\n",
    "\n",
    "\n",
    "mapa_3 = [[0, 0, 1, 0],\n",
    "          [1, 1, 0, 1],\n",
    "          [0, 1, 0, 0],\n",
    "          [0, 0, 0, 1],\n",
    "          [1, 0, 1, 1],\n",
    "          [0, 1, 0, 0],\n",
    "          [0, 0, 1, 0],\n",
    "          [1, 1, 0, 1],\n",
    "          [1, 0, 1, 1],\n",
    "          [0, 0, 0, 1],\n",
    "          [0, 0, 1, 1]]\n",
    "\n",
    "# Resposta esperada: [1, 1, 1, 1, 1, 3, 3, 3, 6]\n",
    "\n",
    "\n",
    "mapa_4 = [[1, 1, 1, 1],\n",
    "          [1, 1, 1, 0],\n",
    "          [1, 1, 1, 0],\n",
    "          [1, 1, 1, 0],\n",
    "          [1, 1, 1, 1],\n",
    "          [1, 0, 1, 0],\n",
    "          [1, 0, 0, 0],\n",
    "          [0, 0, 0, 0],\n",
    "          [0, 0, 0, 0],\n",
    "          [1, 0, 1, 1]]\n",
    "\n",
    "# Resposta esperada: [1, 2, 20]\n",
    "\n",
    "\n",
    "mapa_5 = [[1, 1, 1, 1],\n",
    "          [1, 0, 0, 0],\n",
    "          [1, 0, 0, 0],\n",
    "          [1, 0, 0, 1]]\n",
    "\n",
    "# Resposta esperada: [1, 7]\n",
    "\n",
    "mapa_6 = [[0, 0, 0], \n",
    "          [0, 0, 0]]\n",
    "\n",
    "lista_mapas = [mapa_1, mapa_2, mapa_3, mapa_4, mapa_5, mapa_6]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Contagem dos rios do mapa:\n",
      "\n",
      "| 0 0 0 1 0 |\n",
      "| 1 1 0 1 1 |\n",
      "| 1 0 0 1 0 |\n",
      "| 0 1 0 0 1 |\n",
      "| 0 0 0 0 0 |\n",
      "| 1 1 1 1 1 |\n",
      "| 1 0 1 0 1 |\n",
      "\n",
      "Encontrei 5 rios: [1, 1, 3, 4, 8]\n",
      "\n",
      "Contagem dos rios do mapa:\n",
      "\n",
      "| 1 1 0 |\n",
      "| 1 0 0 |\n",
      "| 1 0 0 |\n",
      "| 1 0 1 |\n",
      "| 0 1 0 |\n",
      "| 1 0 0 |\n",
      "| 1 0 1 |\n",
      "| 1 1 1 |\n",
      "| 1 0 0 |\n",
      "\n",
      "Encontrei 4 rios: [1, 1, 5, 7]\n",
      "\n",
      "Contagem dos rios do mapa:\n",
      "\n",
      "| 0 0 1 0 |\n",
      "| 1 1 0 1 |\n",
      "| 0 1 0 0 |\n",
      "| 0 0 0 1 |\n",
      "| 1 0 1 1 |\n",
      "| 0 1 0 0 |\n",
      "| 0 0 1 0 |\n",
      "| 1 1 0 1 |\n",
      "| 1 0 1 1 |\n",
      "| 0 0 0 1 |\n",
      "| 0 0 1 1 |\n",
      "\n",
      "Encontrei 9 rios: [1, 1, 1, 1, 1, 3, 3, 3, 6]\n",
      "\n",
      "Contagem dos rios do mapa:\n",
      "\n",
      "| 1 1 1 1 |\n",
      "| 1 1 1 0 |\n",
      "| 1 1 1 0 |\n",
      "| 1 1 1 0 |\n",
      "| 1 1 1 1 |\n",
      "| 1 0 1 0 |\n",
      "| 1 0 0 0 |\n",
      "| 0 0 0 0 |\n",
      "| 0 0 0 0 |\n",
      "| 1 0 1 1 |\n",
      "\n",
      "Encontrei 3 rios: [1, 2, 20]\n",
      "\n",
      "Contagem dos rios do mapa:\n",
      "\n",
      "| 1 1 1 1 |\n",
      "| 1 0 0 0 |\n",
      "| 1 0 0 0 |\n",
      "| 1 0 0 1 |\n",
      "\n",
      "Encontrei 2 rios: [1, 7]\n",
      "\n",
      "Contagem dos rios do mapa:\n",
      "\n",
      "| 0 0 0 |\n",
      "| 0 0 0 |\n",
      "\n",
      "Não encontrei nenhum rio!\n"
     ]
    }
   ],
   "source": [
    "for i in lista_mapas:\n",
    "    objeto = Mapa(i)\n",
    "    objeto.acha_rio()\n",
    "    print(objeto)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "9e1544ece40983b9dcb10e0aa1f5d0cdb8d9e539b9b8df5e9e5017ad7f05ebe2"
  },
  "kernelspec": {
   "display_name": "Python 3.9.2 64-bit",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.2"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}