2018noip模拟 路径

【题目描述】
给定一个n个点每个点度数不超过2的无向图，支持两种操作
1.加入一条边[l,r],保证加入后图依然满足每个点度不超过2
2.求长度在[l,r]之间的不经过重复点的路径有多少条
注意:这里的路径长度定义为经过的点数而不是边数
【输入】
第一行三个正整数n,m,q表示点数,初始边数和操作数
接下来m行每行两个整数u,v表示u到v有一条边
接下来q行，每行三个整数op l r
【输出】
对于每个op=2输出答案
【样例】
path1.in path1.out
8 5 8
1 2
2 3
4 5
5 6
7 8
2 2 3
1 4 6
2 3 3
2 1 2
1 1 7
2 3 5
1 3 8
2 1 8 7
4
14
9
34
path2.in path2.out
4 0 5
1 1 2
1 2 3
1 3 4
1 4 1
2 1 4 16
【数据范围及约定】
对于100%的数据 1≤n,m,q≤500000 1≤l≤r≤n

线段树维护等差数列，但是合并比较麻烦QAQ，需要讨论。
1.若两个都是链，减两个等差数列，加一个等差数列，否则会重复。
2.若两个点在一个链上，相当于成环，只需要减一个等差数列，加一个等差数列。
然后你就发现woc为什么这么难写

代码

#include<bits/stdc++.h>
#define N 500005
#define ll long long
using namespace std;
int n,m,q,u,v,l,r,op,fa[N],siz[N];
ll t[N<<2],a1[N<<2],d[N<<2];
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
void update(int l,int r,int o,ll A1,ll D)
{
    int len=r-l+1;
    t[o]+=(1ll*(A1+A1+1ll*D*(len-1))*len/2);
    a1[o]+=A1,d[o]+=D;
}
void pushdown(int o,int l,int r,int mid)
{
    ll A1=a1[o],D=d[o];
    update(l,mid,o<<1,A1,D);
    update(mid+1,r,o<<1|1,A1+1ll*(mid-l+1)*D,D);
    a1[o]=0,d[o]=0;
}
void add(int l,int r,int o,int x,int y,ll D,ll A1)
{
    if(l>=x&&r<=y){update(l,r,o,A1,D);return;}
    int mid=l+r>>1,len;
    if(a1[o]||d[o]) pushdown(o,l,r,mid);
    if(x<=mid) len=min(mid,y)-x+1,add(l,mid,o<<1,x,min(mid,y),D,A1);
    if(y>mid) add(mid+1,r,o<<1|1,max(mid+1,x),y,D,A1+1ll*len*D);
    t[o]=t[o<<1]+t[o<<1|1];
}
void merge(int x,int y)
{
    int p=find(x),q=find(y);
    if(p!=q)
    {
        add(1,n,1,1,siz[p],1,-siz[p]);
        add(1,n,1,1,siz[q],1,-siz[q]);
        if(siz[p]>siz[q]) swap(p,q);
        fa[p]=q,siz[q]+=siz[p];
        add(1,n,1,1,siz[q],-1,siz[q]);
    }
    else
    {
        add(1,n,1,1,siz[p],1,-siz[p]);
        add(1,n,1,1,siz[p],0,siz[p]);
    }
}
ll query(int l,int r,int o,int x,int y)
{
    if(l>=x&&r<=y) return t[o];
    int mid=l+r>>1;
    ll re=0;
    if(a1[o]||d[o]) pushdown(o,l,r,mid);
    if(x<=mid) re+=query(l,mid,o<<1,x,y);
    if(y>mid) re+=query(mid+1,r,o<<1|1,x,y);
    return re;
}
int main()
{
    scanf("%d%d%d",&n,&m,&q);
    for(int i=1;i<=n;i++) fa[i]=i,siz[i]=1;
    add(1,n,1,1,1,0,n);
    for(int i=1;i<=m;i++) scanf("%d%d",&u,&v),merge(u,v);
    while(q--)
    {
        scanf("%d%d%d",&op,&l,&r);
        if(op==1) merge(l,r);
        else printf("%lld\n",query(1,n,1,l,r));
    }
    return 0;
}