B站带我打codeforces（三）

E. Nanosoft

原题地址




代码：
题目大意：定义一个logo为四个小正方形按（R,G,B,Y）颜色顺序拼成一个大正方形，有q次询问，每次询问输入一个范围，求图中这个范围内的面积最大商标
由于q的范围是1e5，可以想到应该采用O（1）或O（log n）的查询方法，自然想到了二维ST表。在预处理时，先将每个红点与周围点（即判断它右边是否为绿点，下方是否为黄点，右下是否为蓝点）的最大log处理出来。查询时二分答案，即可获得结果。
详细讲解
代码借鉴qsc的视频讲解

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<vector>
using namespace std;
typedef long long ll;
#define maxn 505
#define ma 11
int n,m,q,R[4][maxn][maxn],val[maxn][maxn],f[ma][ma][maxn][maxn],lg[maxn];
char s[maxn][maxn];
void build(int n,int m)//二维ST表
{
	int i,j,k1,k2;
	for(i=0;i<n;i++)
	{
		for(j=0;j<m;j++)
		{
			f[0][0][i+1][j+1]=val[i][j];
		}
	}
	for(i=2;i<maxn;i++)
		lg[i]=1+lg[i>>1];
	for(i=1;i<=n;i++)
		for(k2=1;(1<<k2)<=m;k2++)
			for(j=1;j<=m;j++)
				f[0][k2][i][j]=max(f[0][k2-1][i][j],f[0][k2-1][i][j+(1<<(k2-1))]);//先处理列
	for(k1=1;(1<<k1)<=n;k1++)
		for(i=1;i<=n-(1<<k1)+1;i++)
			for(k2=0;(1<<k2)<=m;k2++)
				for(j=1;j<=m-(1<<k2)+1;j++)
					f[k1][k2][i][j]=max(f[k1-1][k2][i][j],f[k1-1][k2][i+(1<<(k1-1))][j]);//再处理行
}
void color(char c,int fx,int fy,int k)
/*寻找颜色为R/G/Y/B的，以(i,j)为右下/左下/右上/左上角的最大正方形
*/
{
	int stx=0,sty=0;
	if(fx==-1) stx=n-1;
	if(fy==-1) sty=m-1;
	while(stx<n&&stx>=0)
	{
		while(sty<m&&sty>=0)
		{
			if(s[stx][sty]==c)
			{
				R[k][stx][sty]=1;
				if(stx-fx>=0&&stx-fx<n&&sty-fy>=0&&sty-fy<m)
				{
					if(s[stx-fx][sty]==c&&s[stx][sty-fy]==c&&s[stx-fx][sty-fy]==c)
					{
						R[k][stx][sty]=min(R[k][stx-fx][sty],min(R[k][stx][sty-fy],R[k][stx-fx][sty-fy]))+1;
						//寻找最大正方形模板
					}
				}
			}
			sty+=fy;
		}
		sty=0;if(fy==-1) sty=m-1;
		stx+=fx;
	}
}
int query(int x1,int y1,int x2,int y2)//ST表查询
{
	int k1=lg[x2-x1+1],k2=lg[y2-y1+1];
	x2=x2-(1<<k1)+1;
	y2=y2-(1<<k2)+1;
	return max(max(f[k1][k2][x1][y1],f[k1][k2][x1][y2]),max(f[k1][k2][x2][y1],f[k1][k2][x2][y2]));
}
bool check(int r1,int c1,int r2,int c2,int mid)//非常有意思的check函数
{
	r1=r1+mid+-1;
	c1=c1+mid-1;
	r2=r2-mid;
	c2=c2-mid;
	if(r1>r2||c1>c2) return false;
	return query(r1,c1,r2,c2)>=mid;
}
int main()
{    
	ios::sync_with_stdio(false);
	cin>>n>>m>>q;
	for(int i=0;i<n;i++)
	{
		cin>>s[i];
	}
	color('R',1,1,0);//预处理
	color('Y',-1,1,1);
	color('G',1,-1,2);
	color('B',-1,-1,3);
	for(int i=0;i<n-1;i++)
	{
		for(int j=0;j<m-1;j++)
		{
			if(s[i][j]=='R'&&s[i+1][j]=='Y'&&s[i][j+1]=='G'&&s[i+1][j+1]=='B')
			{
				val[i][j]=min(R[0][i][j],min(R[1][i+1][j],min(R[2][i][j+1],R[3][i+1][j+1])));
			}
		}
	}
	build(n,m);
	while(q--)//q次询问
	{
		int r1,c1,r2,c2,mid;
		scanf("%d%d%d%d",&r1,&c1,&r2,&c2);
		int lc=0,rc=min(r2-r1,c2-c1),ans=0;
		while(lc<=rc)//二分答案
		{
			mid=(lc+rc)/2;
			if(check(r1,c1,r2,c2,mid))
			{
				lc=mid+1;
				ans=mid;
			}
			else rc=mid-1;
		}
		printf("%d\n",ans*ans*4);
	}
    return 0;
}

