https://vjudge.net/problem/UVA-1423
该题一看很容易懵逼,不知道从哪里想起,看网上思路是说前缀和,才明白。
我们定义
a
i
a_{i}
ai 为第 i 个元素,
S
i
=
∑
j
=
0
n
a
j
S_{i}=\sum_{j=0}^{n}a_{j}
Si=∑j=0naj ,并且
S
0
=
0
S_{0}=0
S0=0,我们很容易得到 S 之间的大小关系,并且有个初始值
S
0
S_{0}
S0,这题就好做了:
我们首先得到S的大小关系,然后根据大小关系给 S 赋值,最终得到 a ,但是网上大部分题解都是用拓扑排序来确定的(好像并不需要?)
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include <set>
#include <vector>
#include <queue>
#include <utility>
#include <map>
#define LL long long
using namespace std;
const int N = 1e5+5;
const int MAXN=100000+5;
const LL INF=1E17;
int n,m;
struct node{
int id,x,ans;
}s[N];
bool cmp1(node &x,node &y){
return x.x<y.x;
}
bool cmp2(node &x,node &y){
return x.id<y.id;
}
int main()
{
int t;
cin>>t;
while(t--)
{
string str;
int n;
cin>>n>>str;
for(int i=0;i<=n;i++)
s[i].id=i,s[i].x=0;
int ch=0;
for(int i=1;i<=n;i++)
for(int j=i;j<=n;j++)
{
if(str[ch]=='+') //j>i-1
s[j].x+=1000;
else if(str[ch]=='-')
s[i-1].x+=1000;
else
s[j].x+=10,s[i-1].x+=10;
ch++;
}
sort(s,s+1+n,cmp1);
int id=0;
for(int i=0;i<=n;i++)
if(s[i].id==0)
id=i;
s[id].ans=0;
for(int i=id+1;i<=n;i++)
if(s[i].x==s[i-1].x)
s[i].ans=s[i-1].ans;
else
s[i].ans=s[i-1].ans+1;
for(int i=id-1;i>=0;i--)
if(s[i].x==s[i+1].x)
s[i].ans=s[i+1].ans;
else
s[i].ans=s[i+1].ans-1;
sort(s,s+1+n,cmp2);
for(int i=1;i<=n;i++)
printf("%d%c",s[i].ans-s[i-1].ans,i==n?'\n':' ');
}
return 0;
}
/*
3
4
-+0++++--+
2
+++
5
++0+-+-+--+-+--
*/