SDOI2017 R2泛做

由于各种原因,在bzoj上我day1的题一题都没过,所以这里就直接贴loj的链接好了。

D1T1 龙与地下城

中心极限定理。

https://en.wikipedia.org/wiki/Central_limit_theorem

由于某些原因这里的公式挂了...直接看维基吧...

要算积分可以用标准库里的erf:https://en.wikipedia.org/wiki/Error_function

对于正态分布image,值落在[-x,x]的概率为image,所以落在[0,x]的概率就是这个值的一半,这玩意儿就可以直接当做不定积分了。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <string>
#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <sstream>
#include <stack>
#include <iomanip>
using namespace std;
#define pb push_back
#define mp make_pair
typedef pair<int,int> pii;
typedef long long ll;
typedef double ld;
typedef vector<int> vi;
#define fi first
#define se second
#define fe first
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define Edg int M=0,fst[SZ],vb[SZ],nxt[SZ];void ad_de(int a,int b){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;}void adde(int a,int b){ad_de(a,b);ad_de(b,a);}
#define Edgc int M=0,fst[SZ],vb[SZ],nxt[SZ],vc[SZ];void ad_de(int a,int b,int c){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;vc[M]=c;}void adde(int a,int b,int c){ad_de(a,b,c);ad_de(b,a,c);}
#define es(x,e) (int e=fst[x];e;e=nxt[e])
#define esb(x,e,b) (int e=fst[x],b=vb[e];e;e=nxt[e],b=vb[e])
#define VIZ {printf("digraph G{\n"); for(int i=1;i<=n;i++) for es(i,e) printf("%d->%d;\n",i,vb[e]); puts("}");}
#define VIZ2 {printf("graph G{\n"); for(int i=1;i<=n;i++) for es(i,e) if(vb[e]>=i)printf("%d--%d;\n",i,vb[e]); puts("}");}
#define SZ 666666
int x,y,a,b,T;
ld f[2333333];
void sol()
{
    scanf("%d%d",&x,&y);
    if((ll)x*y*y<=51200000)
    {
        for(int i=0;i<=x*y;++i) f[i]=1;
        for(int i=1;i<=y;++i)
        {
            for(int j=x*y;j>=x;--j)
                f[j]-=f[j-x];
            for(int j=0;j<=x*y;++j)
            {
                f[j]/=x;
                if(j) f[j]+=f[j-1];
            }
        }
        for(int i=1;i<=10;++i)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            ld ans=f[r];
            if(l) ans-=f[l-1];
            printf("%.10lf\n",ans);
        }
    }
    else
    {
        int n=y; ld m=(x-1)/2.0,s=(x*(ld)x-1)/12;
        ld k=-n*m,b=sqrt(n*s*2);
        for(int i=1;i<=10;++i)
        {
            ld l,r;
            scanf("%lf%lf",&l,&r);
            l-=0.5,r+=0.5;
            l=(l+k)/b; r=(r+k)/b;
            printf("%.10lf\n",(erf(r)-erf(l))/2);
        }
    }
}
int main()
{
    scanf("%d",&T);
    while(T--) sol();
}

D1T2 苹果树

首先如果是树上有父亲限制的01背包,可以参见http://www.cnblogs.com/zzqsblog/p/5537440.html

现在不是01背包了,为了方便起见,一个ai>1的点我们拆成一个ai=1和一个ai比原来少1的点,把这个新点挂在ai=1的点下面进行dp。

多重背包显然可以单调队列优化。

现在剩的就是那个莫名其妙的k。我们枚举一条到叶子的链(注意这里我们忽略ai>1的点),那么我们的要求就是在除了这条链以外最多只能有k个点。

