Codeforces Round #702 (Div. 3)-E. Accidental Victory-迭代器跳数

Posted 2021-06-08 Chivas_/Regal

题目：

思路：
首先分析一下
一个数可以把所有<=它的数吃掉
加上后如果还能吃掉第一个比它大的数
就可以继续往上吃直到吃完所有的数

如果吃完比它小的数后
不能吃掉第一个比它大的数
那么就是“鸿沟区“了：比它小的数无论怎么吃都跨不过这条”鸿沟”

而且还能发现，只要一个数能吃完所有的数，所有与它相同的数也都能吃完，反之亦然。

所以我们不去一个个位置遍历，而是利用map迭代器遍历的去重性质，进行map迭代器降序遍历
存入一个sum代表吃完 >=它的 所有的数，所能达到最大值
遍历完每一个值，sum要减的量是“当前数 * 当前数的个数”
如果在遍历某一个值中发现sum < 比它大的第一个数了，那么就说明鸿沟出现，就可以break了

/*
           ________   _                                              ________                              _
          /  ______| | |                                            |   __   |                            | |
         /  /        | |                                            |  |__|  |                            | |
         |  |        | |___    _   _   _   ___  _   _____           |     ___|   ______   _____   ___  _  | |
         |  |        |  __ \\  |_| | | | | |  _\\| | | ____|          |  |\\  \\    |  __  | |  _  | |  _\\| | | |
         |  |        | |  \\ |  _  | | | | | | \\  | | \\___           |  | \\  \\   | |_/ _| | |_| | | | \\  | | |
         \\  \\______  | |  | | | | \\ |_| / | |_/  |  ___/ |          |  |  \\  \\  |    /_   \\__  | | |_/  | | |
Author :  \\________| |_|  |_| |_|  \\___/  |___/|_| |_____| _________|__|   \\__\\ |______|     | | |___/|_| |_|
                                                                                         ____| |
                                                                                         \\_____/
*/
//#include <unordered_map>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <utility>
#include <string>
#include <vector>
#include <cstdio>
#include <stack>
#include <queue>
#include <cmath>
#include <map>
#include <set>

#define G 10.0
#define LNF 1e18
#define EPS 1e-6
#define PI acos(-1.0)
#define INF 0x7FFFFFFF

#define ll long long
#define ull unsigned long long

#define LOWBIT(x) ((x) & (-x))
#define LOWBD(a, x) lower_bound(a.begin(), a.end(), x) - a.begin()
#define UPPBD(a, x) upper_bound(a.begin(), a.end(), x) - a.begin()
#define TEST(a) cout << "---------" << a << "---------" << '\\n'

#define CHIVAS int main()
#define _REGAL exit(0)

#define SP system("pause")
#define IOS ios::sync_with_stdio(false)
//#define map unordered_map

#define _int(a) int a; cin >> a
#define  _ll(a) ll a; cin >> a
#define _char(a) char a; cin >> a
#define _string(a) string a; cin >> a
#define _vectorInt(a, n) vector<int>a(n); cin >> a
#define _vectorLL(a, b) vector<ll>a(n); cin >> a

#define PB(x) push_back(x)
#define ALL(a) a.begin(),a.end()
#define MEM(a, b) memset(a, b, sizeof(a))
#define EACH_CASE(cass) for (cin >> cass; cass; cass--)

#define LS l, mid, rt << 1
#define RS mid + 1, r, rt << 1 | 1
#define GETMID (l + r) >> 1

using namespace std;

