Codeforces Round #851 (Div. 2)题解

A. One and Two

题意：对于给定长度为 $n$ 的序列 $a$ (已知 $a_i=1$ or $2$),寻找是否存在最小 $k$ 使得 $a_1\times a_2\times ...\times a_k=a_{k+1}\times a_{k+2}\times ...\times a_n$ 且 $1\le k\le n-1$ ，若不存在输出 $-1$
由于序列 $a$ 中只包含数 $1$ 或 $2$ ，只有 $2$ 影响序列连乘结果，故统计序列中 $2$ 的数量，当前缀中 $2$ 的数量等于后缀中 $2$ 的数量，则寻找到符合题意的答案
注意：由于 $2\le n\le 1000$ ，故不可采用直接计算前缀连乘积，会爆数据范围
代码如下:

#include <bits/stdc++.h>
using namespace std;
void solve()
{
    int n,k=-1;
    scanf("%d",&n);
    vector<int>ve(n+1);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&ve[i]);ve[i]=ve[i-1]+(ve[i]==2);
    }
    for(int i=1;i<n;i++)
    {
        if(ve[i]*2==ve[n])
        {
            k=i;break;
        }
    }
    printf("%d\n",k);
}
signed main() {
    int T=1;
    cin>>T;
    while(T--)
        solve();
	return 0;
}

B. Sum of Two Numbers

题意：给定一个正整数 $n$ ，要求寻找 $x$、$y$ ，满足 $x+y=n$ 并且 $x$ 各数位上数字之和与 $y$ 各数位上数字之和相差不超过 $1$
显然对于大部分情况，构造 $x=floor(\frac{n}{2})$ 与 $y=n-floor(\frac{n}{2})$ 符合题意。
但对于 $x=floor(\frac{n}{2})$ 与 $y=n-floor(\frac{n}{2})$ 之间存在进位情况，例如 $1999=999+1000$ ，需进行特殊处理，将 $x=floor(\frac{n}{2})$ 末尾 $9$ 数量进行统计，若为偶数进行均分，否则将一个 $9$ 分为 $4$ 与 $5$ 分别放置在 $x$ 与 $y$ 的最后一位上，然后其他 $9$ 进行均分即可
代码如下:

#include <bits/stdc++.h>
using namespace std;
void solve()
{
    int n,x,y;
    scanf("%d",&n);
    x=n/2;y=n-x;
    if(x<y)swap(x,y);
    if(x%10==0&&x!=y)
    {
        int ty=y,tt=0,num=0,tnow=9;
        while(ty%10==9)
        {
            tnow*=10;num++;ty/=10;
        }
        tnow/=10;
        for(int i=1;i*2<=num;i++)
        y-=tnow,tt+=tnow,tnow/=10;
        x+=tt;
        if(num&1)
            x+=4,y-=4;
    }
    printf("%d %d\n",x,y);
}
signed main() {
    int T=1;
    cin>>T;
    while(T--)
        solve();
	return 0;
}

C. Matching Numbers

题意：给定正整数 $n$ ,判断是否能构造 $n$ 个数对 $(a_i,b_i)$,满足 { a 1 , b 1 , a 2 , b 2 , . . . a n , b n } \{ a_{1},b_{1},a_{2},b_{2},...a_{n},b_{n} \} {a1​,b1​,a2​,b2​,...an​,bn​} 是 $1-2n$ 的排列，且 { a 1 + b 1 , a 2 + b 2 , . . . a n + b n } \{ a_{1}+b_{1},a_{2}+b_{2},...a_{n}+b_{n} \} {a1​+b1​,a2​+b2​,...an​+bn​} 排列后是公差为 $1$ 的等差数列
运用两次等差数列求和公式可得到，若存在符合题意的构造，则 { a 1 + b 1 , a 2 + b 2 , . . . a n + b n } \{ a_{1}+b_{1},a_{2}+b_{2},...a_{n}+b_{n} \} {a1​+b1​,a2​+b2​,...an​+bn​} 排列后所得等差数列首项为 $\frac{3(n+1)}{2}$ ，末项为 $\frac{5n+1}{2}$ 。显然由首项公式可得到若存在构造，则 $n$ 一定为奇数。接着这里采用 $a_i+b_i$ 由 $\frac{3(n+1)}{2}$ 至 $\frac{5n+1}{2}$ 依次构造，保证 $a_i<b_i$ 恒成立， $a_i$ 先运用奇数后运用偶数参与构造， $b_i$ 由 $a_i+b_i$ 总和减去 $a_i$ 的值得到

#include <bits/stdc++.h>
using namespace std;
void solve()
{
    int n,now;
    scanf("%d",&n);
    if(n%2==0)
    {
        puts("No");return ;
    }
    puts("Yes");
    now=(n+1)*3/2;
    for(int i=1;i<now-i;i+=2,now++)
        printf("%d %d\n",i,now-i);
    for(int i=2;now<=(5*n+1)/2;i+=2,now++)
        printf("%d %d\n",i,now-i);
}
signed main() {
    int T=1;
    cin>>T;
    while(T--)
        solve();
	return 0;
}

