[CSP-S模拟测试]:Median(暴力+模拟)

题目描述

定义两个数列:

$$S={S(1),S(2),...,S(n)} ext{和}S_2{S_2(1),S_2(2),...,S_2(n)}$$

$$S(k)=(p_k imes k)mod w,where p_k is the kth prime number$$

$$S_2(k)=S(k)+S(leftlfloorfrac{k}{10} ight floor+1)$$

令$M(i,j)$表示$S_2(i)$到$S_2(j)$的中位数(个数为奇数就是中间的数,否则为中间的两个数除以二)。现给定$n,k$,求:

$$sum limits_{i=1}^{n-k+1}M(i,i+k-1)$$


输入格式

输入只有一行,为三个正整数$n,k,w$(同题意)。


输出格式

输出只有一行,为所求的答案。如果答案不是整数,使用$.5$表示一半,否则用$.0$


样例

样例输入1:

100 10 10007

样例输出1:

387895.5

样例输入2:

100000 10000 10007

样例输出2:

897586519.5


数据范围与提示

对于$20\%$的数据,$n,kleqslant 6,000$
对于另外$30\%$的数据,$n,kleqslant 10,000$
对于另外$20\%$的数据,$w=3$
对于$100\%$的数据,$wleqslant kleqslant nleqslant 10^7$


题解

首先,想说一下我在考试的时候的思路(毕竟对着这道$T1$刚了一个小时……)

$10^7$的数据范围$nlog n$可能差不多,于是我想到了$Splay$……

然后我就打了,还以为$A$了这道题。

然后忽然想到筛素数不能只筛到$10^7$,我们需要$10^7$素数,当场歇逼……

因为我发现要筛到$179424673$……

然后我就打算从$w$入手,推式子,找规律,最后啥也没发现,于是我只筛到了$10^7$,因为我觉得多了会$T$(学校$OJ$太菜)……

然而正解告诉我们,就是要筛到$179424673$……

因为$OJ$太才,于是标程$T$掉了,老师把时限开到了$4s$并重测,那些筛到$179424673$的人(本来都$T$飞了……)都拿到了$70$分,然而我差点跌出了前$10$……

擦干眼泪,笑面未来!!!

于是我们开始讲这道题……

思考一个类似莫队的思路,也类似滑动窗口叭~

维护一个指针指向中位数,挪动窗口时更新位置即可($k$是偶数时维护两个即可)。

总之这是一道卡常题,$Theta(nlog n)$的做法就别想了,因为它是这样的$downarrow$

然后我去尝试了正解$downarrow$

其实我也不知道我到底哪里把常数给写大了,总之别人的是这样的$downarrow$

然而我就筛个素数就$downarrow$

时间复杂度:$Theta(179424673+n)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
int n,k,w;
int S1[10000001],S2[10000001];
int prime[10000001],cnt;
int t[10000001];
char vis[179424674];
double ans;
void pre_work()
{
	for(int i=2;i<179424674;i++)
	{
		if(!vis[i])prime[++cnt]=i;
		for(int j=1,x;j<=cnt&&(x=i*prime[j])<179424674;j++)
		{
			vis[x]=1;
			if(!(i%prime[j]))break;
		}
	}
}
int main()
{
	pre_work();
	scanf("%d%d%d",&n,&k,&w);
	for(long long i=1;i<=n;i++)S1[i]=prime[i]*i%w;
	for(int i=1;i<=n;i++)S2[i]=S1[i]+S1[i/10+1];
	for(int i=1;i<k;i++)t[S2[i]]++;
	if(k&1)
	{
		int median=(k>>1)+1;
		int lst=0;
		int hand=-1;
		for(int i=k;i<=n;i++)
		{
			t[S2[i]]++;
			if(S2[i]<=hand)lst++;
			if(i>k)
			{
				t[S2[i-k]]--;
				if(S2[i-k]<=hand)lst--;
			}
			while(lst<median)lst+=t[++hand];
			while(lst>=median+t[hand])lst-=t[hand--];
			ans+=hand;
		}
	}
	else
	{
		int median=(k>>1);
		int lst1=0,lst2=0;
		int hand1=-1,hand2=-1;
		for(int i=k;i<=n;i++)
		{
			t[S2[i]]++;
			if(S2[i]<=hand1)lst1++;
			if(S2[i]<=hand2)lst2++;
			if(i>k)
			{
				t[S2[i-k]]--;
				if(S2[i-k]<=hand1)lst1--;
				if(S2[i-k]<=hand2)lst2--;
			}
			while(lst1<median)lst1+=t[++hand1];
			while(lst2<median+1)lst2+=t[++hand2];
			while(lst1>=median+t[hand1])lst1-=t[hand1--];
			while(lst2>=median+1+t[hand2])lst2-=t[hand2--];
			ans+=(double)(hand1+hand2)/2;
		}
	}
	printf("%.1lf",ans);
	return 0;
}

rp++

原文地址:https://www.cnblogs.com/wzc521/p/11631528.html