[CCF] 201803-2 碰撞的小球

题面

思路：小球的相对顺序不能变化，碰撞可看作穿了过去

我的代码

#include <bits/stdc++.h>
using namespace std;
struct Node{
	int id,pos;
	Node(int _id=0,int _pos=0):id(_id),pos(_pos){}
	bool operator < (const Node& rhs)const{
		return pos < rhs.pos;
	}	
}node[1000];
int n,L,t;
int cal(int x)
{
	int y = x+t;
	if(y<=L) return y;
	y -= L;
	int d = y/L, r = y%L;
	if(d&1) return r;
	else return L-r;
}
int a[1000];
int main()
{
	cin>>n>>L>>t;
	int pos;
	for(int i=0;i<n;++i)
		scanf("%d",&pos),node[i] = Node(i,pos);
	sort(node,node+n);
	map<int,int> mp;
	for(int i=0;i<n;++i)
		mp[node[i].id] = i;
	for(int i=0;i<n;++i)
		a[i] = cal(node[i].pos);
	sort(a,a+n);
	for(int i=0;i<n;++i)
		printf("%d%c",a[mp[i]],i==n-1?'\n':' ');
	
	return 0;
}