Submission #202165

#	Time	Username	Problem	Language	Result	Execution time	Memory
202165		Segtree	Bulldozer (JOI17_bulldozer)	C++14	20 / 100	8 ms	1008 KiB

bulldozer

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> P;
typedef complex<long double> C;
#define chmax(a,b) a=max(a,b)
#define chmin(a,b) a=min(a,b)
#define all(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<n;i++)
#define mod 1000000007
#define mad(a,b) a=(a+b)%mod
#define PI acosl(-1)
#define N 110
ll n,x[N],y[N],w[N];

ll dat[N];
void upd(ll i,ll x){
    dat[i]=x;
}
ll rsm(){
    ll mi=dat[0],res=0;
    rep(i,n){
	chmin(mi,dat[i+1]);
	chmax(res,dat[i+1]-mi);
    }
    return res;
}

ll pl[N];
ll swp(int a,int b){//cout<<"swp"<<w[a]<<" "<<w[b]<<endl;
    if(pl[a]>pl[b])swap(a,b);
    assert(pl[a]+1==pl[b]);
    upd(pl[a]+1,dat[pl[a]]+w[b]);
    swap(pl[a],pl[b]);
    return rsm();
}

struct qry{
    ld arg;
    ll a,b;
    bool operator<(const qry&key)const{
	return this->arg<key.arg;
    }
};

int main(){
    cin>>n;
    rep(i,n){
	cin>>x[i]>>y[i]>>w[i];
    }
    rep(i,n)rep(j,n-1){
	bool sw=0;
	if(y[j]!=y[j+1])sw=y[j]>y[j+1];
	else if(x[j]!=x[j+1])sw=x[j]<x[j+1];
	if(sw){
	    swap(x[j],x[j+1]);
	    swap(y[j],y[j+1]);
	    swap(w[j],w[j+1]);
	}
    }
    ll rui=0;
    upd(0,0);
    rep(i,n){
	rui+=w[i];
	upd(i+1,rui);
	pl[i]=i;
    }
    
    vector<qry> Q;
    rep(i,n)rep(j,i){
	ld vl=atan2(y[i]-y[j],x[i]-x[j]);
	ld vr=atan2(y[j]-y[i],x[j]-x[i]);
	//cout<<"atan"<<w[i]<<" "<<w[j]<<" "<<vl<<" "<<vr<<endl;
	Q.push_back((struct qry){max(vl,vr),i,j});
    }
    sort(all(Q));
    ll ans=rsm();
    rep(i,Q.size()){
	chmax(ans,swp(Q[i].a,Q[i].b));
    }
    cout<<ans<<endl;
}

Compilation message (stderr)

bulldozer.cpp: In function 'int main()':
bulldozer.cpp:10:31: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,n) for(int i=0;i<n;i++)
bulldozer.cpp:79:9:
     rep(i,Q.size()){
         ~~~~~~~~~~             
bulldozer.cpp:79:5: note: in expansion of macro 'rep'
     rep(i,Q.size()){
     ^~~

• Subtask 1: 0 / 5
• Subtask 2: 20 / 20
• Subtask 3: 0 / 35
• Subtask 4: 0 / 20
• Subtask 5: 0 / 20