JZOJ.5329【NOIP2017模拟8.22】时间机器

Description

 

Input

Output

 

Sample Input

3
2 2
1 4 2
3 5 1
1 4 2
2 5 1
3 2
1 3 1
2 4 1
3 5 1
1 3 2
2 5 1
2 2
1 2 2
1 2 1
1 2 1
1 2 2
 

Sample Output

Yes
No
Yes

 

Data Constraint

 

Hint

很明显这是要让我们匹配，很容易想到可以二分图匹配，但是又有数量，于是我们可以网络流，但是n还是巨大，使得我们不得不想想其他办法。

我们的期望搭配就是电阻的允许电压范围跟节点的电压范围接近，于是我们可以贪心。

我们可以用左端点排序，然后对于每个节点找到第一个右端点大于它的，左端点小于等于它的即可。

但是复杂度仍为O(nm),我们可以考虑用map或set维护符合左端点条件的右端点的电阻。复杂度O((n + m) log m + n log n)。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <set>
#include <algorithm>
#define N 100005
using namespace std;
struct data{
    int l,r,num,v;
    bool operator < (const data &a) const{
        return (a.r>r);
      }
}zu[N];
multiset<data> qwq;
multiset<data>::iterator qaq;
int n,m,t,ans;
bool power;
int read(){
    int x=0,w=1;
    char c=0;
    for (c=getchar();c<'0'||c>'9';c=getchar()) {if (c=='-') w=-1;}
    for (;c>='0'&&c<='9';c=getchar()) x=(x<<3)+(x<<1)+c-'0';
    return x*w;
}
bool comp(const struct data a,const struct data b){
  return ((a.l<b.l)||(a.l==b.l)&&(a.v>b.v));
}
int main(){
    freopen("machine.in","r",stdin);
    freopen("machine.out","w",stdout);
    for (t=read();t;t--){
      qwq.clear();
      power=true;
      ans=0;
      n=read();
      m=read();
      for(int i=1;i<=n;++i){
        zu[i].l=read();
        zu[i].r=read();
        zu[i].num=read();
        zu[i].v=-1;
      }
      for(int i=n+1;i<=n+m;++i){
        zu[i].l=read();
        zu[i].r=read();
        zu[i].num=read();
        zu[i].v=1;
      }
      sort(zu+1,zu+1+n+m,comp);
      for (int i=1;i<=n+m;i++){
        if (zu[i].v==1) 
         qwq.insert(zu[i]);
        else while (zu[i].num){
          qaq=qwq.lower_bound(zu[i]);
          if (qaq==qwq.end()){
            power=false;
            puts("No");
            break;
          }
          data tmp=*qaq;
          qwq.erase(qaq);
          if (zu[i].num<tmp.num) {tmp.num-=zu[i].num,qwq.insert(tmp),zu[i].num=0;}
          else {zu[i].num-=tmp.num;}
        }
        if (power==false) break;
      }
      if (power) puts("Yes");
    }
}

这种纯考STL的题.......还出了两题

第一次打set，第一次打迭代器，第一次打重载运算符啊啊QAQ