Weekly8限时模拟——前缀和+两次二分

A HRZ的序列

题目

相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列a，他对这个序列产生了浓厚的兴趣，他好奇是否存在一个数K，使得一些数加上K，一些数减去K，一些数不变，使得整个序列中所有的数相等，其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算，或是对该位置不进行任何操作。由于瑞神只会刷B站，所以他把这个问题交给了你！

输入格式

输入第一行是一个正整数t表示数据组数。
接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列a。

输出格式

输出共包含t行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。 （输出不包含引号）

Example

Input
2 
5 
1 2 3 4 5 
5
1 2 3 4 5

Output
NO
NO

思路

• 简单的结论题
• 首先寻找序列中不同的数字个数
• 个数为1，代表已经全部相等；个数为2，将其中一个元素+/-两者的差即可
• 个数＞3，因为只有三种操作，必定不能全部相等
• 个数=3，若三个元素构成等差数列，则能全部相等，否则不行

总结

j敲成i，80分没了，记得检查下标啊！╰（‵□′）╯

实现代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;

int t,n,num=0;
long long a[100005];
long long b[100010];

int main()
{
	cin>>t;
	for(int i=0;i<t;i++)
	{
		num=0;
		cin>>n;
		for(int j=0;j<n;j++)
		{
			cin>>a[j];
			//b[j]=0;
		}
		sort(a,a+n);
		b[0]=a[0];
		num=1;
		for(int j=0;j<n-1;j++)
		{
			if(a[j]!=a[j+1]) b[num++]=a[j+1];//检查下标！！！
		}
		if(num==1||num==2) 
			cout<<"YES"<<endl;
		else if(num>3)
			cout<<"NO"<<endl;
		else if(num==3)
		{
			if(b[1]==(b[0]+b[2])/2)
			{
				cout<<"YES"<<endl;
			}
			else 
				cout<<"NO"<<endl;
		}
	}
	return 0;
}

B HRZ学英语

题目

瑞神今年大三了，他在寒假学会了英文的26个字母，所以他很兴奋！于是他让他的朋友TT考考他，TT想 到了一个考瑞神的好问题：给定一个字符串，从里面寻找连续的26个大写字母并输出！但是转念一想， 这样太便宜瑞神了，所以他加大了难度：现在给定一个字符串，字符串中包括26个大写字母和特殊字 符’?’，特殊字符’？'可以代表任何一个大写字母。现在TT问你是否存在一个位置连续的且由26个大写字 母组成的子串，在这个子串中每个字母出现且仅出现一次，如果存在，请输出从左侧算起的第一个出现 的符合要求的子串，并且要求，如果有多组解同时符合位置最靠左，则输出字典序最小的那个解！如果 不存在，输出-1！ 这下HRZ蒙圈了，他刚学会26个字母，这对他来说太难了，所以他来求助你，请你帮 他解决这个问题，报酬是可以帮你打守望先锋。
说明：字典序 先按照第一个字母，以 A、B、C……Z 的顺序排列；如果第一个字母一样，那么比较第二 个、第三个乃至后面的字母。如果比到最后两个单词不一样长（比如，SIGH 和 SIGHT），那么把短者排 在前。例如

AB??EFGHIJKLMNOPQRSTUVWXYZ

ABCDEFGHIJKLMNOPQRSTUVWXYZ 
ABDCEFGHIJKLMNOPQRSTUVWXYZ

上面两种填法，都可以构成26个字母，但是我们要求字典序最小，只能取前者。
注意，题目要求的是 第一个出现的，字典序最小的！

Input

输入只有一行，一个符合题目描述的字符串。

Output

输出只有一行，如果存在这样的子串，请输出，否则输出-1

Example

Input1
ABC??FGHIJK???OPQR?TUVWXY?
Output1
ABCDEFGHIJKLMNOPQRSTUVWXYZ
Input2
AABCDEFGHIJKLMNOPQRSTUVW??M
Output2
-1

思路

• 暴力枚举
• 首先统计字符串中问号的个数，以及出现一次的字母个数
• 若两者相加等于26，证明可以通过将“？”替换，得到想要的字符串
• 题目要求满足字典序最小，则每次从A开始，依次判断是否可以替换
• 也可以使用尺取法解决

实现代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<string.h>
#include<cstring>
#include<algorithm>
using namespace std;

char s[1000005];
char zi[28];
int cun[27]={0};
int num1=0,num2=0;
bool judge=false;