D. Moving Dots

题意:存在某点集，对于点集中的每个点，都将以相同的速度向离其最近的点移动(若某点与左右两点距离均相等，优先向左移动)，直到遇到其他点后停止。现给定 $n$ 个点坐标，求点集中的每个非空子集在移动操作后点聚集地数的总和
现对于点 $i$ 与点 $j$ $(i<j)$ ，计算点 $i$ 向右运动点 $j$ 向左运动后点 $i$ 与点 $j$ 相遇产生一个点聚集地可在几个点子集中对答案产生贡献
代码如下:

#include <bits/stdc++.h>
typedef long long LL;
const int MOD = 1e9+7;
using namespace std;
int n,pw2[3005],x[3005];
LL ans=0;
void solve()
{
    scanf("%d",&n);
    pw2[0]=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x[i]);
        pw2[i]=(pw2[i-1]*2ll)%MOD;
    }
    int l,r;
    for(int i=1;i<=n;i++)//现对于点i与点j(i<j)，计算点i向右运动、点j向左运动后点i与点j相遇产生一个点聚集地可在几个点子集中对答案产生贡献
    {
        l=i-1,r=(i+1)+1;//x[j]-x[i]不断增大，故l不断减小、r不断增大
        for(int j=i+1;j<=n;j++)
        {
            while(x[i]-x[l]<=x[j]-x[i]&&l>0)//寻找最右边的可添加点l不影响点i运动方向
                l--;
            while(x[r]-x[j]<x[j]-x[i]&&r<=n)//寻找最左边的可添加点r不影响点j运动方向
                r++;
            ans+=1ll*pw2[l]*pw2[n-r+1]%MOD;//[1,l]与[r,n]内点任选情况下的点子集均存在点i与点j相遇产生一个点聚集地的贡献
            ans%=MOD;
        }
    }
    printf("%lld\n",ans); 
}
signed main() {
    solve();
	return 0;
}

E. Sum Over Zero

题意：对于给定长度为 $n$ 的序列 $a$ ，将序列 $a$ 划分为若干个子段，寻找最大的子段长度总和，满足选择的每个子段总和大于等于 $0$
将 $pres$ 表示为 $a$ 的前缀和。若一个段 $[x,y]$ 可被选择，则满足 $pre{s}_{x-1}\le pre{s}_y$ 。
让我们把 $f_i$ 表示小于等于 $i$ 且符合题意的段的长度之和的最大值。如果没有结束于i的段符合题意， $f_i=f_{i-1}$ 。如果存在段 $[k+1,i]$ 符合题意， $f_i=ma{x}_{pre{s}_k\le pre{s}_i}(f_k+(i-(k+1)+1))=ma{x}_{pre{s}_k\le pre{s}_i}(f_k-k)+i$ 。整理可得：$f_i=max(f_{i-1},ma{x}_{pre{s}_k\le pre{s}_i}(f_k-k)+i)$，此时可 $O(N^2)$ 解决问题。
现在尝试用树状数组来加速 $dp$ 的转换。首先，在 $pre{s}_i$ 上使用坐标压缩，我们只需看到一个前缀的和是否比另一个大。我们将维护一个树状数组，将 $f_i-i$ 存储在 $pre{s}_i$ 的位置 $in{v}_i$ 。
$f_i=max(f_{i-1},ma{x}_{pre{s}_k\le pre{s}_i}(f_k-k)+i)$ ，则现在可以通过树状数组上的范围查询 $[0,in{v}_i]$ 解决 $ma{x}_{pre{s}_k\le pre{s}_i}(f_k-k)$。因此，对于每个 $i$ 可以在 $O(logN)$ 中解决 $f_i$ ，此时可 $O(NlogN)$ 解决问题。

#include <bits/stdc++.h>
typedef long long LL;
const int N = 1e6 + 5;
using namespace std;
int n;
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int c,int tr[])
{
    for(int i=x;i<=n;i+=lowbit(i))
        tr[i]=max(tr[i],c);
}
int query(int x,int tr[])
{
    int res=-2e5;
    for(int i=x;i;i-=lowbit(i))
        res=max(res,tr[i]);
    return res;
}
int tr[N],f[N],a[N],inv[N],ans=0;
LL pres[N];
void solve()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&pres[i]);
        pres[i]+=pres[i-1];
        tr[i]=-2e5;a[i]=i;
    }
    sort(a+1,a+n+1,[&](int i,int j){
        if(pres[i]!=pres[j])return pres[i]<pres[j];
        else return i<j;
    });
    for(int i=1;i<=n;i++)
        inv[a[i]]=i;
    for(int i=1;i<=n;i++)
    {
        if(pres[i]>=0ll)f[i]=i;
        else f[i]=f[i-1];
        f[i]=max(f[i],i+query(inv[i],tr));
        ans=max(ans,f[i]);
        add(inv[i],f[i]-i,tr);
    }
    printf("%d\n",ans);
}
signed main() {
    solve();
    return 0;
}