牛客周赛 Round 1

A

游游画U

题意：

帮游游画一个大小为n的U。

示例：

题解：

因为n<=50,所以直接模拟即可,时间复杂度为O()。

对于前3*n行，从1到n列和3*n列到4*n列为‘*’，其余都为‘.’。

对于后n行，定义k=1，从1+k到n+k列和3*n-k列到4*n-k列为‘*’，其余都为‘.’。从3*n+1行到4*n行每行k+1。 

 代码：

#include<iostream>
using namespace std;
char s[500][500];
int n;
int main()
{
	int k = 1;
	cin >> n;
	for (int i = 1; i <= 4 * n; i++)
		for (int j = 1; j <= 4 * n; j++)
			s[i][j] = '.';
	for (int i = 1; i <= 3 * n; i++)
		for (int j = 1; j <= 4 * n; j++)
		{
			if (j <= n || j > 3 * n)
				s[i][j] = '*';
		}
	for (int i = 3 * n + 1; i <= 4 * n; i++, k++)
		for (int j = 1; j <= 4 * n; j++)
		{
			if ((j > k) && (j <= n + k) || (j > 3 * n - k) && (j <= 4 * n - k))
				s[i][j] = '*';
		}
	for (int i = 1; i <= 4 * n; i++)
	{
		for (int j = 1; j <= 4 * n; j++)
			cout << s[i][j];
		cout << endl;
	}
	return 0;
}

B

游游的数组染色

题意：
游游拿到了一个数组，其中一些数被染成红色，一些数被染成蓝色。
游游想知道，取两个不同颜色的数，且它们的数值相等，有多少种不同的取法？

题解：

因为n<=200000，直接模拟时间复杂度为O()，所以不可行。

可以用数组记录对应颜色中每个数字出现的频率，再将两种颜色中相同数字的频率相乘，来求解。时间复杂度约为O(n)。

代码：

#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
const int N = 2 * 1e5 + 10;
int n;
map<int, pair<int, int>>s;
long long int a[N];
string c;
int main()
{
	long long int sum = 0;
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	cin >> c;
	for (int i = 0; i < n; i++)
	{
		if (c[i] == 'R')
			s[a[i]].first++;
		else
			s[a[i]].second++;
	}
	for (auto& v : s)
	{
		sum += s[v.first].first * s[v.first].second;
	}
	cout << sum;
	return 0;
}