135 lines
3.9 KiB
Plaintext
135 lines
3.9 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "google",
|
|
"metadata": {},
|
|
"source": [
|
|
"##### Copyright 2022 Google LLC."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "apache",
|
|
"metadata": {},
|
|
"source": [
|
|
"Licensed under the Apache License, Version 2.0 (the \"License\");\n",
|
|
"you may not use this file except in compliance with the License.\n",
|
|
"You may obtain a copy of the License at\n",
|
|
"\n",
|
|
" http://www.apache.org/licenses/LICENSE-2.0\n",
|
|
"\n",
|
|
"Unless required by applicable law or agreed to in writing, software\n",
|
|
"distributed under the License is distributed on an \"AS IS\" BASIS,\n",
|
|
"WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n",
|
|
"See the License for the specific language governing permissions and\n",
|
|
"limitations under the License.\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "basename",
|
|
"metadata": {},
|
|
"source": [
|
|
"# cp_sat_example"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "link",
|
|
"metadata": {},
|
|
"source": [
|
|
"<table align=\"left\">\n",
|
|
"<td>\n",
|
|
"<a href=\"https://colab.research.google.com/github/google/or-tools/blob/master/examples/notebook/sat/cp_sat_example.ipynb\"><img src=\"https://raw.githubusercontent.com/google/or-tools/master/tools/colab_32px.png\"/>Run in Google Colab</a>\n",
|
|
"</td>\n",
|
|
"<td>\n",
|
|
"<a href=\"https://github.com/google/or-tools/blob/master/ortools/sat/samples/cp_sat_example.py\"><img src=\"https://raw.githubusercontent.com/google/or-tools/master/tools/github_32px.png\"/>View source on GitHub</a>\n",
|
|
"</td>\n",
|
|
"</table>"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "doc",
|
|
"metadata": {},
|
|
"source": [
|
|
"First, you must install [ortools](https://pypi.org/project/ortools/) package in this colab."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "install",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"!pip install ortools"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "description",
|
|
"metadata": {},
|
|
"source": [
|
|
"Simple solve."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "code",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"from ortools.sat.python import cp_model\n",
|
|
"\n",
|
|
"\n",
|
|
"def main():\n",
|
|
" \"\"\"Minimal CP-SAT example to showcase calling the solver.\"\"\"\n",
|
|
" # Creates the model.\n",
|
|
" model = cp_model.CpModel()\n",
|
|
"\n",
|
|
" # Creates the variables.\n",
|
|
" var_upper_bound = max(50, 45, 37)\n",
|
|
" x = model.NewIntVar(0, var_upper_bound, 'x')\n",
|
|
" y = model.NewIntVar(0, var_upper_bound, 'y')\n",
|
|
" z = model.NewIntVar(0, var_upper_bound, 'z')\n",
|
|
"\n",
|
|
" # Creates the constraints.\n",
|
|
" model.Add(2 * x + 7 * y + 3 * z <= 50)\n",
|
|
" model.Add(3 * x - 5 * y + 7 * z <= 45)\n",
|
|
" model.Add(5 * x + 2 * y - 6 * z <= 37)\n",
|
|
"\n",
|
|
" model.Maximize(2 * x + 2 * y + 3 * z)\n",
|
|
"\n",
|
|
" # Creates a solver and solves the model.\n",
|
|
" solver = cp_model.CpSolver()\n",
|
|
" status = solver.Solve(model)\n",
|
|
"\n",
|
|
" if status == cp_model.OPTIMAL or status == cp_model.FEASIBLE:\n",
|
|
" print(f'Maximum of objective function: {solver.ObjectiveValue()}\\n')\n",
|
|
" print(f'x = {solver.Value(x)}')\n",
|
|
" print(f'y = {solver.Value(y)}')\n",
|
|
" print(f'z = {solver.Value(z)}')\n",
|
|
" else:\n",
|
|
" print('No solution found.')\n",
|
|
"\n",
|
|
" # Statistics.\n",
|
|
" print('\\nStatistics')\n",
|
|
" print(f' status : {solver.StatusName(status)}')\n",
|
|
" print(f' conflicts: {solver.NumConflicts()}')\n",
|
|
" print(f' branches : {solver.NumBranches()}')\n",
|
|
" print(f' wall time: {solver.WallTime()} s')\n",
|
|
"\n",
|
|
"\n",
|
|
"main()\n",
|
|
"\n"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 5
|
|
}
|