我们进行一次这个背包,可以得到一个dfs序,即我们可以得到一个dfs序后缀的背包信息,那我们把每个点的儿子反过来再得到一个dfs序,把两个dfs序这两个后缀并在一起,恰好就是整棵树去掉这条链,泛化背包类似搞搞就行。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <string>
#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <sstream>
#include <stack>
#include <iomanip>
using namespace std;
#define pb push_back
#define mp make_pair
typedef pair<int,int> pii;
typedef long long ll;
typedef double ld;
typedef vector<int> vi;
#define fi first
#define se second
#define fe first
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define Tree vector<int> ch[SZ];\
void setf(int x,int f) {ch[f].pb(x);}
#define esb(x,e,b) \
(unsigned __=0,b=(__<ch[x].size())?(ch[x][__]):0;\
__<ch[x].size();++__,b=(__<ch[x].size())?(ch[x][__]):0)
#define VIZ {printf("digraph G{\n"); for(int i=1;i<=n;i++) for es(i,e) printf("%d->%d;\n",i,vb[e]); puts("}");}
#define VIZ2 {printf("graph G{\n"); for(int i=1;i<=n;i++) for es(i,e) if(vb[e]>=i)printf("%d--%d;\n",i,vb[e]); puts("}");}
#define SZ 1234567
namespace FF
{
char ch,B[1<<20],*S=B,*T=B;
#define getc() (S==T&&(T=(S=B)+fread(B,1,1<<20,stdin),S==T)?0:*S++)
#define isd(c) (c>='0'&&c<='9')
int aa,bb;int F(){
    while(ch=getc(),!isd(ch)&&ch!='-');ch=='-'?aa=bb=0:(aa=ch-'0',bb=1);
    while(ch=getc(),isd(ch))aa=aa*10+ch-'0';return bb?aa:-aa;
}
}
#define gi FF::F()
int T,n,k;
#define max(a,b) (((a)>(b))?(a):(b))
#define cmax(a,b) (((a)<(b))?((a)=(b)):0)
Tree
int m1[52111111],m2[52111111];
#define R(x,a) (x+(a)*(k+1))
inline void pack(int*r,int v,int x) //x个重1价值v
{
    if(!x);
    else if(x==1)
        for(int i=k;i>=1;--i)
            cmax(r[i],r[i-1]+v);
    else if(x>=k)
        for(int i=1;i<=k;++i)
            cmax(r[i],r[i-1]+v);
    else
    {
        static pii qs[SZ];
        int h=0,t=0;
        for(int j=1;j<=k;j++)
        {
            int cv=r[j]-j*v;
            while(h!=t&&qs[t-1].fi<cv) --t;
            qs[t++]=pii(cv,j);
            while(qs[h].se<j-x) ++h;
            r[j]=qs[h].fi+j*v;
        }
    }
}
int fa[SZ],a[SZ],v[SZ],dep[SZ],fc[SZ];
int d1[SZ],c1=0,l1[SZ],d2[SZ],c2=0,l2[SZ],r1[SZ],r2[SZ];
inline void dfs(int x,int*d,int&c,int*l,int*r)
{
    d[x]=++c; r[c]=x;
    for esb(x,e,b)
        dep[b]=dep[x]+1,fc[b]=fc[x]+v[b],
        dfs(b,d,c,l,r);
    l[x]=c;
}
inline void work(int*d,int&c,int*l,int*p,int*r)
{
    for(int i=c;i>=1;--i)
    {
        int x=r[i],*A=R(p,d[x]),*B=R(p,l[x]+1),*C=R(p,d[x]+1);
        for(int s=1;s<=k;s++)
            A[s]=max(B[s],C[s-1]+v[x]);
        pack(A,v[x],a[x]-1);
    }
}
bool good[SZ];
void sol()
{
    n=gi,k=gi; int N=n; c1=c2=0;
    for(int i=1;i<=n+n;++i)
        good[i]=1,ch[i].clear();
    ll tot=0;
    for(int i=1;i<=N;++i)
    {
        fa[i]=gi,a[i]=gi,v[i]=gi; tot+=a[i];
        if(fa[i]) setf(i,fa[i]),good[fa[i]]=0;
        if(a[i]>1)
        {
            ++n; setf(n,i); a[n]=a[i]-1; good[n]=0;
            v[n]=v[i]; fa[n]=i; a[i]=1;
        }
    }
    int tt=(n+2)*(k+1);
    memset(m1,0,tt*sizeof(int));
    memset(m2,0,tt*sizeof(int));
    dep[1]=1; fc[1]=v[1];
    dfs(1,d1,c1,l1,r1);
    work(d1,c1,l1,m1,r1);
    for(int i=1;i<=n;++i)
        reverse(ch[i].begin(),ch[i].end());
    dfs(1,d2,c2,l2,r2);
    work(d2,c2,l2,m2,r2);
    int ans=0;
    for(int i=1;i<=n;++i)
    {
        if(!good[i]) continue;
        int x=min(tot-dep[i],(ll)k);
        int*f=R(m1,d1[i]+1),*g=R(m2,l2[i]+1);
        for(int s=0;s<=x;++s)
            cmax(ans,fc[i]+f[s]+g[x-s]);
    }
    printf("%d\n",ans);
}
int main()
{
    T=gi; while(T--) sol();
}

D1T3 切树游戏

这里讲一下本人常数很大的做法。首先众所周知如果有一个序列A和一个序列B,A和B做异或卷积等于C,那么fwt(A)*fwt(B)(点乘)=fwt(C)。

我们记v[i]为单点值fwt之后的数组,那么有如下的dp方程:

$s[i]=v[i]\prod_cs[c]+1$(c是i的孩子,+1是加上空子树的情况)

最后我们要求的是$\sum_{i=1}^ns[i]$。求出这玩意儿之后ifwt回去,把n个空子树扣掉就有答案了。

考虑记$g[i]=v[i]\prod_{c'}s[c']$,c'是i的轻孩子。

这玩意儿可以支持修改,只要修改的时候跳重链就好。

现在的问题就是要求出重链顶端的s值以及把一整条重链的s值求和。

我们只考虑重链的话,可以发现$s[i]=s[w]g[i]+1$,w是i的重孩子,如果我们把g预处理成矩阵(每个元素都是向量),然后这玩意儿是可以矩乘的,同时也可以顺便把s求和。然后线段树维护一下就好了。注意到矩阵有一些位置是不变的可以减小常数。

