Codeforces Round #670(Div. 2) A,B,C,D,E

A. Subset Mex

桶排序，遇到没有的x，直接加上A,B剩下数量*x，遇到只剩一个的，如果A,B都还在，那么让一个结束，答案加上x，否则继续。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
int T,n;
int a[105];
int s[105];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        memset(s,0,sizeof(s));
        for(int i=1;i<=n;++i)
        {
            scanf("%d",&a[i]);
            s[a[i]]++;
        }
        int tag=0;
        int ans=0;
        for(int i=0;i<=100;++i)
        {
            if(!s[i])
            {
                ans+=(2-tag)*i;
                tag=2;
                break;
            }
            else
            {
                if(s[i]==1&&!tag)
                {
                    tag++;
                    ans+=i;
                }
            }
        }
        if(!tag)
        ans+=202;
        else
        if(tag==1)
        ans+=101;
        printf("%d\n",ans);
    }
    return 0;
}

B. Maximum Product

从排好序的数组中前五个和后五个中找连续的五个，取乘积最大值。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N=1e5+5;
typedef long long ll;
int T,n,cnt;
ll a[N];
vector<ll>pos,neg;
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        cnt=0;
        scanf("%d",&n);
        pos.clear();
        neg.clear();
        for(int i=1;i<=n;++i)
        {
            scanf("%lld",&a[i]);
            if(a[i]>=0)
            pos.push_back(a[i]);
            else
            neg.push_back(a[i]);
        }
        if(n==5)
        {
            printf("%lld\n",a[1]*a[2]*a[3]*a[4]*a[5]);
            continue;
        }
        sort(pos.begin(),pos.end(),greater<ll>());
        if(!pos.size())
        {
            sort(neg.begin(),neg.end(),greater<ll>());
            printf("%lld\n",neg[0]*neg[1]*neg[2]*neg[3]*neg[4]);
            continue;
        }
        sort(neg.begin(),neg.end());
        ll t1,t2,t3,t4,t5,t6;
        if(neg.size()>=4)
        {
            t1=neg[0]*neg[1];
            t2=neg[2]*neg[3];
        }
        else
        if(neg.size()>=2)
        {
            t1=neg[0]*neg[1];
            t2=0;
        }
        else
        t1=t2=0;
        if(pos.size()>=4)
        {
            t3=pos[0]*pos[1];
            t4=pos[2]*pos[3];
            t6=pos[2];
            if(pos.size()>4)
            t5=pos[4];
            else
            t5=0;
        }
        else
        if(pos.size()>=2)
        {
            t3=pos[0]*pos[1];
            t4=0;
            t5=0;
            if(pos.size()>2)
            t6=pos[2];
            else
            t6=0;
        }
        else
        t3=t4=t5=t6=0;
        //cout<<neg[0]<<" "<<neg[1]<<" "<<t1<<" "<<t2<<" "<<t3<<" "<<t4<<" "<<t5<<" "<<t6<<endl;
        printf("%lld\n",max(t1*t2*pos[0],max(t1*t3*t6,t3*t4*t5)));
    }
    return 0;
}

C. Link Cut Centroids

我们发现，如果能找到的话，原树重心要么是一个要么是两个，如果是一个的话，随便找一条边剪掉再连上。如果是两个，让其中一个重心断一条边，并与另一个重心连上。注意断的那条边的另一个点不能是另一个重心。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e5+5;
const int inf=0x3f3f3f3f;
struct node
{
    int to,next;
}edge[N<<1];
int head[N];
int size[N],ma[N];
int T,n,cnt,up,s;
void add(int x,int y)
{
    edge[++cnt].to=y;
    edge[cnt].next=head[x];
    head[x]=cnt;
}
void get_heavy(int u,int fa)
{
    //puts("1");
    ma[u]=0;
    size[u]=1;
    for(int i=head[u];i;i=edge[i].next)
    {
        //puts("1");
        int v=edge[i].to;
        if(v==fa)
        continue;
        get_heavy(v,u);
        ma[u]=max(ma[u],size[v]);
        size[u]+=size[v];
    }
    ma[u]=max(ma[u],n-size[u]);
    up=min(up,ma[u]);
    //puts("1");
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        s=cnt=0;
        up=inf;
        scanf("%d",&n);
        int x,y;
        for(int i=1;i<=n;++i)
        head[i]=0;
        for(int i=1;i<n;++i)
        {
            scanf("%d%d",&x,&y);
            add(x,y);
            add(y,x);
        }
        //puts("1");
        get_heavy(1,0);
        int t1,t2;
        for(int i=1;i<=n;++i)
        {
            if(ma[i]==up)
            {
                if(!s)
                t1=i;
                else
                t2=i;
                s++;
            }
        }
        //puts("1");
        if(s==1)
        {
            int v=edge[head[t1]].to;
            printf("%d %d\n",t1,v);
            printf("%d %d\n",t1,v);
        }
        else
        {
            int v;
            for(int i=head[t1];i;i=edge[i].next)
            {
                v=edge[i].to;
                if(v!=t2)
                break;
            }
            printf("%d %d\n",t1,v);
            printf("%d %d\n",t2,v);
        }
    }
    return 0;
}

