1 题目及要求
1.1 题目描述
小毅有一个长度为N的整数数列 A={A[1],A[2], A[3], …., A[N]}。马教授给小毅出了一个难题:对数列A进行重新排序,使得数列A满足所有的A[i] * A[i+1]都是4的整数(1 <=i < N)。
小毅现在需要判断一个数列是否可以满足重重排后满足马教授的要求。
1)输入:
第一行,输入列数的个数t (1 <= t <= 50);
接下来的每两行描述一个数列A,第一行为数列长度n (1 <=n <=10^5),
第二行为n个整数A[i] (-10^9<= A[i] <= 10^9)
2)输出:
对于每个数列输出一行表示能否满足要求,如果满足输出Yes,否则输出No 。
3)输入范例:
2
3
1 10 100
4
1 2 3 4
4)输出范例:
Yes
No
2 解答
2.1 题目分析
用n1 表示不能被2整除的数(这样的数用1表示)的个数;n2表示能被2整除但不能被4整除(用2表示)的个数;n4表示能被4整除(用4表示)的个数。
很显然有以下规则:1必须挨着4放置,2绝不能挨着1放置。
这样,当不存在2时(n2==0), 则1和4间隔摆放,因此n1至多比n4大1;
当存在2时(0 < n2),由于2不能1放置,所以2至少得挨着一个4(如果有的话)放置,因此n1不能大于n4 。
2.2 代码
#include <iostream>
#include <vector>
using namespace std;
bool fun(vector<int> &nums) {
int nl = nums.size();
int n1(0), n2(0), n4(0);
for (int k1(0);k1<nl;++k1) {
if (nums[k1] & 1) ++n1; // 不能被2整除 (记为 1)
else if (nums[k1] & 3) ++n2; // 能被2整除, 不能被4整除 (记为 2)
else ++n4; // 能被4整除 (记为 4)
}
// 有2时,1的个数不能多于4的个数; 没有2时, 1的个数最多比4的个数多1
return n2 ? (n1<n4 + 1) : (n1<n4 + 2);
}
int main() {
int n, l;
cin >> n;
for (int k1(0);k1<n;++k1) {
cin >> l;
vector<int> nums(l);
for (int k2(0);k2<l;++k2) cin >> nums[k2];
cout << (fun(nums) ? "Yes" : "No") << endl;
}
return 0;
}