8.1 习题日志 T3序列I

题目描述

给定一个序列 a=a1,a2,…,an，有以下两种操作：

• 若 ai=ai+1，则可将 ai 与 ai+1 同时加 1

• 将 ai 加 2。

求问是否可经过多此操作后使得所有 ai相等。

输入格式

第一行，一个正整数T，代表数据组数。

接下来T*2行，每两行一个数据，其中第一行一个正整数n代表序列长度，第二行n个正整数。

输出格式

共T行，每行"YES"或者"NO"代表是否可行。

样例

Input 1

4 5 2 1 1 2 5 3 4 5 3 2 10 10 3 1 2 3

Output 1

YES YES YES NO

Input 2

2 8 3 2 5 8 4 1 14 11 7 3 7 4 2 13 6 5

Output 2

YES NO

数据范围

对于60%的数据，T=1，1≤n≤10

对于100%的数据，1≤T≤10,1≤n≤200000

仔细查看题目：操作1需要两个相同的相邻数字，而操作2不改变数字的奇偶性，所以我们可以简化本体，不考虑操作2，将数字按奇偶变为1与0，同时，任何两个1或0可以互相转变，所以可以直接删去。结合栈的结构，可以将数字压入，相同则删除。若最后栈剩下0个或1个元素则成立。

AC代码如下

#include <bits/stdc++.h>
using namespace std;
int T, n;
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> T;
    while (T--)
    {
        cin >> n;
        stack<bool> st;
        for (int i = 1; i <= n;i++){
            int t;
            cin >> t;
            t %= 2;
            if(!st.empty() && st.top() == t)
                st.pop();
            else
                st.push(t);
        }
        if(st.size()>1)
            cout << "NO\n";
        else
            cout << "YES\n";
    }
}