这玩意儿还是常数爆炸了,那我也没有办法,复杂度都是一样的...

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <string>
#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <sstream>
#include <stack>
#include <iomanip>
using namespace std;
#define pb push_back
#define mp make_pair
typedef pair<int,int> pii;
typedef long long ll;
typedef double ld;
typedef vector<int> vi;
#define fi first
#define se second
#define fe first
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define Edg int M=0,fst[SZ],vb[SZ],nxt[SZ];void ad_de(int a,int b){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;}void adde(int a,int b){ad_de(a,b);ad_de(b,a);}
#define Edgc int M=0,fst[SZ],vb[SZ],nxt[SZ],vc[SZ];void ad_de(int a,int b,int c){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;vc[M]=c;}void adde(int a,int b,int c){ad_de(a,b,c);ad_de(b,a,c);}
#define es(x,e) (int e=fst[x];e;e=nxt[e])
#define esb(x,e,b) (int e=fst[x],b=vb[e];e;e=nxt[e],b=vb[e])
#define VIZ {printf("digraph G{\n"); for(int i=1;i<=n;i++) for es(i,e) printf("%d->%d;\n",i,vb[e]); puts("}");}
#define VIZ2 {printf("graph G{\n"); for(int i=1;i<=n;i++) for es(i,e) if(vb[e]>=i)printf("%d--%d;\n",i,vb[e]); puts("}");}
#define SZ 66666
#pragma pack(1)
int n,m,q; Edg
#define SZ 33333
const int MOD=10007;
int inv[MOD+3];
struct num
{
short x; short z;
num() {x=z=1;}
num(int p) {(p%=MOD)?(x=p,z=0):(x=z=1);}
inline operator int() const {return z?0:x;}
};
inline num operator * (num a,num b)
{a.x=int(a.x)*b.x%MOD; a.z+=b.z; return a;}
inline num operator / (num a,num b)
{a.x=int(a.x)*inv[b.x]%MOD; a.z-=b.z; return a;}
template<class T>
struct arr
{
T g[129];
arr() {for(int i=0;i<m;++i) g[i]=0;}
void fwt()
{
    for(int i=1;i<m;i<<=1)
    {
        for(int j=0;j<m;++j)
        {
            if(j&i) continue;
            int a=g[j],b=g[j|i];
            g[j]=(a+b)%MOD;
            g[j|i]=(a+MOD-b)%MOD; 
        }
    }
}
void ifwt() //only for <short>
{
    fwt();
    for(int i=0;i<m;++i)
        g[i]=g[i]*(int)inv[m]%MOD;
}
template<class T1> 
operator arr<T1> ()
{
    arr<T1> r;
    for(int i=0;i<m;++i) r.g[i]=g[i];
    return r;
}
};
arr<num> one_n,zero_n;
arr<short> one_i,zero_i;
template<class T>
arr<T> operator + (arr<T> a,arr<T> b)
{
    for(int i=0;i<m;++i)
        a.g[i]=(int(a.g[i])+b.g[i])%MOD;
    return a;
}
template<class T>
arr<T> operator - (arr<T> a,arr<T> b)
{
    for(int i=0;i<m;++i)
        a.g[i]=(int(a.g[i])+MOD-b.g[i])%MOD;
    return a;
}
arr<num> operator * (arr<num> a,arr<num> b)
{
    for(int i=0;i<m;++i)
        a.g[i]=a.g[i]*b.g[i];
    return a;
}
arr<short> operator * (arr<short> a,arr<short> b)
{
    for(int i=0;i<m;++i)
        a.g[i]=a.g[i]*(int)b.g[i]%MOD;
    return a;
}
arr<num> operator / (arr<num> a,arr<num> b)
{
    for(int i=0;i<m;++i)
        a.g[i]=a.g[i]/b.g[i];
    return a;
}
struct mat
{
arr<short> g00,g01,g20,g21;
/*
x y 0
0 1 0
z s 1
*/
};
mat operator * (mat a,mat b)
{
    mat p;
    p.g00=a.g00*b.g00;
    p.g01=a.g00*b.g01+a.g01;
    p.g20=a.g20*b.g00+b.g20;
    p.g21=a.g20*b.g01+a.g21+b.g21;
    return p;
}
//pair(w,s)
pair<arr<short>,arr<short> > cp(mat s)
{
    return make_pair(s.g00+s.g01,s.g20+s.g21);
}
inline mat gs(arr<short> x)
{
    mat s;
    s.g01=s.g21=one_i;
    s.g00=s.g20=x; return s;
}
arr<short> sw,isw; //sum of w[i]
arr<num> v[SZ],g[SZ],w[SZ]; mat gm[SZ]; //重链顶端的w保证正确
int fa[SZ],son[SZ],sz[SZ],top[SZ],fi[SZ],bt[SZ],C=0;
arr<num> ts[130];
void dfs(int x,int f=0)
{
    fa[x]=f; sz[x]=1;
    for esb(x,e,b) if(b!=f)
    {
        dfs(b,x),sz[x]+=sz[b];
        if(sz[b]>sz[son[x]]) son[x]=b;
    }
}
void dfs2(int x,int t,int f=0)
{
    top[x]=t; fi[x]=++C; bt[t]=C;
    if(!son[x]) return;
    dfs2(son[x],t,x);
    for esb(x,e,b) if(b!=f&&b!=son[x])
        dfs2(b,b,x);
}
const int Z=32768;
mat seg[Z+Z+3],I;
void dfs3(int x,int f=0,int d=0)
{
    for esb(x,e,b) if(b!=f) dfs3(b,x,d+1);
    g[x]=v[x];
    for esb(x,e,b) if(b!=f) g[x]=g[x]*w[b];
    w[x]=g[x]+one_n; if(son[x]) g[x]=g[x]/w[son[x]];
    seg[Z+fi[x]]=gm[x]=gs(g[x]); sw=sw+arr<short>(w[x]);
}
void upd(int x)
{
    for((x+=Z)>>=1;x;x>>=1) seg[x]=seg[x+x]*seg[x+x+1];
}
mat qry(int l,int r)
{
    mat rs=I; static int st[SZ]; int sn=0;
    for(l+=Z-1,r+=Z+1;l^r^1;l>>=1,r>>=1)
    {
        if(~l&1) rs=rs*seg[l^1];
        if(r&1) st[++sn]=r^1;
    }
    while(sn) rs=rs*seg[st[sn--]];
    return rs;
}
int main()
{
    inv[1]=1;
    for(int i=2;i<MOD;++i)
        inv[i]=(MOD-MOD/i)*inv[MOD%i]%MOD;
    scanf("%d%d",&n,&m);
    static int a[SZ];
    for(int i=1;i<=n;++i) scanf("%d",a+i);
    for(int i=1;i<n;++i)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        adde(a,b);
    }
    for(int i=0;i<m;++i) one_n.g[i]=1,zero_n.g[i]=0;
    for(int i=0;i<m;++i) one_i.g[i]=1,zero_i.g[i]=0;
    I.g00=one_i; I.g01=I.g20=I.g21=zero_i;
    sw=zero_i; dfs(1); dfs2(1,1);
    for(int i=0;i<m;++i)
        ts[i]=zero_n, ts[i].g[i]=1, ts[i].fwt();
    for(int i=1;i<=n;++i) v[i]=ts[a[i]];
    dfs3(1); isw=sw; isw.ifwt();
    for(int i=Z-1;i>=1;--i)
        seg[i]=seg[i+i]*seg[i+i+1];
    int q; scanf("%d",&q);
    while(q--)
    {
        char s[10]; scanf("%s",s);
        if(s[0]=='Q')
        {
            int a; scanf("%d",&a);
            ll ans=isw.g[a];
            if(!a) ans-=n;
            ans=(ans%MOD+MOD)%MOD;
            printf("%d\n",int(ans));
        }
        else
        {
            int x,y;
            scanf("%d%d",&x,&y);
            mat r=qry(fi[top[x]],bt[top[x]]);
            pair<arr<short>,arr<short> > s=cp(r); sw=sw-s.se;
            g[x]=g[x]/v[x]*ts[y]; v[x]=ts[y];
            seg[Z+fi[x]]=gm[x]=gs(g[x]); upd(fi[x]); x=top[x];
            r=qry(fi[top[x]],bt[top[x]]);
            s=cp(r); sw=sw+s.se;
            while(fa[x])
            {
                int y=fa[x]; g[y]=g[y]/w[x]*arr<num>(s.fi);
                w[x]=s.fi; x=y;
                r=qry(fi[top[x]],bt[top[x]]);
                s=cp(r); sw=sw-s.se;
                seg[Z+fi[x]]=gm[x]=gs(g[x]); upd(fi[x]); x=top[x];
                r=qry(fi[top[x]],bt[top[x]]);
                s=cp(r); sw=sw+s.se;
            }
            w[x]=s.fi; isw=sw; isw.ifwt();
        }
    }
}

