题意:T组测试数据,然后给你n,给你P序列让你求W序列。P[i]的意义是的在第i个右括号前面有多少个左括号,w[i]的意义是,在第i个右括号和与他匹配的左括号中有多少个括号,包括自己。
思路:先通过P序列重建括号序列。规律:第i+p[i]位置一定是右括号,剩下的都是左括号。总长度为2*n。
然后与第i个右括匹配的左括号一定是在他左边第一个没跟其他右括号匹配过的右括号,开个标记数组去维护即可,过程中查一下有多少个左括号就有多少个括号。
//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
const int MAXN = 105;
int n, p[MAXN], w[MAXN];
char s[MAXN];
bool vis[MAXN];
int main()
{
int T; scanf("%d", &T);
while (T--)
{
memset(s, '#', sizeof(s));
memset(vis, 0, sizeof(vis));
memset(w, 0, sizeof(w));
scanf("%d", &n);
for (int i = 0; i < n; i++)
{
scanf("%d", &p[i]);
s[i + p[i]] = ')';
}
for (int i = 0; i < n*2; i++)
if (s[i] == '#') s[i] = '(';
//s[n*2] = '\0'; cout << s << endl;
int id = 0;
for (int i = 0; i < n*2; i++)
{
if (s[i] == '(') continue;
int cnt = 0;
for (int j = i - 1; j >= 0; j--)
{
if (s[j] == '(')
{
cnt++;
if (!vis[j])
{
vis[j] = true;
w[id++] = cnt;
break;
}
}
}
}
for (int i = 0; i < n; i++)
{
printf("%d", w[i]);
if (i != n-1) printf(" ");
}
printf("\n");
}
return 0;
}
/*
20
6
4 5 6 6 6 6
9
4 6 6 6 6 8 9 9 9
*/