标签：insert set -- s1 Two erase 咒语 CF1398 sumb

题目链接：https://codeforces.com/problemset/problem/1398/E

分析

不难发现，每次我们要让能被双倍使用的咒语尽量大，能被双倍使用的咒语数量就是1类咒语的数量。而能被双倍使用的咒语即是除了第一次使用的1类咒语外的所有咒语。
根据贪心的思想，第一次使用的1类咒语肯定是1类咒语种最小的，其他咒语根据大小排序之后选择最大的几个即可。
可以用set来维护，具体实现看代码。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define fi first
#define se second
#define lson (k << 1)
#define rson (k << 1 | 1)
 
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const int M = 1e6 + 10;
const ll P = 1e9 + 7;
 
int n, op, db, cnt;
set<int> s1, s2; //分别存储第1类咒语以及所有能被双倍使用的咒语
set<int>::iterator f;
ll suma, sumb;
 
void check()
{
    while(cnt < db && f != s2.begin())
    {
        f--;
        cnt++;
        sumb += *f;
    }
    while(cnt > db)
    {
        sumb -= *f;
        f++;
        cnt--;
    }
}
 
void insert(int x)
{
    s2.insert(x);
    if(x > *f)
    {
        sumb += x;
        cnt++;
    }
    check();
}
 
void erase(int x)
{
    if(x == *f)
    {
        cnt--;
        f++;
        sumb -= x;
    }
    else if(x > *f)
    {
        cnt--;
        sumb -= x;
    }
    s2.erase(x);
    check();
}
 
int main() {
    int x;
    s2.insert(INF);
    f = --s2.end();
    scanf("%d",&n);
    while(n--)
    {
        scanf("%d%d",&op,&x);
        if(op == 1)
        {
            if(x > 0)
            {
                db++;
                suma += x;
                if(!s1.size())
                {
                    s1.insert(x);
                }
                else
                {
                    if(x > *s1.begin())
                    {
                        s1.insert(x);
                        insert(x);
                    }
                    else
                    {
                        insert(*s1.begin());
                        s1.insert(x);
                    }
                }
            }
            else
            {
                x = -x;
                suma -= x;
                db--;
                if(x == *s1.begin())
                {
                    s1.erase(x);
                    if(s1.size()) erase(*s1.begin());
                }
                else
                {
                    erase(x);
                    s1.erase(x);
                }
            }
        }
        else
        {
            if(x > 0)
            {
                suma += x;
                insert(x);
            }
            else
            {
                x = -x;
                suma -= x;
                erase(x);
            }
        }
        check();
        printf("%lld\n",suma + sumb);
    }
    return 0;
}

标签：insert,set,--,s1,Two,erase,咒语,CF1398,sumb
来源： https://blog.csdn.net/Sankkl1/article/details/120903044

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。