思路:不知道有没有把这个题做复杂,,就是直接按序号读入,读入完毕后,将有效的(没有被撤销的)记录分别插入买和卖的容器,按价格进行排序。然后暴力枚举答案的价格,找出最大的。
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
const int N = 6005;
typedef long long ll;
ll sum1[N],sum2[N];
struct node
{
int flag;//0是买,1是卖
double val; //价格
int s; //股数
int cancal; //是否被撤销
}a[N];
string getstring(int &i,string &s)
{
if(i>=s.size()) return NULL;
string ans;
while(s[i]!=' '&&i<s.size())
ans+=s[i++];
i++;
return ans;
}
double getdouble(int &i,string &s)
{
if(i>=s.size()) return 0;
double ans=0,tem=0.1;
while(s[i]!='.'&&i<s.size())
ans=ans*10+(s[i++]-'0');
i++;
while(s[i]!=' '&&i<s.size())
ans=ans+(double)(s[i++]-'0')*tem,tem*=0.1;
i++;
return ans;
}
int getint(int &i,string &s)
{
if(i>=s.size()) return 0;
int ans=0;
while(s[i]!=' '&&i<s.size())
ans=ans*10+s[i++]-'0';
i++;
return ans;
}
bool cmp(int x,int y)
{
return a[x].val<a[y].val;
}
int main()
{
string s;
int n=1;
map<string,int> mp;
mp["buy"]=0;mp["sell"]=1;
while(getline(cin,s))
{
int i=0;
if(s[0]=='c') //标记撤销
{
string ss=getstring(i,s);a[getint(i,s)].cancal=1;n++;continue;
}
a[n]={mp[getstring(i,s)],getdouble(i,s),getint(i,s),0};
n++;
// if(n>7) break;
}
vector<int> buy,sell;
for(int i=1;i<n;i++)
{
if(a[i].cancal==0&&a[i].s)//整理有效数据
{
if(a[i].flag==0) buy.push_back(i);
else sell.push_back(i);
}
}
sort(buy.begin(),buy.end(),cmp);
sort(sell.begin(),sell.end(),cmp);
ll sum=0,now=0,ans_s=0;//sum的作用:大于等于价格的股数即总股数减去比它价格小的总股数
double ans_val=0;
for(int i=0;i<buy.size();i++)
sum+=a[buy[i]].s;
for(int i=0;i<buy.size();i++)
{
ll x=sum-now,y=0;
for(int j=0;j<sell.size()&&a[sell[j]].val<=a[buy[i]].val;j++) //这里其实还可以优化,但是数据不大,所以就直接循环了,其实可以二分
y+=a[sell[j]].s;
if(min(x,y)>=ans_s) ans_s=min(x,y),ans_val=a[buy[i]].val;
now+=a[buy[i]].s;
}
printf("%.2lf ",ans_val);
cout<<ans_s<<endl;
return 0;
}