template<typename T> inline void Read(T &x){T f = 1; x = 0;char s = getchar();while(s < '0' || s > '9'){if(s == '-') f = -1; s = getchar();}while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}x*=f;}
template<typename T> inline T MAX(T a, T b){return a > b? a : b;}
template<typename T> inline T MIN(T a, T b){return a > b? b : a;}
template<typename T> inline void SWAP(T &a, T &b){T tp = a; a = b; b = tp;}
template<typename T> inline T GCD(T a, T b){return b > 0? GCD(b, a % b) : a;}
template<typename T> inline void ADD_TO_VEC_int(T &n, vector<T> &vec){vec.clear(); cin >> n; for(int i = 0; i < n; i ++){T x; cin >> x, vec.PB(x);}}
template<typename T> inline pair<T, T> MaxInVector_ll(vector<T> vec){T MaxVal = -LNF, MaxId = 0;for(int i = 0; i < (int)vec.size(); i ++) if(MaxVal < vec[i]) MaxVal = vec[i], MaxId = i; return {MaxVal, MaxId};}
template<typename T> inline pair<T, T> MinInVector_ll(vector<T> vec){T MinVal = LNF, MinId = 0;for(int i = 0; i < (int)vec.size(); i ++) if(MinVal > vec[i]) MinVal = vec[i], MinId = i; return {MinVal, MinId};}
template<typename T> inline pair<T, T> MaxInVector_int(vector<T> vec){T MaxVal = -INF, MaxId = 0;for(int i = 0; i < (int)vec.size(); i ++) if(MaxVal < vec[i]) MaxVal = vec[i], MaxId = i; return {MaxVal, MaxId};}
template<typename T> inline pair<T, T> MinInVector_int(vector<T> vec){T MinVal = INF, MinId = 0;for(int i = 0; i < (int)vec.size(); i ++) if(MinVal > vec[i]) MinVal = vec[i], MinId = i; return {MinVal, MinId};}
template<typename T> inline pair<map<T, T>, vector<T> > DIV(T n){T nn = n;map<T, T> cnt;vector<T> div;for(ll i = 2; i * i <= nn; i ++){while(n % i == 0){if(!cnt[i]) div.push_back(i);cnt[i] ++;n /= i;}}if(n != 1){if(!cnt[n]) div.push_back(n);cnt[n] ++;n /= n;}return {cnt, div};}
template<typename T>             vector<T>& operator--            (vector<T> &v){for (auto& i : v) --i;            return  v;}
template<typename T>             vector<T>& operator++            (vector<T> &v){for (auto& i : v) ++i;            return  v;}
template<typename T>             istream& operator>>(istream& is,  vector<T> &v){for (auto& i : v) is >> i;        return is;}
template<typename T>             ostream& operator<<(ostream& os,  vector<T>  v){for (auto& i : v) os << i << ' '; return os;}

map<ll, ll> res;//first = 某个数，second = 这个数能不能赢
map<ll, ll, greater<ll> > num;//first = 某个数，second = 这个数的个数

inline void Init(){
        res.clear();
        num.clear();
}

inline void solve(){Init();
        _ll(n); _vectorLL(a, n);
        ll sum = 0;//所有数的总和
        for(int i = 0; i < n; i ++) num[a[i]] ++, sum += a[i];

        map<ll, ll>::iterator las = num.begin();//比当前数大的第一个数（上一个数）
        for(map<ll, ll>::iterator it = num.begin(); it != num.end(); it ++){

                if(it == num.begin()) res[it->first] = 1, sum -= it->first * it->second;//最大的肯定能选
                else{
                        if(las->first > sum) break;//如果当前总和比上一个数要小，鸿沟出现，就跳出循环
                        res[it->first] = 1;//记录一下这个数能被选
                        sum -= it->first * it->second;
                        las ++;
                }

        }

        int res_cnt = 0;//能选的数的个数
        for(int i = 0; i < n; i ++) if(res[a[i]]) res_cnt ++;
        cout << res_cnt << endl;
        for(int i = 0; i < n; i ++) if(res[a[i]]) cout << i + 1 << " ";//如果这个数可以被选，就输出下标
        cout << endl;
}

CHIVAS{
        int cass;
        EACH_CASE(cass){
                solve();
        }
        _REGAL;
}