hi.Math

Math

链接：https://ac.nowcoder.com/acm/contest/90041/C
来源：牛客网

题目描述

智乃有N种不同颜色的小球，第i种颜色的小球有 $a_i$ 个，放在同一个盒子中。

假设智乃每次任意取出两个不同颜色的小球并丢弃，然后重复这一过程，直到盒子为空或者只剩下一种颜色的小球。

智乃想要知道每种颜色的小球是否为最终可能被剩下的颜色。

一种小球的颜色为最终可能被剩下的颜色，当且仅当至少存在一种丢弃小球的方案使得盒子中最后只剩下（至少存在1个）该种颜色的小球。

输入描述:

第一行输入一个正整数 $T$ $(1\le T\le 10^6)$
对于每组测试案例，第一行输入一个正整数 $N$ $(1\le N\le 10^5)$ 表示有 $N$ 种颜色的小球，接下来一行输入 $N$ 个正整数 $(1\le a_i\le 10^9)$ 表示每种颜色的小球数目。

输入数据保证 $\sum$ $N\le 10^6$。

输出描述:

输出一行 $N$ 个整数 $an{s}_i,an{s}_i\in$ {0,1}，表示第i种颜色的小球是否有可能成为最终剩下的小球，如果可能输出1，否则输出0，整数之间用一个空格隔开。

输入示例1

1
2
10 10

输出示例1

0 0

说明

只有两种颜色，个数相同，两两相消

输入示例2

2
2
10 110
3
1 1 1

输出示例2

0 1
1 1 1

说明

对于第一个样例，两两相消后只会剩下第2种球。
对于第二个样例，一开始消除1,2可以剩下3；消除1,3剩下2；消除2,3剩下1。所以每种小球都可能被留到最后。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <sstream>//整型转字符串
// #include <stack>//栈
// #include <deque>//堆/优先队列
// #include <queue>//队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const ll N=1e5+10;
ll a[N];

int main()
{
    ll t;
    cin>>t;

    while(t--)
    {
        ll n;
        cin>>n;

        ll sum=0,mx=0,sx=0;
        for(ll i=0;i<n;i++)
        {
            cin>>a[i];
            sum+=a[i];

            // mx1,sx2
            if(a[i]>=mx)
            {
                sx=max(sx,mx);
                mx=a[i];
            }
            else sx=max(a[i],sx);
        }

        ll p=1;
        if(sum%2==0) p=2;
        for(ll i=0;i<n;i++)
        {
            if(a[i]<p) cout<<"0 ";
            else if((a[i]==mx&&(sum-p)>=sx*2)||(sum-p)>=mx*2)
                cout<<"1 ";
            else cout<<"0 ";
        }
        cout<<endl;
    }
    
    return 0;
}