牛客挑战赛67-A,B

 A.构造

链接：https://ac.nowcoder.com/acm/contest/51727/A
来源：牛客网
 

题目描述

给定n,a,b,请构造一个长度为2n的排列P，满足如下条件：

∀1≤i<2n,prei​!=sufi​，且Pa​=b。其中prei​代表排列的前i个数之和，sufi​则代表排列的后i个数之和。

如果存在多种答案，任意输出一种即可。可以证明，答案一定存在。

输入描述:

第一行数据组数t(1≤t≤1e4)。
对于每组数据，一行三个整数n,a,b(1≤a,b≤2n,1≤n≤2⋅1e5)。
保证所有的n之和不超过2⋅1e5。

输出描述:

对于每组数据，输出一行2n个整数，表示该排列。

示例1

输入

3
2 3 4
4 6 1
5 1 3

输出

3 1 4 2
2 3 7 5 6 1 4 8
3 1 6 10 2 4 7 8 9 5

说明

对于第一个样例
pre1=3≠2=suf1​
pre2=4≠6=suf2
pre3=8≠7=suf3
故满足条件

该题构造题，题目要求排列的前i个数的和，和后i个数的和不相等，很明显只要让最小的n个数在其中一半，最大的n个数在其中另一半,就可以保证前i个数的和和后i个数的和不相等了，同时题目又要求第a个数等于b，所以构造时，要考虑4种可能：

1. a小于等于n且b小于等于n，这时候得让小的n个数放在前一半，大的后一半;
2. a小于等于n且b大于n,  这时候让大的n个数放在前一半，小的后一半;
3. a大于n且b大于n，这时候让小的前一半，大的后一半;
4. a大于n且b小于等于n，这时候让大的前一半，小的后一半. 

代码如下：

#include<iostream>
using namespace std;
const int N = 4e5+5;
int nums[N] {0};
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int t;
    cin >> t;
    while(t --)
    {
        int n, a, b;
        
        cin >> n >> a >> b;
        nums[a] = b;
        if(a<=n && b<=n)
        {
            for(int i=1, j = 1; i<=2*n; i++)
            {
                if(i==a) continue;
                if(j==b) j ++;
                nums[i] = j ++;
            }
        }
        else if(a<=n && b>n)
        {
            for(int i=1, j=2*n; i<=2*n; i ++)
            {
                if(i==a) continue;
                if(j==b) j --;
                nums[i] = j --;
            }
        }
        else if(a>n && b>n)
        {
            for(int i=1, j=1; i<=2*n; i ++)
            {
                if(i==a) continue;
                if(j==b) j ++;
                nums[i] = j ++;
            }
        }
        else if(a>n && b<=n)
        {
            for(int i=1, j=2*n; i<=2*n; i ++)
            {
                if(i==a) continue;
                if(j==b) j --;
                nums[i] = j --;
            }
        }
        for(int i=1; i<=2*n; i ++) cout << nums[i] << " ";
        cout << endl;
    }
    return 0;
}

B.数据结构

链接：https://ac.nowcoder.com/acm/contest/51727/B
来源：牛客网
 

题目描述

你有一个长度为n的字符串，其中仅含'0','1','2'三个字符。

你希望知道，这个字符串有多少个子串，满足该子串的'0','1','2'个数相等？

输入描述:

第一行一个正整数，代表数据组数t(1≤t≤1e4)。
对于每组数据，第一行一个正整数n(1≤n≤3⋅1e5)，接下来一行一个长度为n的仅含'0','1','2'的字符串。
保证所有的n之和不超过3⋅1e5。

输出描述:

对于每组数据，一行一个正整数表示答案。

示例1

输入

4
3
012
6
001122
18
102100120120120012
18
012012012012012012

输出

1
1
25
51

说明

对于前两个样例，仅有串本身满足条件。
对于第三个样例，举一个满足条件的子串：从第3个字符到第8个字符的子串210012满足条件。

 这题看大佬的题解写出来的，不得不说好妙的解法啊。 

该题用哈希和前缀和的思想实现，先开一个pair类型变量，当s[i]等于'1'时，可以等效成pair.first ++, s[i]等于'2'时，可以等效成pair.second ++, s[i]等于'0'时，可以等效成 pair.first --, pair.second --。

如果当遍历到i时，更新后pair的值等于i之前的点j的pair的值的话，那么这表示[j+1, i]这个区间的pair的变化为0，该区间内'0','1','2'个数是相等的。 所以每次遍历一个点后，查找之前是否存在过当前的pair的值，存在过几次，相当于有几个区间'0','1','2'个数是相等的，而查找的方法用哈希表实现即可。 

代码如下：

#include<iostream>
#include<string>
#include<map>
using namespace std;
int main()
{
	int t;
	cin >> t;
	while(t --)
	{
		int n;
        long long res = 0;
		string s;
		cin >> n >> s;
		map<pair<int, int>, int> mm; //用来记录pair 
		pair<int, int> sum;
		sum.first = sum.second = 0; //最初时'0''1''2'都为0，数量相等
		mm[sum] = 1;
		for(int i=0; i<s.size(); i++)
		{
			if(s[i]=='0') sum.first --, sum.second --;
			else if(s[i]=='1') sum.first ++;
			else sum.second ++;
			res += mm[sum]; //加上前面出现过的相同的pair的数目 
			mm[sum] ++;	//当前pair的大小的数量加一 
		}
		cout << res << endl;
	}
	return 0;
}