D. Three Sequences

因为只有区间修改，所以可以考虑差分，以样例为例，a数组为 2,-1,7,3。差分结果为 -3,8,-4。b_i<=b_i+1<=b_i+2…，所以其差分之后是大于等于0的数，c_i>=c_i+1>=c_i+2…，其差分之后是小于等于0的数。对于a的每个差分，把负的差分值给c，正的差分值给b。例如-3，如果给b 1，给 c -4，肯定不如给b 0，给 c -3。这样才能保证b数组不会变的太大。对于答案，可以转化为 max(b₁+sum,c₁)，sum为差分值为正数的和。又 b₁+c₁=a₁，因此又可以将答案转化为 max(b₁+sum,a₁-b₁)，要使其最小，那么 b₁+sum=a₁-b₁。那么 b₁=(a₁-sum)/2。对于每次修改，利用差分维护差分正数数组即可。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=1e5+5;
ll a[N];
ll diff[N];
int n,q,l,r;
ll x,sum,d;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        scanf("%lld",&a[i]);
        diff[i]=a[i]-a[i-1];
        if(i>1&&diff[i]>0)
        sum+=diff[i];
    }
    d=(a[1]-sum)>>1;
    printf("%lld\n",max(d+sum,a[1]-d));
    scanf("%d",&q);
    while(q--)
    {
        scanf("%d%d%lld",&l,&r,&x);
        if(l==1)
        a[1]+=x;
        else
        {
            if(diff[l]>0&&x<0)
            sum-=min(-x,diff[l]);
            else
            if(diff[l]+x>0&&x>0)
            sum+=min(diff[l]+x,x);
            diff[l]+=x;
        }
        x=-x;
        if(r==n)
        ;
        else
        {
            if(diff[r+1]>0&&x<0)
            sum-=min(-x,diff[r+1]);
            else
            if(diff[r+1]+x>0&&x>0)
            sum+=min(diff[r+1]+x,x);
            diff[r+1]+=x;
        }
        d=(a[1]-sum)>>1;
        printf("%lld\n",max(d+sum,a[1]-d));
    }
    return 0;
}

E. Deleting Numbers

我们知道一个数可以分成好几个质因子相乘，如果我们能找到其所有质因子及其次幂就能确定这个数。n<=1e5，1e5的素数有9592个，与我们上界次数10000很接近了。我们知道，我们可以每次删除一个素数，及其所有倍数，再利用A 1操作。就可以知道其是否属于x的质因子。但是这样肯定会超过上界次数。所以我们考虑分块，将m个素数分为$\sqrt{m}$块，每块$\sqrt{m}$个素数，然后对于每块删完素数及其倍数后，再来一次 A 1操作，就可以找到 A的最小质因子在哪个块里，然后就从该块起始位置开始枚举，对于每个素数p，A p¹ A p² … 直到找到 A ^k|x , A^k+1 不能整除 x。我们就找到其指定次幂，我们所以，我们找在n内其一共有多少次幂 res，然后 A p^k 判断是否和res相等，如果不相等意味着，x有该质因子指定次幂所以没被删掉。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
#include <math.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
long long su[maxn],cnt;
bool isprime[maxn];
bool vis[maxn];
void prime(int n)//素数筛
{
    cnt=0;
    memset(isprime,1,sizeof(isprime));//初始化认为所有数都是素数
    isprime[0]=isprime[1]=0;//0和1不是素数
    for(long long i=2;i<=n;i++)
    {
        if(isprime[i])
        su[cnt++]=i;//保存素数i
        for(long long j=0;j<cnt&&su[j]*i<=n;j++)
        {
            isprime[su[j]*i]=0;//筛掉小于等于i的素数和i的积构成的合数
        }
    }
}
int ask(int i)
{
    printf("A %d\n",i);
    fflush(stdout);
    int x;
    scanf("%d",&x);
    return x;
}
int del(int i)
{
    printf("B %d\n",i);
    fflush(stdout);
    int x;
    scanf("%d",&x);
    return x;
}
int main()
{
    int n;
    scanf("%d",&n);
    prime(n);
    //cout<<cnt<<endl;
    int blk=sqrt(cnt);
    int ans=1;
    int res=n;
    int siz=cnt-1;
    int tmp=n+1;
    for(int i=0;i<=siz;++i)
    {
        int nxt=i+blk-1;
        nxt=min(nxt,siz);
        for(int j=i;j<=nxt;++j)
        {
            int x=su[j];
            del(x);
            for(int k=x;k<=n;k+=x)
            if(!vis[k])
            vis[k]=1,res--;
        }
        int num=ask(1);
        if(num!=res)
        {
            tmp=i;
            break;
        }
        i=nxt;
    }
    for(int i=tmp;i<=siz;++i)
    {
        int x=su[i];
        int res=0;
        if(ans*x>n)
        break;
        for(ll k=x;k<=n;k*=x)
        if(!vis[k])
        res++;
        for(ll k=x;k<=n;k*=x)
        {
            int num=ask(k);
            if(num!=res)
            ans*=x;
            else
            break;
            if(!vis[k])
            res--;
        }
    }
    printf("C %d\n",ans);
    fflush(stdout);
    return 0;
}