LeetCode 1237. 找出给定方程的正整数解

1. 题目

给出一个函数 f(x, y) 和一个目标结果 z，请你计算方程 f(x,y) == z 所有可能的正整数数对 x 和 y。

给定函数是严格单调的，也就是说：

f(x, y) < f(x + 1, y)
f(x, y) < f(x, y + 1)

函数接口定义如下：

interface CustomFunction {
public:// Returns positive integer f(x, y) for any given positive integer x and y.int f(int x, int y);
};

如果你想自定义测试，你可以输入整数 function_id 和一个目标结果 z 作为输入，其中 function_id 表示一个隐藏函数列表中的一个函数编号，题目只会告诉你列表中的 2 个函数。

你可以将满足条件的结果数对按任意顺序返回。

示例 1：
输入：function_id = 1, z = 5
输出：[[1,4],[2,3],[3,2],[4,1]]
解释：function_id = 1 表示 f(x, y) = x + y示例 2：
输入：function_id = 2, z = 5
输出：[[1,5],[5,1]]
解释：function_id = 2 表示 f(x, y) = x * y提示：
1 <= function_id <= 9
1 <= z <= 100
题目保证 f(x, y) == z 的解处于 1 <= x, y <= 1000 的范围内。
在 1 <= x, y <= 1000 的前提下，题目保证 f(x, y) 是一个 32 位有符号整数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-positive-integer-solution-for-a-given-equation
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：搜索二维矩阵
x=1,y=1,相当于矩阵左上角，x=1000，y=1000,相当于矩阵右下角

在这里插入图片描述

class Solution {
public:vector<vector<int>> findSolution(CustomFunction& cf, int z) {int x = 1, y = 1000, val;vector<vector<int>> ans;while(x<=1000 && y >=1){val = cf.f(x,y);if(val < z)x++;else if(val > z)y--;else{ans.push_back({x,y});x++;}}return ans;}
};

class Solution {
public:vector<vector<int>> findSolution(CustomFunction& cf, int z) {int x = 1000, y = 1, val;vector<vector<int>> ans;while(x>=1 && y<=1000){val = cf.f(x,y);if(val < z)y++;else if(val > z)x--;else{ans.push_back({x,y});x--;}}return ans;}
};