F. Super Jaber

原题地址



代码：
题目大意：在一张图中，规定一步可以向相邻的格子移动一步，或跳转到相同颜色的格子，有q次询问，问从（r1,c1）到(r2,c2)最少要几步。
显然又是一道疯狂预处理的题目，bfs处理每个点到不同颜色的最短距离
详细讲解
代码借鉴qsc的视频

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<queue>
using namespace std;
typedef long long ll;
#define maxn 1010
#define mkp make_pair
int n,m,q,k,a[maxn][maxn],dis[44][maxn][maxn],vis[maxn][maxn],viscol[44];
vector<pair<int,int> >col[44];
int dx[4]={-1,1,0,0};
int dy[4]={0,0,1,-1};
void bfs(int x)
{
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			vis[i][j]=0;
	for(int i=1;i<=k;i++) viscol[i]=0;
	viscol[x]=1;
	queue<pair<int,int> >Q;
	for(int i=0;i<col[x].size();i++)
	{
		Q.push(col[x][i]);
		dis[x][col[x][i].first][col[x][i].second]=0;
		vis[col[x][i].first][col[x][i].second]=1;
	}
	while(!Q.empty())
	{
		int nowx=Q.front().first;
		int nowy=Q.front().second;
		Q.pop();
		if(!viscol[a[nowx][nowy]])
		{
			for(int i=0;i<col[a[nowx][nowy]].size();i++)
			{
				int nx=col[a[nowx][nowy]][i].first;
				int ny=col[a[nowx][nowy]][i].second;
				if(!vis[nx][ny])
				{
					dis[x][nx][ny]=1+dis[x][nowx][nowy];
					vis[nx][ny]=1;
					Q.push(mkp(nx,ny));
				}
			}
			viscol[a[nowx][nowy]]=1;
		}
		for(int i=0;i<4;i++)
		{
			int nx=nowx+dx[i];
			int ny=nowy+dy[i];
			if(nx<0||nx>=n||ny<0||ny>=m) continue;
			if(!vis[nx][ny])
			{
				dis[x][nx][ny]=1+dis[x][nowx][nowy];
				vis[nx][ny]=1;
				Q.push(mkp(nx,ny));
			}
		}
	}
}
int main()
{    
	ios::sync_with_stdio(false);
	cin>>n>>m>>k;
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
		{
			scanf("%d",&a[i][j]);
			col[a[i][j]].push_back(mkp(i,j));
		}
	for(int i=1;i<=k;i++)
		bfs(i);
	cin>>q;
	while(q--)
	{
		int r1,c1,r2,c2;
		scanf("%d%d%d%d",&r1,&c1,&r2,&c2);
		r1--;c1--;r2--;c2--;
		int ans=abs(r2-r1)+abs(c1-c2);
		for(int j=1;j<=k;j++)//通过k颜色将两点相连
			ans=min(ans,dis[j][r1][c1]+dis[j][r2][c2]+1);
		printf("%d\n",ans);
	}
    return 0;
}