背景:
IOI
\text{IOI}
IOI的水 题。
题目传送门:
题意:
n
n
n双鞋子,现在要按照大小(
∣
a
i
∣
|a_i|
∣ai∣)配对,且要满足对于一双鞋,负数在正数的左边,求最小交换次数。
思路:
考虑配对。
两两大小一样的鞋子可以配对,找出对于每一个正数
x
x
x,向右找出离它最近的
−
x
-x
−x,将其配对。这个操作可以用
vector
\text{vector}
vector完成。
考虑贪心,每一次找出一只没有被操作(没有配对)过的鞋子,找出它所配对的鞋子,那么最优策略一定是将另一只鞋子移到它的位置,区间中没有被操作过的鞋子就为操作次数,可以用树状数组来维护。
具体来说,树状数组维护某一只鞋子是否被操作,一开始全是
1
1
1,每操作一只鞋子,将它和与它配对的鞋子
−
1
-1
−1,即可。
考虑负数要在正数的左边,因此这一个贡献我们可以在配对的时候就做好。
正确性显然。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define LL long long
#define lowbit(x) ((x)&(-(x)))
using namespace std;
vector<int> f[100010],g[100010];//f:存负的位置;g:存正的位置
int n;
LL ans=0;
int pos[200010],tr[200010];
bool bz[200010];
void add(int x,int y)
{
for(;x<=(n<<1);x+=lowbit(x))
tr[x]+=y;
}
int getsum(int x)
{
int tot=0;
for(;x;x-=lowbit(x))
tot+=tr[x];
return tot;
}
int main()
{
int x;
scanf("%d",&n);
for(int i=1;i<=(n<<1);i++)
{
scanf("%d",&x);
if(x<0) f[-x].push_back(i); else g[x].push_back(i);
}
for(int i=1;i<=n;i++)
for(int j=0;j<f[i].size();j++)
{
pos[g[i][j]]=f[i][j];
pos[f[i][j]]=g[i][j];
ans+=f[i][j]>g[i][j];
}
for(int i=1;i<=(n<<1);i++)
add(i,1);
for(int i=1;i<=(n<<1);i++)
if(!bz[i])
{
add(i,-1),add(pos[i],-1);
ans+=getsum(pos[i]);
bz[i]=true,bz[pos[i]]=true;
}
printf("%lld",ans);
}