glguy/aocpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix divmod

Browse Source
This commit is contained in:
Eric Mertens 2023-01-13 17:03:44 -08:00
parent 978a8ad39d
commit 0d3c9e3422
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
2022/25.cpp
View File

@ -19,7 +19,7 @@ auto DivMod(T dividend, T divisor) -> std::pair<T, T>
{ {
auto const quotient = dividend / divisor; auto const quotient = dividend / divisor;
auto const remainder = dividend % divisor; auto const remainder = dividend % divisor;
if ((remainder >= 0) == (divisor >= 0)) { if (remainder == 0 || (remainder > 0) == (divisor > 0)) {
return {quotient, remainder}; return {quotient, remainder};
} else { } else {
return {quotient - 1, remainder + divisor}; return {quotient - 1, remainder + divisor};
@ -101,6 +101,11 @@ TEST_SUITE("2022-25 examples") {
CHECK(DivMod<int>(-2,5) == std::make_pair(-1,3)); CHECK(DivMod<int>(-2,5) == std::make_pair(-1,3));
CHECK(DivMod<int>(2,-5) == std::make_pair(-1,-3)); CHECK(DivMod<int>(2,-5) == std::make_pair(-1,-3));
CHECK(DivMod<int>(-2,-5) == std::make_pair(0,-2)); CHECK(DivMod<int>(-2,-5) == std::make_pair(0,-2));
CHECK(DivMod<int>(-5,-5) == std::make_pair(1,0));
CHECK(DivMod<int>(-5,5) == std::make_pair(-1,0));
CHECK(DivMod<int>(5,-5) == std::make_pair(-1,0));
CHECK(DivMod<int>(-5,-5) == std::make_pair(1,0));
} }
} }