Codeforces Round 962 (Div. 3)

链接

C题：

        思路：

        直接暴力求每个字母的前缀和，对于区间l，r的最小操作就是区间不同数的一半，因为可以把一个数变成另一个不一样的数，一下抵消两个。

#include<bits/stdc++.h>
using namespace std;
//#define int long long
typedef long long ll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const int N=2e5+10;
const int mod=1e9+7;
#define fi first
#define se second
 

int n,m;
int pre[N][30],nex[N][30];
void solve(){
	cin>>n>>m;
	string s,ss;
	cin>>s>>ss;
	s=' '+s;
	ss=' '+ss;
	for(int i=1;i<=n;i++){
		for(int j=0;j<=27;j++){
			pre[i][j]=pre[i-1][j];
			nex[i][j]=nex[i-1][j];
		}
		pre[i][s[i]-'a']++;
		nex[i][ss[i]-'a']++;
	}
	while(m--){
		int cnt=0;
		int l,r;
		cin>>l>>r;
		for(int i=0;i<=27;i++){
			cnt+=abs((pre[r][i]-pre[l-1][i])-(nex[r][i]-nex[l-1][i]));
			
		}
		int ans=cnt/2;
		cout<<ans<<endl;
	}
}
signed  main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int t=1;
	cin>>t;
	while(t--){
		solve();
	}
	return 0;
}

D题

思路：

        abc有三种情况，一种是全相等，一种是两个相等，最后一种是全不相等。根据排列组合，第一种只有一种排列，第二种有三种排列，第三种有六种排列。

        对于第一种情况，我们直接暴力遍历一个数即可。第二种情况我们可以分两种情况考虑：1.不一样的数小于一样的数，2.不一样的数大于一样的数，我们直接暴力遍历这两种情况即可。第三种情况我们假定先找最小数，再找次小数，最后找最大数即可。

代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const int N=1e5+10;
const int mod=1e9+7;
#define fi first
#define se second
#define ls u<<1
#define rs u<<1|1


int n,x;
void solve(){
	cin>>n>>x;
	int cnt=0;
	if(x<3){
		cout<<0<<endl;
		return;
	}
	for(int i=1;3*i<=x&&i*i*3<=n;i++){
		cnt++;
	}
	for(int i=1;i*2<=x&&i*i<=n;i++){
		for(int j=1;i*2+j<=x&&j<i&&i*i+j*2*i<=n;j++){
			cnt+=3;
		}
		for(int j=i+1;i*2+j<=x&&j<=x&&i*i+j*2*i<=n;j++){
			cnt+=3;
		}
	}
	for(int i=1;i*3<=x&&i*i*3<=n;i++){
		for(int j=i+1;i*2+j<=x&&i*i*2+j*j<=n;j++){
			for(int k=j+1;i+j+k<=x&&i*j+i*k+k*j<=n;k++)
				cnt+=6;
		}
	}
	cout<<cnt<<endl;
}
signed  main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int t=1;
	cin>>t;
	while(t--){
		solve();
	}
	return 0;
}

E题

思路：

        由于要知道区间个数，我们很容易想到要前缀和。我们另1为+1,0为-1，那么前缀和差为0的区间就是01数量一样的区间。我们很容易想到用一个vector来存sub的值，那么我们枚举左端点的时候，只需要去同样sub的位置去找即可。会有0101010101...这种情况，容易知道这样肯定是超时的。

        但我们一般都是找到暴力的方法然后再找方法优化，这题也是这样。

        如果我们找到的一对数为（x，y），根据分步乘法原理，容易知道这样的贡献是x*（n-y+1）。我们让x不动，去找所有右端点，也就是之前超时的暴力遍历sub。我们可以发现每次都是l*（n-xi+1）作为增加的贡献。我们可以发现 l 在我们遍历这次的时候是定值，n和1也是定值，我们都可以提取出来。假如后面满足条件的右端点有k个，那么以 l 为左端点的贡献就是 l*（k*n+k-）。对于k我们可以二分查找下标容易求，然后对于后面那求和部分我们很容易发现是一个后缀和，而后缀和又可以看成总和减去前缀和，所以我们去维护每个sub的前缀和，这样我们就可以log的时间求出以 l 为左端点的贡献了。

tle代码：

string s;
int pre[N];
map<int,vector<int> > v;
void solve(){
	cin>>s;
	s=' '+s;
	v.clear();
	int n=s.size()-1;
	for(int i=1;i<=n;i++){
		pre[i]=pre[i-1];
		if(s[i]=='1')
			pre[i]++;
		v[pre[i]-(i-pre[i])].push_back(i);
	}
	int cnt=0;
	for(int i=1;i<=n;i++){
		int sub=pre[i-1]-(i-1-pre[i-1]);
		for(auto &x:v[sub]){
			if(x>i){
				cnt=(cnt+(i)*(n-x+1))%mod;
			}
			
		}
	}
	cout<<cnt<<endl;
}

优化代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const int N=2e5+10;
const int mod=1e9+7;
#define fi first
#define se second
 

string s;
int pre[N];
map<int,vector<int> > v;
map<int,vector<int> > vv;
void solve(){
	cin>>s;
	s=' '+s;
	v.clear();
	vv.clear();
	int n=s.size()-1;
	set<int> p;
	for(int i=1;i<=n;i++){
		pre[i]=pre[i-1];
		if(s[i]=='1')
			pre[i]++;
		v[pre[i]-(i-pre[i])].push_back(i);
		p.insert(pre[i]-(i-pre[i]));
	}
	for(auto &x:p){
		int sum=0;
		for(int i=0;i<v[x].size();i++){
			sum+=v[x][i];
			vv[x].push_back(sum);
		}
	}
	int cnt=0;
	for(int i=1;i<=n;i++){
		int sub=pre[i-1]-(i-1-pre[i-1]);
		if(v[sub].size()==0) continue;
		int p=lower_bound(v[sub].begin(),v[sub].end(),i+1)-v[sub].begin();
		int pp=v[sub].size()-1;
		int t=0;
		if(p>0) t=vv[sub][pp]-vv[sub][p-1];
		else t=vv[sub][pp];
		cnt=(cnt+i*(-t+(pp-p+1)*n+(pp-p+1)))%mod;
	}
	cout<<cnt<<endl;
}
signed  main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int t=1;
	cin>>t;
	while(t--){
		solve();
	}
	return 0;
}