2016-2017 National Taiwan University World Final Team Selection Contest (Codeforces Gym) 部分题解
D
考虑每个点被删除时其他点对它的贡献，然后发现要求出距离为1~k的点对有多少个。
树分治+FFT。分治时把所有点放一起做一遍FFT，然后减去把每棵子树单独做FFT求出来的值。
复杂度$nlog^2n$
#include<bits/stdc++.h>
#define N 270000
#define pi acos(-1)
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int p = 1000000007;
int pw(int x,int y)
{
int lst=1;
while(y)
{
if(y&1)lst=1LL*lst*x%p;
y>>=1;
x=1LL*x*x%p;
}
return lst;
}
int head[N],ver[2*N],nxt[2*N],tot;
void add(int a,int b)
{
tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;return ;
}
struct E
{
double x,y;
E(){;}
E(double _x,double _y)
{
x=_x;y=_y;
}
friend E operator + (E &a,E &b)
{
return E(a.x+b.x,a.y+b.y);
}
friend E operator - (E &a,E &b)
{
return E(a.x-b.x,a.y-b.y);
}
friend E operator * (E &a,E &b)
{
return E(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);
}
}a[N],b[N],c[N];
int R[N];
ll ans[N];
void fft(E *a,int f,int n)
{
for(int i=0;i<n;i++)if(i>R[i])swap(a[i],a[R[i]]);
for(int i=1;i<n;i<<=1)
{
E wn(cos(pi/i),f*sin(pi/i));
for(int j=0;j<n;j+=(i<<1))
{
E w(1,0);
for(int k=0;k<i;k++,w=w*wn)
{
E x=a[j+k],y=a[j+k+i]*w;
a[j+k]=x+y;a[j+k+i]=x-y;
}
}
}
if(f==-1)
{
for(int i=0;i<n;i++)a[i].x/=n;
}
return ;
}
void FFT(int *sa,int m,int f)
{
int l=0,n=1;
while(n<=2*m)n<<=1,l++;
for(int i=0;i<n;i++)
{
a[i].y=a[i].x=b[i].x=b[i].y=0;
if(i<=m)a[i].x=b[i].x=sa[i];
}
for(int i=0;i<n;i++)R[i]=(R[i>>1]>>1)|((i&1)<<(l-1));
fft(a,1,n);fft(b,1,n);
for(int i=0;i<n;i++)c[i]=a[i]*b[i];
fft(c,-1,n);
for(int i=2;i<n;i++)
{
ll tmp=(ll)(c[i].x+0.5);
ans[i-1]+=f*tmp;
}
return ;
}
int size[N],mn,id,sum,v[N];
int n;
void dfs(int x,int f)
{
int mx=0;
size[x]=1;
for(int i=head[x];i;i=nxt[i])
{
if(v[ver[i]]||ver[i]==f)continue;
dfs(ver[i],x);
size[x]+=size[ver[i]];
mx=max(mx,size[ver[i]]);
}mx=max(mx,sum-size[x]);
if(mx<mn)
{
mn=mx;
id=x;
}
return ;
}
int now[N],nw[N];
int mxx=0;
void dffs(int x,int f,int dp)
{
size[x]=1;ans[dp]+=2;now[dp]++,nw[dp]++;
if(dp>mxx)mxx=dp;
for(int i=head[x];i;i=nxt[i])
{
if(v[ver[i]]||ver[i]==f)continue;
dffs(ver[i],x,dp+1);
size[x]+=size[ver[i]];
}
return ;
}
void solve(int x)
{
sum=size[x];mn=inf;id=x;
dfs(x,-1);
x=id;
v[x]=1;size[x]=1;ans[1]++;
for(int i=0;i<=sum;i++)now[i]=0;
int mx=0;
for(int i=head[x];i;i=nxt[i])
{
if(v[ver[i]])continue;
mxx=0;
dffs(ver[i],x,2);
size[x]+=size[ver[i]];
FFT(nw,mxx,-1);
mx=max(mx,mxx);
for(int j=0;j<=mxx;j++)nw[j]=0;
}
if(mx)FFT(now,mx,1);
for(int i=head[x];i;i=nxt[i])
{
if(!v[ver[i]])solve(ver[i]);
}
}
int main()
{
scanf("%d",&n);
int t1,t2;
for(int i=1;i<n;i++)
{
scanf("%d%d",&t1,&t2);
add(t1,t2);add(t2,t1);
}
size[1]=n;
solve(1);
ll as=0;
for(int i=1;i<=n;i++)
{
ans[i]%=p;
as+=ans[i]*pw(i,p-2)%p;
}
as%=p;
for(int i=2;i<=n;i++)
{
as=as*i%p;
}
printf("%I64d\n",as);
return 0;
}
E
把每条线段看成二维平面上的一个点。
相当于求从(0,0)点到(n+1,n+1)的一条权值和最小的一条路径，且相邻两个点之间不能有其他点。
CDQ分治+单调栈+线段树
和bzoj 4273很像。
#include<bits/stdc++.h> #define N 100055 #define inf 2147483647 #define ls x<<1,l,mid #define rs x<<1|1,mid+1,r using namespace std; int n; int p[N],v[N]; int a[N*4]; void gai(int x,int l,int r,int pos,int z) { if(l==r) { a[x]=z;return ; } int mid=(l+r)>>1; if(pos<=mid)gai(ls,pos,z); else gai(rs,pos,z); a[x]=min(a[x<<1],a[x<<1|1]); return ; } int qur(int x,int l,int r,int ll,int rr) { if(l>=ll&&r<=rr)return a[x]; int mid=(l+r)>>1; if(ll>mid)return qur(rs,ll,rr); if(rr<=mid)return qur(ls,ll,rr); return min(qur(rs,ll,rr),qur(ls,ll,rr)); } int st1[N],top1,st2[N],top2; int q1[N],cnt1,q2[N],cnt2,f[N]; bool cmp(int x,int y) { return p[x]<p[y]; } void solve(int l,int r) { if(l==r)return ; int mid=(l+r)>>1; solve(l,mid); top1=top2=cnt1=cnt2=0; for(int i=l;i<=mid;i++)q1[++cnt1]=i; for(int i=mid+1;i<=r;i++)q2[++cnt2]=i; sort(q1+1,q1+cnt1+1,cmp);sort(q2+1,q2+cnt2+1,cmp); int pt=1; for(int i=1;i<=cnt2;i++) { while(pt<=cnt1&&p[q1[pt]]<=p[q2[i]]) { while(top1&&st1[top1]<q1[pt]) { gai(1,1,n,p[st1[top1]],inf); top1--; } gai(1,1,n,p[q1[pt]],f[q1[pt]]+v[q1[pt]]); st1[++top1]=q1[pt]; pt++; } while(top2&&st2[top2]>q2[i])top2--; int tmp=0; if(top2)tmp=p[st2[top2]]+1; f[q2[i]]=min(f[q2[i]],qur(1,1,n,tmp,p[q2[i]])); st2[++top2]=q2[i]; } while(top1)gai(1,1,n,p[st1[top1]],inf),top1--; solve(mid+1,r); } int main() { scanf("%d",&n); for(int i=1;i<=n;i++)scanf("%d",&p[i]); for(int i=1;i<=n;i++)scanf("%d",&v[i]); for(int i=1;i<=4*(n+1);i++)a[i]=inf; p[n+1]=n+1;p[0]=0; memset(f,0x3f,sizeof(f)); f[0]=0; solve(0,n+1); printf("%d\n",f[n+1]); return 0; }
H
按极角排序，然后用一条线扫过去。
然后狂WA第5个点，去膜了下Claris的代码，发现有些细节写错了。。。
当一个点左右两个点极角比它小的时候，那么答案会加一，否则会减一。
如果连续一条直线上极角都相等，只拿端点算贡献。
然后这些点分为两类，假设现在答案要加一。
如果这个点在右下(意会一下)，那么只有扫描线扫过这个点时答案才会加一。
如果在左上，那么扫到这个点时答案已经加了一。
讨论一下，具体看代码。
#include<bits/stdc++.h>
#define N 100005
#define ll long long
using namespace std;
int n;
struct node
{
int x,y;
node(){;}
node(int _x,int _y)
{
x=_x;y=_y;
}
friend node operator - (const node &aa,const node &bb)
{
return node(aa.x-bb.x,aa.y-bb.y);
}
}a[N];
int p[N];
ll cj(const node &aa,const node &bb)
{
return 1LL*aa.x*bb.y-1LL*aa.y*bb.x;
}
bool in[N],ok[N];
bool cmp(int x,int y)
{
return cj(a[x],a[y])>0;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d%d",&a[i].x,&a[i].y);
a[0]=a[n];a[n+1]=a[1];
for(int i=1;i<=n;i++)p[i]=i;
sort(p+1,p+n+1,cmp);
int sum=0,ans=0;
for(int i=1;i<=n;)
{
node as=a[p[i]];
int tmp=0;
for(;i<=n&&cj(as,a[p[i]])==0;i++)
{
if(cj(a[p[i]+1],a[p[i]])<0&&cj(a[p[i]-1],a[p[i]])<=0)
{
if(cj(a[p[i]-1]-a[p[i]],a[p[i]+1]-a[p[i]])>0)sum++;
else tmp++;
}
if(cj(a[p[i]+1],a[p[i]])>0&&cj(a[p[i]-1],a[p[i]])>=0)
{
if(cj(a[p[i]-1]-a[p[i]],a[p[i]+1]-a[p[i]])<0)sum--;
else tmp--;
}
}
ans=max(ans,sum);
sum+=tmp;
ans=max(ans,sum);
}
printf("%d\n",ans+1);
return 0;
}
I
傻逼题，一个子树要么给上边提供一个两个叶子的小子树，要么是一个叶子或零个，其他的只能直接配对，画画图大力分类讨论+贪心。
#include<bits/stdc++.h>
#define N 100005
using namespace std;
int n;
int head[N],ver[N*2],nxt[N*2],tot;
void add(int a,int b)
{
tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;return ;
}
int du[N],root;
int ans;
int dfs(int x,int f)
{
int t1=0,t2=0;
int cnt=0;
for(int i=head[x];i;i=nxt[i])
{
if(ver[i]==f)continue;
cnt++;
int tmp=dfs(ver[i],x);
if(tmp==1)t1++;
else if(tmp==2)t2++;
}
if(!cnt)return 1;
while(t2>=2)t2-=2,ans++;
while(t1>=3)t1-=2,ans++;
if(!t1)
{
if(t2==1)return 2;
return 0;
}
if(t1==1)
{
if(t2==1)return 2;
return 1;
}
else
{
if(!t2)return 2;
else if(t2==1){ans++;return 1;}
else {ans++;return 2;}
}
}
int main()
{
scanf("%d",&n);
if(n==2)
{
puts("1");
return 0;
}
int t1,t2;
for(int i=1;i<n;i++)
{
scanf("%d%d",&t1,&t2);
add(t1,t2);add(t2,t1);
du[t1]++;du[t2]++;
}
for(int i=1;i<=n;i++)if(du[i]>1)root=i;
int tmp=dfs(root,-1);
if(tmp==2)ans++;
printf("%d\n",ans);
return 0;
}
J
先假设一共有无数个0。
枚举右端点，枚举左端点，然后把中间的1去掉，剩下的操作往里边塞0。
列下式子发现左端点单调，可以用单调队列优化。
最后把ans和0的个数取个min。
#include<bits/stdc++.h>
#define N 1000005
using namespace std;
char s[N];
int n,mx,sum[N];
int q[N];
void solve(int x)
{
int ha=1,ta=1;
q[1]=0;int ans=0;
for(int i=1;i<=n;i++)
{
while(ta>=ha&&sum[i]-sum[q[ha]]>x)ha++;
if(ta>=ha)
{
ans=max(ans,i-2*sum[i]+x+2*sum[q[ha]]-q[ha]);
}
while(ta>=ha&&2*sum[q[ta]]-q[ta]<=2*sum[i]-i)ta--;
q[++ta]=i;
}
printf("%d\n",min(ans,mx));
return ;
}
int main()
{
scanf("%s",s+1);
n=strlen(s+1);
for(int i=1;i<=n;i++)
{
sum[i]=sum[i-1];
if(s[i]=='1')sum[i]++;
}
for(int i=1;i<=n;i++)if(s[i]=='0')mx++;
int q;scanf("%d",&q);
for(int i=1;i<=q;i++)
{
int cnt;scanf("%d",&cnt);
solve(cnt);
}
return 0;
}
