WA x3,应该是一些小细节出了问题
主要思路是定义指针 p,代表从右往左第一个没有石头的点,然后批量模拟有石头的地方该把石头给哪里,计算移动总次数。
#include<iostream>
#include<algorithm>
using namespace std;
#define int long long
struct node{
int id,num;
}a[200005];
bool cmp(node x,node y){
return x.id<y.id;
}
signed main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=m;i++) cin>>a[i].id;
for(int i=1;i<=m;i++) cin>>a[i].num;
sort(a+1,a+m+1,cmp);
int p=n,cnt=m,ans=0;
while(cnt>=1){
if(p-a[cnt].id+1<a[cnt].num){
cout<<-1;
return 0;
}
p-=a[cnt].num;
ans+=(p+1-a[cnt].id+p+a[cnt].num-a[cnt].id)*a[cnt].num/2;
cnt--;
}
cout<<ans;
return 0;
}