来自学长的快乐AK题——Day2

---

A——蜜汁做法

Description

n个小球排成一列，第i个小球颜色为ai

定义小球i与j的距离为|(i-j)*(ai-aj)|

求所有小球间距离之和

i与j 和 j与i 不必重复计算

100%：n<=1000000，0<=ai<=1

思路

由条件容易得到：只有当ai-aj不为0，即ai和aj不相等时，距离才不为0。也就是说，只有ai与aj不相等的两个小球才能产生贡献。

观察数据范围，很容易发现一个突破口：ai不是0就是1。那么对于一个0色球来说，只有1色球能与它产生贡献，1色球同理。

对于任意一个小球i来说，设它前面有a1个0色球，编号和为a2；有b1个1色球，编号和为b2。

若i为0色球，则前面的1色球对它的贡献为b1*i-b2，为1色球同理。（只需要考虑i之前的小球，因为题目要求无需重复计算i，j和j，i）

代码

（代码与上面的题解除了基本思想外一点关系也没有，属于我比赛时乱搞。。。）

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e6;
int n,cnt,tot,a[N+1];
long long ans,s[N+1],p[N+1],q[N+1];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		if(!a[i]) p[++cnt]=i;
		else q[++tot]=i;
	}
	for(int i=1;i<=tot;i++)
	s[i]=s[i-1]+q[i];
	for(int i=1;i<=cnt;i++)
	{
		int k=lower_bound(q+1,q+tot+1,p[i])-q;
		int len=tot-k+1;
		ans+=(s[tot]-s[k-1]-len*p[i])+((k-1)*p[i]-s[k-1]);
	}
	printf("%lld",ans);
	return 0;
}

B——最小生成树

Description

n个人，每个人都有一份情报

人之间都可以交流，交流有对应的费用，交流完后各自享有对方的情报

A与B交流，A与C交流，B与C也可以互相享有

派遣n个人中任意一人去执行任务都有对应的费用

要求：派遣人去执行任务，派遣的人需要包含所有的情报(不一定只派遣一个)，问最小花费

思路

比赛时一眼看过去，就知道遇见了你，我的最小生成树！

首先，考虑两个人花一定费用见面，就能互相知道对方的情报，就联想到用一条边将两个点连接起来。考虑派遣出去的人必须得到所有人的情报，就联想到一张图中所有的点互相连通。
用最小的花费使得所有的点互相连通，这还不是最小生成树吗？

最后，执行任务的费用可以视为图中再加入一个n+1号节点，与每个人新建一条边，边权为这个人执行任务的费用。

对于这个n+1个节点的图跑一遍最小生成树算法即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e6;
struct edge
{
	int u,v,w;
}a[N];
int n,k,cnt,fa[N];
long long ans;
void add(int u,int v,int w)
{
	cnt++;
	a[cnt].u=u;
	a[cnt].v=v;
	a[cnt].w=w;
}
int find(int x)
{
	if(fa[x]==x) return fa[x];
	return fa[x]=find(fa[x]);
}
void join(int x,int y)
{
	int fx=find(x),fy=find(y);
	fa[fx]=fy;
}
bool cmp(edge a,edge b){return a.w<b.w;}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	{
		scanf("%d",&k);
		if(i<j) add(i,j,k);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&k);
		add(n+1,i,k);
	}
	sort(a+1,a+cnt+1,cmp);
	for(int i=1;i<=n+1;i++) fa[i]=i;
	for(int i=1,j=1;i<=n;i++)
	{
		while(find(a[j].u)==find(a[j].v)) j++;
		join(a[j].u,a[j].v);
		ans+=a[j].w;
	}
	printf("%lld",ans);
	return 0;
}

C——维护第k大的数

Description

给出一个长度为n的序列A描述一个长度为n的全排列a，A_i表示前i-1个数中有多少个数比a_i大。
请求出这个全排列a

思路

从后往前遍历A，根据题目中的定义，a的前n-1个数中有A_n个数比a_n大，换言之，a_n就是1~n中第A_n+1大的数。同理，a_n-1就是除去已经确定的a_n后第A_n-1+1大的数。以此类推，就将问题转化为如何维护一个序列中的第k大数。

算法还是有很多，例如快排（nlogn），归并(nlogn或n)，我是采用更好写也更快的桶排(n)，卡到了O(n²)。但数据一大（超时）或一恶心（超出桶的范围），桶排就不顶用了。

学长教了一个更优秀的算法：利用树状数组动态维护前缀和。

对于每个存在的数i，将数组中下标为i的位置赋值为1。对于任意一个数i，只要求出数组下标为1~i的前缀和，就能得到i的大小排名。如果这个数不存在或被删除，将其对应下标的位置赋值为0即可。
原理也容易明白，就是优化了桶排，通过前缀和快速求出有多少个数小于i。

有了这个东东，我们只需要二分数组的下标，得到前缀和正好为k时的下标，即为第k大的数。

代码

（我懒得再写树状数组的优化了，反正我都过了）

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1e4;
int n,a[N],vis[N],ans[N];
int query(int k)
{
	for(int i=n,cnt=0;i;i--)
	{
		if(!vis[i]) cnt++;
		if(cnt==k) return i;
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	for(int i=n;i;i--)
	ans[i]=query(a[i]+1),vis[ans[i]]=1;
	for(int i=1;i<=n;i++)
	printf("%d ",ans[i]);
	return 0;
}

D——差分。。。之类的东东

Description

题面有误，应该为子串长度。（否则这题就sb到爆了）

思路

考虑差分的思想，为男生则赋值+1，为女生则赋值-1，做前缀和。

倘若一段区间i~j内男女个数相等，则该区间的所有值之和显然等于0（因为个数相等的+1和-1互相抵消了）。
用前缀和来表示，即s[j]-s[i-1]=0。简单移项可以得到s[j]=s[i-1]。

也就是说，如果s[j]=s[i]，那么i+1~j就是一个男女人数相等的区间。用桶记录之前出现过的最早的s数组值的位置（因为要求最大），倘若之后再有相同的值，用当前位置减去之前位置再与答案比较即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e6;
int n,s[N],h[N],ans; 
int main()
{
	scanf("%d",&n);
	for(int i=1,k;i<=n;i++)
	{
		scanf("%d",&k);
		if(k==0) k=-1;
		s[i]=s[i-1]+k;
	}
	for(int i=1;i<=n;i++)
	{
		int t=s[i]+n;
		if(h[t]==0) h[t]=i;
		else ans=max(ans,i-h[t]);
	}
	printf("%d",ans);
	return 0;
}

Tips

我今天AK啦！！！！！！！！！！！