编程题1-球队比赛(2018字节跳动校招)

题目描述

有三只球队，每只球队编号分别为球队1，球队2，球队3，这三只球队一共需要进行 n 场比赛。现在已经踢完了k场比赛，每场比赛不能打平，踢赢一场比赛得一分，输了不得分不减分。已知球队1和球队2的比分相差d1分，球队2和球队3的比分相差d2分，每场比赛可以任意选择两只队伍进行。求如果打完最后的 (n-k) 场比赛，有没有可能三只球队的分数打平。

输入描述

第一行包含一个数字 t (1 <= t <= 10)
接下来的t行每行包括四个数字 n, k, d1, d2(1 <= n <= 10^12; 0 <= k <= n, 0 <= d1, d2 <= k)

输出描述

每行的比分数据，最终三只球队若能够打平，则输出“yes”，否则输出“no”

示例1

输入：

2
3 3 0 0
3 3 3 3

输出：

yes
no

说明

case1: 球队1和球队2 差0分，球队2 和球队3也差0分，所以可能的赛得分是三只球队各得1分
case2: 球队1和球队2差3分，球队2和球队3差3分，所以可能的得分是 球队1得0分，球队2得3分, 球队3 得0分，比赛已经全部结束因此最终不能打平。

算法分析

题目中只说队伍之间相差的分数，并没有说哪支队伍得分多，哪支队伍得分少。所以，本题应该分4种情况讨论。假设球队1得分为m (m >= 0) ，至少需要need 场比赛才能持平。

当 球队1< 球队2，球队2<球队3 时，得分情况：

        球队1：m
        球队2：m + d1
        球队3：m + d1 + d2 。此时有3 * m = k - d1 - d1 - d2    need = d1 + d2 + d2（此时球队3得分最多，所以球队1还需要赢d1 + d2场，球队2还需要赢d2场）

当 球队1< 球队2，球队2>球队3 时，得分情况：

        球队1：m
        球队2：m + d1
        球队3：m + d1 - d2 。此时有3 * m = k - d1 - d1 + d2    need = d1 + d2 （此时球队2得分最多，所以球队1还需要赢d1场，球队3还需要赢d2场）

当 球队1> 球队2，球队2>球队3 时，得分情况：
        球队1：m
        球队2：m - d1
        球队3：m - d1 - d2 。此时有3 * m = k + d1 + d1 + d2    need = d1 + d1 + d2 （此时球队1得分最多，所以球队2还需要赢d1场，球队3还需要赢d1 + d2场）

当 球队1> 球队2，球队2<球队3 时，得分情况：
        球队1：m
        球队2：m - d1
        球队3：m - d1 + d2 。
        此时有3 * m = k + d1 + d1 - d2  。这时不能确定哪个球队得分最多，还要分情况：
                    当 d1  >= d2 时，球队1得分最多need = d1 + d1 - d2
                    当 d1  <   d2 时，球队3得分最多need = d2 - d1 + d2

但是写代码时，不能用if   else if  else if....这种结构，因为同一组数据可能满足两种以上的情况，所以要把这几种情况都判断一遍，如果都不能输出yes，那么最后输出no
 

提交代码：

#include<iostream>
#include<vector>

using namespace std;

int main()
{
	long long t, n, k, d1, d2;
	cin >> t;
	while (t--) {
		cin >> n >> k >> d1 >> d2;

		long long need, m;
		//#1 a>b>c
		m = k + d2 + 2 * d1;
		if (m >= 0 && m % 3 == 0) {
			need = (n - k) - (2 * d1 + d2);
			if (need >= 0 && need % 3 == 0) {
				cout << "yes" << endl;
				continue;
			}
		}
		//#2 a<b<c
		m = k - d2 - 2 * d1;
		if (m >= 0 && m % 3 == 0) {
			need = (n - k) - (d1 + 2 * d2);
			if (need >= 0 && need % 3 == 0) {
				cout << "yes" << endl;
				continue;
			}
		}
		//#3 a<b, b>c
		m = k + d2 - 2 * d1;
		if (m >= 0 && m % 3 == 0) {
			need = (n - k) - (d1 + d2);
			if (need >= 0 && need % 3 == 0) {
				cout << "yes" << endl;
				continue;
			}
		}
		//#4 a>b, b<c
		m = k - d2 + 2 * d1;
		if (m >= 0 && m % 3 == 0) {
			if (d2 > d1) need = (n - k) - (2 * d2 - d1);
			else need = (n - k) - (2 * d1 - d2);

			if (need >= 0 && need % 3 == 0) {
				cout << "yes" << endl;
				continue;
			}
		}

		cout << "no" << endl;
	}

	return 0;
}