int main()
{
	scanf("%s",s);
	int len=strlen(s);
	if(len<26) 
	{
		cout<<"-1";
		return 0;
	}
	for(int i=0;i<=len-26;i++)
	{
		num1=0,num2=0;
		for(int j=0;j<27;j++)
			cun[j]=0;
		for(int j=i;j<i+26;j++)
		{
			if(s[j]>='A'&&s[j]<='Z')
				cun[s[j]-'A'+1]++;
			else 
				num2++;
		}
		for(int j=1;j<=26;j++)
		{
			if(cun[j]==1) num1++;
		}
		if(num1+num2==26)
		{
			int ss=1;
			for(int j=i;j<i+26;j++)
			{
				if(s[j]=='?')
				{
					for(int k=1;k<=26;k++)
					{
						if(cun[k]==0)
						{
							s[j]='A'+k-1;
							cun[k]=1;
							break;
						}
					}
				}
			}
			for(int j=i;j<i+26;j++)
			{
				zi[ss++]=s[j];
			}
			for(int j=1;j<=26;j++)
				cout<<zi[j];
			return 0;
		}
	}
	if(judge==false)
		cout<<"-1";
	return 0;
}

C 咕咕东的奇妙序列

咕咕东 正在上可怕的复变函数，但对于稳拿A Plus的 咕咕东 来说，她早已不再听课，此时她在睡梦中 突然想到了一个奇怪的无限序列：112123123412345 …这个序列由连续正整数组成的若干部分构成，其 中第一部分包含1至1之间的所有数字，第二部分包含1至2之间的所有数字，第三部分包含1至3之间的所 有数字，第i部分总是包含1至i之间的所有数字。
所以，这个序列的前56项会是11212312341234512345612345671234567812345678912345678910，其中第1项是1，第3项是2，第20项是 5，第38项是2，第56项是0。咕咕东 现在想知道第 k 项数字是多少！但是她睡醒之后发现老师讲的东西 已经听不懂了，因此她把这个任务交给了你。

Input

输入由多行组成。
第一行一个整数q表示有q组询问（1≤q≤500）
接下来第i+1行表示第i个输入 ，表示询问第ki项数字。（1≤ki≤1e18）

Output

输出包含q行
第i行输出对询问ki的输出结果。

Example

Input
5
1
3
20
38
56
Output
1
2
5
2
0

思路

• 前缀和+两次二分
• 设sum[i]为该数到达i的长度，add[i]为第i个数那一块的长度，因此对add求和，即可得到sum[i]
• 分析可知[10ⁱ，10ⁱ⁺¹)的长度均为i+1，所以在这一段中的add[i]都是一个等差数列，可以利用求和公式求解
• 而计算sum[i]可以分成两块来计算，10ⁱ之前的前面已经求出，10ⁱ之后的计算：令cnt=(x-10ⁱ+1)，sum[x]=sum[10ⁱ-1]+add[10ⁱ-1]cnt+cnt(cnt+1)/2*(i+1)
• 然后两次二分，第一次二分找到最大的i，使sum[i]<k，那么解就在第i+1块中；第二次二分找到最大的i使得add[i]<k，那么解就在数i+1中，最后数i+1的第k位即为答案

总结

• 测试的时候没有思路，只暴力得了30分
• 通过两次二分减少查找时间
• 找规律好难(ノ｀Д)ノ

实现代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

long long Sum(long long x,bool flag)
{
	long long sum=0,add=0,len=0,judge=1,cnt=0;
	while(1)
	{
		judge=judge*10;
		len=len+1;
		if(x>judge-1)
		{
			cnt=judge-judge/10;
			sum=sum+add*cnt+cnt*(cnt+1)/2*len;
			add=add+cnt*len;
		}
		else
		{
			cnt=x-judge/10+1;
			sum=sum+add*cnt+cnt*(cnt+1)/2*len;
			add=add+cnt*len;
			break;
		}
	}
	if(flag==true) return sum;
	else if(flag==false) return add;
}

long long binary_search(long long l,long long r,bool flag,long long k)
{
	long long mid,res=0;
	while(l<=r)
	{
		mid=(l+r)>>1;
		if(Sum(mid,flag)<k)
		{
			res=mid;
			l=mid+1;
		}
		else r=mid-1;
	}
	return res;
}

int main()
{
	int num;
	cin>>num;
	long long k;
	for(int i=0;i<num;i++)
	{
		long long res=0;
		cin>>k;
		res=binary_search(0,1e9,true,k);
		k=k-Sum(res,true);
		res=binary_search(0,res+1,false,k);
		k=k-Sum(res,false);
		res=res+1;
		long long it=res,len=0;
		while(it)
		{
			it=it/10;
			len++;
		}
		long long is=len-k+1;
		while(is!=1)
		{
			res=res/10;
			is--;
		}
		cout<<res%10<<endl;
	}
	return 0;
}