D2T1 天才黑客

好像这种做法不是最优的,比标解多了一个log的样子。

考虑把边拆点,拆成入点和出点,中间连边权。然后1号点连上入点,出点连上边另一个端点。接下来我们要做的就是把同一个点的入边的出点 和 出边的入点 之间连边。

考虑把这些点提取出来在字典树上建个虚树,然后连边的话考虑lca为一个点的有哪些点对,要么是子树之间的,要么是自己和整棵树之间的,在dfs序上建两棵线段树辅助一下连边就好了。

代码难度还是挺高的...

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <string>
#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <sstream>
#include <stack>
#include <iomanip>
using namespace std;
#define pb push_back
#define mp make_pair
typedef long long ll;
typedef double ld;
typedef vector<int> vi;
#define fi first
#define se second
#define fe first
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define Edg int M=0,fst[SZ],vb[SZ],nxt[SZ];void ad_de(int a,int b){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;}void adde(int a,int b){ad_de(a,b);ad_de(b,a);}
#define Edgc int M=0,fst[SZ],vb[SZ],nxt[SZ],vc[SZ];void ad_de(int a,int b,int c){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;vc[M]=c;}void adde(int a,int b,int c){ad_de(a,b,c);ad_de(b,a,c);}
#define es(x,e) (int e=fst[x];e;e=nxt[e])
#define esb(x,e,b) (int e=fst[x],b=vb[e];e;e=nxt[e],b=vb[e])
#define VIZ {printf("digraph G{\n"); for(int i=1;i<=n;i++) for es(i,e) printf("%d->%d;\n",i,vb[e]); puts("}");}
#define VIZ2 {printf("graph G{\n"); for(int i=1;i<=n;i++) for es(i,e) if(vb[e]>=i)printf("%d--%d;\n",i,vb[e]); puts("}");}
#define SZ 2000000
int N=0;
#define SS 3500000
int M=0,fst[SS],vb[SS],nxt[SS],vc[SS];
void ad_de(int a,int b,int c=0){if(a&&b);else return;
//cout<<a<<"->"<<b<<"[label="<<c<<"]\n";
++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b; vc[M]=c;}
void ad_fe(int a,int b,int c=0) {ad_de(b,a,c);}
typedef pair<ll,int> pii;
ll dist[SS];
#define fi first
#define se second
void dj()
{
    priority_queue<pii,vector<pii>,greater<pii> > pq;
    for(int i=1;i<=N;++i) dist[i]=1e18;
    dist[1]=0; pq.push(pii(0,1));
    while(!pq.empty())
    {
        pii t=pq.top(); pq.pop();
        if(t.fi!=dist[t.se]) continue;
        int x=t.se;
        for(int e=fst[x];e;e=nxt[e])
        {
            int b=vb[e];
            if(dist[b]<=dist[x]+vc[e])
                continue;
            dist[b]=dist[x]+vc[e];
            pq.push(pii(dist[b],b));
        }
    }
}
template<void f(int,int,int)>
struct seg
{
int Z,s[SZ];
void init(int n)
{
    Z=1; while(Z<=n+3) Z<<=1;
    for(int i=1;i<=Z+Z;++i) s[i]=++N;
    for(int i=Z-1;i>=1;--i)
        f(s[i],s[i+i],0),f(s[i],s[i+i+1],0);
}
int gr(int l,int r)
{
    int p=++N;
    if(l>r) return p;
    for(l+=Z-1,r+=Z+1;l^r^1;l>>=1,r>>=1)
    {
        if(~l&1) f(p,s[l^1],0);
        if(r&1) f(p,s[r^1],0);
    }
    return p;
}
};
seg<ad_de> zs;
seg<ad_fe> fs;
int fl[SZ];
namespace vt
{
Edg
int C,df[SZ],ls[SZ];
void dfs(int x)
{
    df[x]=++C;
    for esb(x,e,b) dfs(b);
    ls[x]=C;
}
void dfs2(int x)
{
    int l=df[x],r=df[x];
    ::ad_de(fs.gr(l,r),zs.gr(l,r),fl[x]);
    for esb(x,e,b) dfs2(b);
    for esb(x,e,b)
    {
        ::ad_de(fs.gr(l,r),zs.gr(df[b],ls[b]),fl[x]),
        ::ad_de(fs.gr(df[b],ls[b]),zs.gr(l,r),fl[x]);
        r=ls[b];
    }
}
void doit(int r)
{
    C=0; dfs(r); zs.init(C);
    fs.init(C); dfs2(r);
}
}
namespace rt
{
Edg
int n,dfn[SZ],C=0,sz[SZ],son[SZ];
void dfs(int x,int f=0)
{
    if(f) fl[x]=fl[f]+1; dfn[x]=++C; sz[x]=1; son[x]=0;
    for esb(x,e,b)
    {
        dfs(b,x); sz[x]+=sz[b];
        if(sz[b]>sz[son[x]]) son[x]=b;
    }
}
int top[SZ],fa[SZ];
void dfs2(int x,int t,int f=0)
{
    fa[x]=f; top[x]=t;
    if(!son[x]) return;
    dfs2(son[x],t,x);
    for esb(x,e,b) if(b!=son[x]) dfs2(b,b,x);
}
int lca(int a,int b)
{
    while(top[a]!=top[b])
        if(fl[top[a]]<fl[top[b]]) b=fa[top[b]];
        else a=fa[top[a]];
    return (fl[a]<fl[b])?a:b;
}
bool cmp(int a,int b) {return dfn[a]<dfn[b];}
void init()
{
    C=M=0;
    for(int i=1;i<=n;++i) fst[i]=0;
    for(int i=1;i<n;++i)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        ad_de(a,b);
    }
    dfs(1);dfs2(1,1);
}
int vs[SZ],st[SZ],vn=0,stn=0,vfa[SZ];
void buildrt(int*g,int gn)
{
    sort(g+1,g+1+gn,cmp);
    gn=unique(g+1,g+1+gn)-g-1;
    vn=stn=0;
    for(int i=1;i<=gn;i++) vs[++vn]=g[i],vfa[g[i]]=0;
    for(int i=1;i<=gn;i++)
    {
        int x=g[i];
        if(!stn) {st[++stn]=x; vfa[x]=0; continue;}
        int lca=rt::lca(x,st[stn]);
        for(;fl[st[stn]]>fl[lca];--stn)
            if(fl[st[stn-1]]<=fl[lca])
                vfa[st[stn]]=lca;
        if(st[stn]!=lca)
        {
            vs[++vn]=lca;
            vfa[lca]=st[stn];
            st[++stn]=lca;
        }
        vfa[x]=lca; st[++stn]=x;
    }
    int ro=0; vt::M=0;
    for(int i=1;i<=vn;++i)
        vt::fst[vs[i]]=0;
    for(int i=1;i<=vn;++i)
        if(vfa[vs[i]]) vt::ad_de(vfa[vs[i]],vs[i]); else ro=vs[i];
    vt::doit(ro);
}
}
namespace Sol
{
int n,m,k,ea[SZ],eb[SZ],ec[SZ],ed[SZ],i1[SZ],i2[SZ];
namespace Z{Edgc}
namespace F{Edgc}
void sol()
{
    scanf("%d%d%d",&n,&m,&k); Z::M=F::M=0;
    ::M=0; ::N=n; memset(::fst,0,sizeof ::fst);
    for(int i=1;i<=n;++i) Z::fst[i]=F::fst[i]=0;
    for(int i=1;i<=m;++i)
    {
        scanf("%d%d%d%d",ea+i,eb+i,ec+i,ed+i);
        Z::ad_de(ea[i],eb[i],i);
        F::ad_de(eb[i],ea[i],i);
        i1[i]=++N; i2[i]=++N;
        ad_de(i1[i],i2[i],ec[i]);
        if(ea[i]==1) ad_de(1,i1[i]);
        ad_de(i2[i],eb[i]);
    }
    rt::n=k; rt::init();
    for(int i=1;i<=n;++i)
    {
        static int tmp[SZ]; int tn=0;
        for(int e=Z::fst[i];e;e=Z::nxt[e])
            tmp[++tn]=ed[Z::vc[e]];
        for(int e=F::fst[i];e;e=F::nxt[e])
            tmp[++tn]=ed[F::vc[e]];
        rt::buildrt(tmp,tn);
        for(int e=F::fst[i];e;e=F::nxt[e])
            ad_de(i2[F::vc[e]],fs.s[vt::df[ed[F::vc[e]]]+fs.Z]);
        for(int e=Z::fst[i];e;e=Z::nxt[e])
            ad_de(zs.s[zs.Z+vt::df[ed[Z::vc[e]]]],i1[Z::vc[e]]);
    }
    dj();
    for(int i=2;i<=n;++i)
        printf("%lld\n",dist[i]);
}
}
int main()
{
    int T; scanf("%d",&T);
    for(int i=1;i<=T;++i) Sol::sol();
}

