题解 | #编程题1#
编程题1
http://www.nowcoder.com/practice/ff518f3162c849b9a84d1fab8e7179be
对MemoryLimitExceeded大佬公式解法进行了一个小小的优化,将求a,b,c值的公式进行了精简,代码看起来更加的简洁,易懂:
思路:穷举d1,d2的正负号,求出a,b,c的值,解决不平衡的情况,剩下的都是平衡的。
代码:
#include<iostream> using namespace std; int main(){ int t; long long n, k , d1, d2, a, b, c; cin >> t; while(t--){ cin >> n >> k >> d1 >> d2; if(n % 3 != 0){ cout << "no" << endl; continue; } int flag = 0; for(int i = -1; i <= 1; i = i + 2 ){ for(int j = -1; j <= 1; j = j + 2){ b = (k + d1*i + d2*j)/3; a = b - d1*i; c = b - d2*j; if(a < 0 || b < 0 || c < 0 || a + b+ c != k)continue; if(a > n/3 || b > n/3 || c > n/3)continue; cout << "yes" <<endl; flag = 1; break; } if(flag == 1)break; if(flag == 0)cout << "no" << endl; } return 0; }