Codeforces - 848C - Goodbye Souvenir（CDQ分治）

最新推荐文章于 2021-09-22 19:59:20 发布

jinglinxiao

最新推荐文章于 2021-09-22 19:59:20 发布

阅读量470

点赞数

分类专栏： cdq分治

本文链接：https://blog.csdn.net/jinglinxiao/article/details/78034539

版权

cdq分治专栏收录该内容

1 篇文章 0 订阅

订阅专栏

Codeforces - 848C - Goodbye Souvenir

设数列中第 $i$ 个点前一个与 $i$ 颜色相同的点是 $prev[i]$ 。
那么每次询问的答案是

\sum i = l 且 p r e v [i] \geq l r i - p r e v [i]

$\sum_{i=l且prev[i] \geq l}^r i-prev[i]$ 。
如果考虑二维平面上点

(i,prev[i]) $(i,prev[i])$ 的权值为

i−prev[i] $i-prev[i]$ 。那么询问转化为求两端点分别为

(l,l)和(r,r) $(l,l)和(r,r)$ 的正方形内框住的点的权值和。
设

x=i $x=i$ ，

y=prev[i] $y=prev[i]$ ，且考虑到

y≤x $y \leq x$ 恒成立，那么上式转化为

\sum y i \geq l 且 x i \leq r v a l [i]

$\sum_{y_i \geq l 且 x_i \leq r} val[i]$
由于

yi≤xi≤r $y_i \leq x_i \leq r$ ，可以知道询问的区域应该是

x∈[0,r]且y∈[l,r] $x\in[0,r]且y\in[l,r]$ 这一段区域。
用树状数组处理

y $y$ 这一维，cdq处理

x $x$ 这一维就可以做了。

CDQ分治教程
cdq分治时间，每次处理左边的修改对右边询问的影响。
$prev$ 可以用 set 来维护。
时间复杂度最大为 $O(nlognlogn)$ 。

其实也可以直接求正方形。。如果这样的化把询问转化成四个角的询问就行了。。。这样应该更好理解一点。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5+7;
int n,a[N],m,vis[N],tot;
ll c[N],ans[N];
bool hq[N];
set<int> s[N];
struct Node
{
    //对于修改，x为横轴，y为纵轴，f为修改值，id为0
    //对于询问，x为右端点，y为左端点，f为0，id为询问id
    int x,y,f,id;
}Q[N*6],tmp[N*6];
void modify(int x,int y)
{
    if(a[x]==y) return ;
    auto i=s[a[x]].find(x);
    int p=0;
    if(i!=s[a[x]].begin())
    {
        --i;p=*i;++i;
        Q[m++]=(Node){x,p,p-x,0};
    }
    ++i;
    if(i!=s[a[x]].end())
    {
        Q[m++]=(Node){*i,x,x-*i,0};
        if(p) Q[m++]=(Node){*i,p,*i-p,0};
    }
    s[a[x]].erase(x);
    a[x]=y;
    i=s[y].insert(x).first;
    p=0;
    if(i!=s[y].begin())
    {
        --i;p=*i;++i;
        Q[m++]=(Node){x,p,x-p,0};
    }
    ++i;
    if(i!=s[y].end())
    {
        if(p) Q[m++]=(Node){*i,p,p-*i,0};
        Q[m++]=(Node){*i,x,*i-x,0};
    }
}
int lowbit(int x) { return x&-x; }
void update(int x,int y)
{
    for(;x<=n;x+=lowbit(x))
        if(vis[x]!=tot) vis[x]=tot,c[x]=y;
        else c[x]+=y;
}
ll query(int x)
{
    ll res=0;
    for(;x>0;x-=lowbit(x))
        res+=(vis[x]==tot?c[x]:0);
    return res;
}
void solve(int L,int R)
{
    if(L>=R) return ;
    int mid=(L+R)>>1;
    solve(L,mid);
    solve(mid+1,R);
    ++tot;
    int t1=L,t2=mid+1;
    for(int i=L;i<=R;++i)
    {
        if(t2>R||(t1<=mid&&Q[t1].x<=Q[t2].x)) //控制询问的点在(x,x) 的右下角
        {
            tmp[i]=Q[t1++];
            if(tmp[i].id==0) update(n-tmp[i].y+1,tmp[i].f);
        }
        else
        {
            tmp[i]=Q[t2++];
            if(tmp[i].id) ans[tmp[i].id]+=query(n-tmp[i].y+1); //询问 y=y上方的部分。
        }
    }
    for(int i=L;i<=R;++i) Q[i]=tmp[i];
}
//二维坐标系中若x=i，那么y=pre[i]。y<x恒成立。
int main()
{
    int q;
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        auto it=s[a[i]].insert(i).first;
        if(it!=s[a[i]].begin()) --it,Q[m++]=(Node){i,*it,i-*it,0};
    }
    for(int i=1;i<=q;++i)
    {
        int op,l,r;
        scanf("%d%d%d",&op,&l,&r);
        if(op==1) modify(l,r);
        else hq[i]=true,Q[m++]=(Node){r,l,0,i};
    }
    solve(0,m-1);
    for(int i=1;i<=q;++i) if(hq[i]) printf("%I64d\n",ans[i]);
    return 0;
}