D2T2 遗忘的集合

euler transform裸题,详见 http://l0nl1f3.leanote.com/post/Euler-Transform-discussion

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <string>
#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <sstream>
#include <stack>
#include <iomanip>
using namespace std;
#define pb push_back
#define mp make_pair
typedef pair<int,int> pii;
typedef long long ll;
typedef double ld;
typedef vector<int> vi;
#define fi first
#define se second
#define fe first
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define Edg int M=0,fst[SZ],vb[SZ],nxt[SZ];void ad_de(int a,int b){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;}void adde(int a,int b){ad_de(a,b);ad_de(b,a);}
#define Edgc int M=0,fst[SZ],vb[SZ],nxt[SZ],vc[SZ];void ad_de(int a,int b,int c){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;vc[M]=c;}void adde(int a,int b,int c){ad_de(a,b,c);ad_de(b,a,c);}
#define es(x,e) (int e=fst[x];e;e=nxt[e])
#define esb(x,e,b) (int e=fst[x],b=vb[e];e;e=nxt[e],b=vb[e])
#define VIZ {printf("digraph G{\n"); for(int i=1;i<=n;i++) for es(i,e) printf("%d->%d;\n",i,vb[e]); puts("}");}
#define VIZ2 {printf("graph G{\n"); for(int i=1;i<=n;i++) for es(i,e) if(vb[e]>=i)printf("%d--%d;\n",i,vb[e]); puts("}");}
#define SZ 666666

