由于m=2就变成了水题一道。。。(整天水题刷刷没救了)
可以把两个栈变成一个数组,然后就变成了每次从当前最大的移动到第二大的,统计路径上还有多少个点累加入答案,然后把最大的删除。直接排序后用树状数组维护前缀和即可。
AC代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#define ll long long
using namespace std;
int n,m,c[100005]; struct node{ int x,id; }a[100005];
void del(int x){
for (; x<=n; x+=x&-x) c[x]--;
}
int getsum(int x){
int t=0; for (; x; x-=x&-x) t+=c[x]; return t;
}
bool cmp(node u,node v){ return u.x>v.x; }
int main(){
scanf("%d%d",&n,&m); int i; a[0].id=n;
for (i=n; i; i--){ scanf("%d",&a[i].x); a[i].id=i; }
for (i=1; i<=m; i++){
scanf("%d",&a[++n].x); a[n].id=n;
}
sort(a+1,a+n+1,cmp);
for (i=1; i<=n; i++) c[i]=i&-i;
ll ans=0;
for (i=1; i<=n; i++){
if (a[i].id>a[i-1].id) ans+=getsum(a[i].id-1)-getsum(a[i-1].id);
else ans+=getsum(a[i-1].id)-getsum(a[i].id);
del(a[i].id);
}
printf("%lld\n",ans);
}
by lych
2016.3.14