第 1 场周赛

Posted 2021-05-31 caifxh

3577. 选择数字

签到。

int main()
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>a[i],vis[a[i]]=true;
    cin>>m;
    for(int i=0;i<m;i++) cin>>b[i],vis[b[i]]=true;
    
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
            if(!vis[a[i]+b[j]])
            {
                cout<<a[i]<<\' \'<<b[j]<<endl;
                return 0;
            }
}

3578. 最大中位数

贪心。

首先将给定数组排序。

如果我们想增大中位数，那么一定是先让\\(mid\\)位置上的数增加至与\\(mid+1\\)位置上的数相等(这样中位数变为\\(mid+1\\)位置上的数)，之后再同时增加\\(mid\\)与\\(mid+1\\)位置上的数直到与\\(mid+2\\)位置上的数相等(这样中位数变为\\(mid+2\\)位置上的数)\\(\\cdots\\)以此类推至第\\(n\\)个位置。

如果最后\\(k\\)还有剩余，则平均分配给\\(mid \\sim n\\)位置上的数。

const int N = 2e5+10;

int a[N];
int n,k;

int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++) cin>>a[i];
    
    sort(a,a+n);

    int cnt=1;
    for(int i=n/2+1;i<n;i++)
    {
        int x=min(k/cnt,a[i]-a[n/2]);
        k-=cnt*x;
        a[n/2]+=x;
        cnt++;
    }
    
    if(k) a[n/2]+=k/cnt;
    
    cout<<a[n/2]<<endl;
    
    return 0;
}

3579. 数字移动

统计环(连通块)内点的数量。

每个数字第一次回到自己的次数，就是其所在环的长度。

const int N = 2e5+10;
int p[N];
int cnt[N];
int n;

int find(int x)
{
    if(x != p[x]) p[x]=find(p[x]);
    return p[x];
}

int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n;
        
        for(int i=1;i<=n;i++) p[i]=i,cnt[i]=1;
        
        for(int i=1;i<=n;i++)
        {
            int j;
            cin>>j;
            int pi=find(i),pj=find(j);
            if(pi != pj)
            {
                p[pi]=p[j];
                cnt[pj]+=cnt[pi];
            }
        }
        
        for(int i=1;i<=n;i++)
        {
            int pi=find(i);
            cout<<cnt[pi]<<\' \';
        }
        cout<<endl;
    }
}