以为跟蚂蚁差不多,,想了半天没结果,,翻别人的才知道直接暴力。。。。
两种写法,第二种是纯粹没事找事。。
思路都一样,每个球直接加一,碰到墙或小球转变方向,为了方便,我用 0和n+1 的小球直接代替墙
保存原来位置用了散列(直接也行,还是想复杂了= =)
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
#define maxn 1050
int n , l , t;
struct Node{
int val;
int idx;
int state;
bool operator < ( Node a )const{
return val < a.val;
};
};
Node ball[maxn];
int main()
{
cin >> n >> l >> t;
for(int i = 1 ; i <= n ; ++i){
cin >> ball[i].val;
ball[i].idx = i;
ball[i].state = 0;
}
sort( ball + 1 , ball + n + 1 );
ball[n+1].val = l;
while(t--){
for(int i = 1 ; i <= n ; ++i){
ball[i].val += !ball[i].state ? 1 : -1;
}
for(int i = 1 ; i <= n ; ++i){
if( ball[i].val == ball[i+1].val || ball[i].val == ball[i-1].val )ball[i].state = !ball[i].state;
}
}
int ans[maxn];
for(int i = 1 ; i <= n ; ++i)
ans[ball[i].idx] = ball[i].val;
for(int i = 1 ; i <= n ; ++i)
printf("%d ",ans[i]);
}
第二种
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
#define maxn 1050
int n , l , t;
struct Node{
int val;
int idx;
int state;
bool operator < ( Node a )const{
return val < a.val;
};
};
Node ball[maxn];
int main()
{
cin >> n >> l >> t;
for(int i = 1 ; i <= n ; ++i){
cin >> ball[i].val;
ball[i].idx = i;
ball[i].state = 0;
}
sort( ball + 1 , ball + n + 1 );
ball[n+1].val = l;
ball[n+1].state = -1;
while(t--){
for(int i = 1 ; i <= n ; ++i){
ball[i].val += !ball[i].state ? 1 : -1;
if( ball[i].val == (ball[i+1].val + (ball[i+1].state == -1 ? 0 : ball[i+1].state == 0 ? 1 : -1 )) || ball[i].val == ball[i-1].val )ball[i].state = !ball[i].state;
}
}
int ans[maxn];
for(int i = 1 ; i <= n ; ++i)
ans[ball[i].idx] = ball[i].val;
for(int i = 1 ; i <= n ; ++i)
printf("%d ",ans[i]);
}