//mtt
#define REP(i, a, b) for (int i = (a), _end_ = (b); i < _end_; ++i)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define mp make_pair
#define x first
#define y second
#define pb push_back
#define ALL(x) (x).begin(), (x).end()

template<typename T> inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

typedef long long LL;

const int oo = 0x3f3f3f3f;

int Mod = 1e9 + 7;
#define MOD Mod
ll qp(ll a,ll b)
{
    ll x=1;
    while(b)
    {
        if(b&1) x=x*a%MOD;
        a=a*a%MOD; b>>=1;
    }
    return x;
}
const int max0 = 524288;

struct comp
{
    double x, y;

    comp(): x(0), y(0) { }
    comp(const double &_x, const double &_y): x(_x), y(_y) { }

};

inline comp operator+(const comp &a, const comp &b) { return comp(a.x + b.x, a.y + b.y); }
inline comp operator-(const comp &a, const comp &b) { return comp(a.x - b.x, a.y - b.y); }
inline comp operator*(const comp &a, const comp &b) { return comp(a.x * b.x - a.y * b.y, a.x * b.y + a.y * b.x); }
inline comp conj(const comp &a) { return comp(a.x, -a.y); }

const double PI = acos(-1);

int N,L;

comp w[max0 + 5];
int bitrev[max0 + 5];

void fft(comp *a, const int &n)
{
    REP(i, 0, n) if (i < bitrev[i]) swap(a[i], a[bitrev[i]]);
    for (int i = 2, lyc = n >> 1; i <= n; i <<= 1, lyc >>= 1)
        for (int j = 0; j < n; j += i)
        {
            comp *l = a + j, *r = a + j + (i >> 1), *p = w;
            REP(k, 0, i >> 1)
            {
                comp tmp = *r * *p;
                *r = *l - tmp, *l = *l + tmp;
                ++l, ++r, p += lyc;
            }
        }
}

inline void fft_prepare()
{
    REP(i, 0, N) bitrev[i] = bitrev[i >> 1] >> 1 | ((i & 1) << (L - 1));
    REP(i, 0, N) w[i] = comp(cos(2 * PI * i / N), sin(2 * PI * i / N));
}

inline void conv(int *x, int *y, int *z)
{
    REP(i, 0, N) (x[i] += Mod) %= Mod, (y[i] += Mod) %= Mod;
    static comp a[max0 + 5], b[max0 + 5];
    static comp dfta[max0 + 5], dftb[max0 + 5], dftc[max0 + 5], dftd[max0 + 5];

    REP(i, 0, N) a[i] = comp(x[i] & 32767, x[i] >> 15);
    REP(i, 0, N) b[i] = comp(y[i] & 32767, y[i] >> 15);
    fft(a, N), fft(b, N);
    REP(i, 0, N)
    {
        int j = (N - i) & (N - 1);
        static comp da, db, dc, dd;
        da = (a[i] + conj(a[j])) * comp(0.5, 0);
        db = (a[i] - conj(a[j])) * comp(0, -0.5);
        dc = (b[i] + conj(b[j])) * comp(0.5, 0);
        dd = (b[i] - conj(b[j])) * comp(0, -0.5);
        dfta[j] = da * dc;
        dftb[j] = da * dd;
        dftc[j] = db * dc;
        dftd[j] = db * dd;
    }
    REP(i, 0, N) a[i] = dfta[i] + dftb[i] * comp(0, 1);
    REP(i, 0, N) b[i] = dftc[i] + dftd[i] * comp(0, 1);
    fft(a, N), fft(b, N);
    REP(i, 0, N)
    {
        int da = (LL)(a[i].x / N + 0.5) % Mod;
        int db = (LL)(a[i].y / N + 0.5) % Mod;
        int dc = (LL)(b[i].x / N + 0.5) % Mod;
        int dd = (LL)(b[i].y / N + 0.5) % Mod;
        z[i] = (da + ((LL)(db + dc) << 15) + ((LL)dd << 30)) % Mod;
    }
}
int x[SZ],g[SZ],tmp[SZ],tx[SZ],d[SZ],r[SZ];
void ginv(int s)
{
    if(s==0)
    {
        g[0]=qp(x[0],Mod-2);
        return;
    }
    ginv(s-1);
    L=s+1; N=1LL<<L; fft_prepare();
    for(int i=N/4;i<N;++i) g[i]=0;
    for(int i=0;i<N/2;++i) tx[i]=x[i];
    conv(tx,g,tmp);
    for(int i=N/2;i<N;++i) tmp[i]=0;
    for(int i=0;i<N;++i) tmp[i]=((i?0:2)-tmp[i]+MOD)%MOD;
    conv(g,tmp,g);
    for(int i=N/2;i<N;++i) g[i]=0;
}
int n,rs[SZ];
int main()
{
    scanf("%d%d",&n,&MOD); ++n; x[0]=1;
    for(int i=1;i<n;++i) scanf("%d",x+i);
    int s=0;
    while(n>(1<<s)) ++s;
    ginv(s);
    L=s+1; N=1<<L; fft_prepare();
    for(int i=1;i<n;++i) d[i-1]=x[i]*ll(i)%MOD;
    conv(d,g,r);
    vector<int> ans;
    for(int i=1;i<n;++i)
    {
        rs[i]=(r[i-1]-rs[i]+MOD)%MOD;
        for(int j=i+i;j<=n;j+=i) (rs[j]+=rs[i])%=MOD;
        if(rs[i]) ans.pb(i);
    }
    printf("%d\n",int(ans.size()));
    for(unsigned j=0;j<ans.size();++j)
        printf("%d ",ans[j]);puts("");
}

