一道题能不能写多个不同的交互库？

比如，对于测试点 #1，交互库中的 int f(int x) 会返回 $x$；对于测试点 #2，交互库中的 int f(int x) 会返回 $\log x$；对于测试点 #3，交互库中的 int f(int x) 会返回 $x^2$；……

或者有没有什么方法能达到类似这种效果的？