[CSP-S模拟测试]:Median（暴力+模拟）

题目描述

定义两个数列：

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

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

$$S_2(k)=S(k)+S(\left\lfloor\frac{k}{10}\right\rfloor+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,k\leqslant 6,000$
对于另外$30\%$的数据，$n,k\leqslant 10,000$
对于另外$20\%$的数据，$w=3$
对于$100\%$的数据，$w\leqslant k\leqslant n\leqslant 10^7$

---

题解

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

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

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

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

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

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

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

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

擦干眼泪，笑面未来！！！

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

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

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

总之这是一道卡常题，$\Theta(n\log 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++