D2T3 文本校正

假设把原串切成123,那么有以下拼合方法:

123 (那可真蠢)

231、312 (枚举个切分点切开,然后哈希判一判)

213:枚举3的位置,然后可以发现1和2的长度的可能取值必须是两个前缀的最大匹配之一,证明参见相关题目题解,kmp+hash即可。

132:把上面一个倒过来。

321:考虑把两个串做C变换(C(a,b)=a[1]b[n]a[2]b[n-1]...a[n]b[1]),然后就是要判断这个玩意儿是不是一个三偶回文串。枚举一个切分点,就是要看后缀是不是双偶回文串,两个回文串中至少有一个可以是最长回文前缀/最长回文后缀,最长回文后缀直接暴力判,最长回文前缀跑完manacher之后更新一遍。(我们需要知道以每个下标作为左端点的最长回文串。从右到左扫描,维护一下当前的左边界,做第i位时我们只要更新当前i-p[i]~左边界-1这一段,然后更新左边界。正确性显然。)

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <string>
#include <bitset>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <sstream>
#include <stack>
#include <iomanip>
using namespace std;
#define pb push_back
#define mp make_pair
typedef pair<int,int> pii;
typedef long long ll;
typedef double ld;
typedef vector<int> vi;
#define fi first
#define se second
#define fe first
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define Edg int M=0,fst[SZ],vb[SZ],nxt[SZ];void ad_de(int a,int b){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;}void adde(int a,int b){ad_de(a,b);ad_de(b,a);}
#define Edgc int M=0,fst[SZ],vb[SZ],nxt[SZ],vc[SZ];void ad_de(int a,int b,int c){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;vc[M]=c;}void adde(int a,int b,int c){ad_de(a,b,c);ad_de(b,a,c);}
#define es(x,e) (int e=fst[x];e;e=nxt[e])
#define esb(x,e,b) (int e=fst[x],b=vb[e];e;e=nxt[e],b=vb[e])
#define VIZ {printf("digraph G{\n"); for(int i=1;i<=n;i++) for es(i,e) printf("%d->%d;\n",i,vb[e]); puts("}");}
#define VIZ2 {printf("graph G{\n"); for(int i=1;i<=n;i++) for es(i,e) if(vb[e]>=i)printf("%d--%d;\n",i,vb[e]); puts("}");}
#define SZ 4444444
int l[3],r[3];
int T,n,m;
const int MOD=666623333,S=10000007;
ll po[SZ];
template<const int G>
struct str
{
int s[G],n;
ll qz[G];
int fail[G],pal[G],sp[G];
void mlc() //malache
{
    int ml=0,p=0; s[n+1]=-1;s[0]=-2;
    for(int i=1;i<=n;++i)
    {
        if(i<=ml) pal[i]=min(pal[p+p-i],ml-i);
        else pal[i]=0;
        while(i>=pal[i]+1&&
        s[i-pal[i]-1]==s[i+pal[i]+1])
            ++pal[i];
        if(i+pal[i]>ml)
            ml=i+pal[i],p=i;
    }
    for(int i=n;i>=1;--i) sp[i]=0;
    int cl=n+1;
    for(int i=n;i>=1;--i)
    {
        for(int j=min(cl-1,i);j>=i-pal[i];--j)
            cl=j,sp[j]=max(sp[j],i-j);
    }
}
void init()
{
    for(int i=1;i<=n;++i)
        qz[i]=(qz[i-1]*S+s[i])%MOD;
    fail[1]=0;
    for(int i=2,j=0;i<=n;++i)
    {
        while(j&&s[j+1]!=s[i]) j=fail[j];
        if(s[j+1]==s[i]) ++j; fail[i]=j;
    }
}
inline ll hsh(int l,int r)
{
    ll rs=qz[r]-qz[l-1]*po[r-l+1];
    rs%=MOD; return (rs<0)?(rs+MOD):rs;
}
void r()
{
    reverse(s+1,s+1+n);
}
const str& operator = (const str<G>& b)
{
    n=b.n; memcpy(s,b.s,sizeof(int)*(n+2));
}
};
str<1000005> a,b,ra,rb;
bool w213(str<1000005>&a,str<1000005>&b)
{
    int o=n+1;
    for(int i=n;i>=3&&a.s[i]==b.s[i];--i) o=i;
    for(int i=1,j=0;i<n;++i)
    {
        while(j&&a.s[j+1]!=b.s[i]) j=a.fail[j];
        if(a.s[j+1]==b.s[i]) ++j;
        if(i+1<o||!j) continue;
        if(a.hsh(j+1,i)==b.hsh(1,i-j))
        {
            l[0]=1; r[0]=j;
            l[1]=j+1; r[1]=i;
            l[2]=i+1; r[2]=n;
            return 1;
        }
    }
    for(int i=1,j=0;i<n;++i)
    {
        while(j&&b.s[j+1]!=a.s[i]) j=b.fail[j];
        if(b.s[j+1]==a.s[i]) ++j;
        if(i+1<o||!j) continue;
        if(b.hsh(j+1,i)==a.hsh(1,i-j))
        {
            l[0]=1; r[0]=i-j;
            l[1]=i-j+1; r[1]=i;
            l[2]=i+1; r[2]=n;
            return 1;
        }
    }
    return 0;
}
str<2000005> t,rt;
str<4000005> tt;
bool w321()
{
    t.n=0;
    for(int i=1;i<=n;++i)
        t.s[++t.n]=a.s[i],
        t.s[++t.n]=b.s[n+1-i];
    rt=t; rt.r(); t.init(); rt.init();
    tt.n=0; tt.s[++tt.n]=m+1;
    for(int i=1;i<=t.n;++i)
        tt.s[++tt.n]=t.s[i],tt.s[++tt.n]=m+1;
    tt.mlc();
    int rm=0;
    for(int r=n+n;r>=2;--r)
    {
        if(t.hsh(r,n+n)==rt.hsh(1,n+n-r+1)) rm=n+n-r+1;
        int gs=tt.sp[r*2-1]/2*2;
        if(r&1);else continue;
        //case1: r-1|...|rm
        {
            int a=r-1,c=rm,b=n+n-a-c;
            if(a>0&&b>0&&c>0&&a%2==0&&b%2==0&&c%2==0&&
            t.hsh(1,a)==rt.hsh(n+n-a+1,n+n)
            &&t.hsh(a+1,a+b)==rt.hsh(n+n-a-b+1,n+n-a)
            //&&t.hsh(n+n-c+1,n+n)==rt.hsh(1,c)
            )
            {
                a/=2,b/=2,c/=2;
                l[0]=1; ::r[0]=a;
                l[1]=a+1; ::r[1]=a+b;
                l[2]=a+b+1; ::r[2]=n;
                return 1;
            }
        }
        //case2: r-1|gs|...
        {
            int a=r-1,b=gs,c=n+n-a-b;
            if(a>0&&b>0&&c>0&&a%2==0&&b%2==0&&c%2==0&&
            t.hsh(1,a)==rt.hsh(n+n-a+1,n+n)
            &&t.hsh(a+1,a+b)==rt.hsh(n+n-a-b+1,n+n-a)
            &&t.hsh(n+n-c+1,n+n)==rt.hsh(1,c)
            )
            {
                a/=2,b/=2,c/=2;
                l[0]=1; ::r[0]=a;
                l[1]=a+1; ::r[1]=a+b;
                l[2]=a+b+1; ::r[2]=n;
                return 1;
            }
        }
    }
    return 0;
}
namespace FF
{
char ch,B[1<<20],*S=B,*T=B;
#define getc() (S==T&&(T=(S=B)+fread(B,1,1<<20,stdin),S==T)?0:*S++)
#define isd(c) (c>='0'&&c<='9')
int aa,bb;int F(){
    while(ch=getc(),!isd(ch)&&ch!='-');ch=='-'?aa=bb=0:(aa=ch-'0',bb=1);
    while(ch=getc(),isd(ch))aa=aa*10+ch-'0';return bb?aa:-aa;
}
}
#define gi FF::F()
void sol()
{
    n=gi,m=gi; a.n=b.n=ra.n=rb.n=n;
    for(int i=1;i<=n;++i) b.s[i]=gi;
    for(int i=1;i<=n;++i) a.s[i]=gi;
    ra=a; rb=b; ra.r(); rb.r();
    a.init(); b.init(); ra.init(); rb.init();
    l[0]=r[0]=-1;
    if(n<3)
    {
        puts("NO");
        return;
    }
    bool s=1;
    for(int i=1;i<=n;++i) s&=a.s[i]==b.s[i];
    if(s)
    {
        puts("YES");
        puts("1 1");
        puts("2 2");
        printf("3 %d\n",n);
        return;
    }
    for(int i=1;i<n;++i)
    {
        if(a.hsh(1,i)==b.hsh(n-i+1,n)&&
        a.hsh(i+1,n)==b.hsh(1,n-i))
        {
            if(i!=1)
            {
                puts("YES");
                printf("%d %d\n",i+1,n);
                puts("1 1");
                printf("2 %d\n",i);
                return;
            }
            else
            {
                puts("YES");
                printf("%d %d\n",i+1,i+1);
                printf("%d %d\n",i+2,n);
                puts("1 1");
                return;
            }
        }
    }
    bool f=w213(a,b);
    if(f)
    {
        puts("YES");
        printf("%d %d\n",l[1],r[1]);
        printf("%d %d\n",l[0],r[0]);
        printf("%d %d\n",l[2],r[2]);
        return;
    }
    f=w213(ra,rb);
    if(f)
    {
        for(int i=0;i<3;++i)
            l[i]=n+1-l[i],r[i]=n+1-r[i],
            swap(l[i],r[i]);
        puts("YES");
        printf("%d %d\n",l[2],r[2]);
        printf("%d %d\n",l[0],r[0]);
        printf("%d %d\n",l[1],r[1]);
        return;
    }
    f=w321();
    if(f)
    {
        puts("YES");
        printf("%d %d\n",l[2],r[2]);
        printf("%d %d\n",l[1],r[1]);
        printf("%d %d\n",l[0],r[0]);
        return;
    }
    puts("NO");
}
int main()
{
    po[0]=1;
    for(int i=1;i<SZ;++i)
        po[i]=po[i-1]*S%MOD;
    T=gi;
    while(T--) sol();
}

image

完结撒花

posted @ 2017-09-14 21:07 fjzzq2002 阅读(...) 评论(...) 编辑 收藏