CodeForces 253E Printer

题目链接

题意：有N份文件需要打印，每行输入该文件传入时间第ti秒，需要打印的时间si秒，和打印优先级pi。每一秒打印机都打印优先级最高的文件，传入的N份文件中有一份优先级为-1代表未知。已知那份优先级未知的文件打印结束的时间为T，求优先级未知的文件的优先级可以为多少，按顺序输出每份文件打印结束的时间。

题解：二分文件的优先级，判断是否成立，最后按输入顺序输出。

#include<stdio.h>
#include<queue>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<assert.h>
#include<stdlib.h>
#include<time.h>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<fstream>
#define pi acos(-1.0)
#define INF 1000000000
#define debug printf("---------------\n");
using namespace std;

struct Work
{
int st;
int v;
int yu;
int num;
friend bool operator < (Work a,Work b)
{
return a.yu<b.yu;
}
} w[50005];

bool cmp1(Work a,Work b)
{
if(a.st==b.st)
{
return a.yu>b.yu;
}
return a.st<b.st;
}
int n;
priority_queue<Work>q;
int check(int p,long long k)
{
//printf("p=%d k=%I64d\n",p,k);
Work tt;
while(!q.empty())
{
q.pop();
}
long long t=0;
for(int i=0; i<n; i++)
{
while(!q.empty()&&q.top().v<=w[i].st-t)
{
t+=q.top().v;
if(q.top().yu==p)
{
if(t==k)
{
return 0;
}
else if(t<k)
{
return -1;
}
else
{
//debug
return 1;
}
}
q.pop();
}
if(q.empty())
{
tt=w[i];
if(tt.yu==-1)
{
tt.yu=p;
}
q.push(tt);
t=tt.st;
}
else if(w[i].st-t<q.top().v)
{
tt=q.top();
q.pop();
tt.v-=w[i].st-t;
q.push(tt);
tt=w[i];
if(tt.yu==-1)
{
tt.yu=p;
}
q.push(tt);
t=tt.st;
}
}
//printf("size=%d\n",q.size());
while(!q.empty())
{
//printf("t=%d\n",t);
t+=q.top().v;
//printf("st=%d v=%d p=%d num=%d\n",q.top().st,q.top().v,q.top().yu,q.top().num);
if(q.top().yu==p)
{
if(t==k)
{
return 0;
}
else if(t<k)
{

return -1;
}
else
{
//printf("t=%d\n",t);
//debug
return 1;
}
}
q.pop();
}
}
long long s[50005];
void print(int p)
{
Work tt;
while(!q.empty())
{
q.pop();
}
long long t=0;
for(int i=0; i<n; i++)
{
while(!q.empty()&&q.top().v<=w[i].st-t)
{
t+=q.top().v;
s[q.top().num]=t;
q.pop();
}
if(q.empty())
{
tt=w[i];
if(tt.yu==-1)
{
tt.yu=p;
}
q.push(tt);
t=tt.st;
}
else if(w[i].st-t<q.top().v)
{
tt=q.top();
q.pop();
tt.v-=w[i].st-t;
q.push(tt);
tt=w[i];
if(tt.yu==-1)
{
tt.yu=p;
}
q.push(tt);
t=tt.st;
}
}
while(!q.empty())
{
t+=q.top().v;
s[q.top().num]=t;
q.pop();
}

for(int i=0;i<n;i++)
{
if(i==n-1)
{
printf("%I64d\n",s[i]);
}
else
{
printf("%I64d ",s[i]);
}
}
}
set<int>S;
int main()
{
freopen("input.txt", "r", stdin);
freopen("output.txt", "w", stdout);
long long k;
scanf("%d",&n);
for(int i=0; i<n; i++)
{
scanf("%d%d%d",&w[i].st,&w[i].v,&w[i].yu);
if(w[i].yu!=-1)
{
S.insert(w[i].yu);
}
w[i].num=i;
}
scanf("%I64d",&k);
sort(w,w+n,cmp1);
int left=1,right=INF,mid,ans;
while(left<=right)
{
mid=(left+right)>>1;
int f=check(mid,k);
//printf("f=%d\n",f);
if(f==0)
{
ans=mid;
//printf("ans=%d\n",ans);
break;
}
else if(f==-1)
{
right=mid-1;
}
else
{
left=mid+1;
}
}
int L=ans;
while(L>=1&&S.find(L)!=S.end())
{
L--;
}
int R=ans;
while(R<=INF&&S.find(R)!=S.end())
{
R++;
}
if(L>1&&check(L,k)==0)
{
//debug
printf("%d\n",L);
print(ans);
}
else
{
//debug
printf("%d\n",R);
print(R);
}

return 0;
}