Codeforces242 D. Dispute（贪心，只增不减思维，cf2100）

最新推荐文章于 2024-07-24 15:50:22 发布

live4m

最新推荐文章于 2024-07-24 15:50:22 发布

阅读量112

点赞数 2

文章标签：算法

本文链接：https://blog.csdn.net/weixin_44178736/article/details/140221705

版权

题意：

在这里插入图片描述
数据范围：n<=1e5

解法：

设b[i]为点i目前的值, 一开始为0.

用队列存储b[i]==a[i]的点,
对于队列中的点x，按点x的按钮使得b[x]>a[x]，
由于按钮只会增加权值，因此后续x一定不会出现b[x]==a[x].

显然每个点上的按钮最多被点击一次，一定有解。

Code：

#include <bits/stdc++.h>
using namespace std;
#define X first
#define Y second
#define int long long
#define PI pair<int, int>
const int maxm=3e5+5;
// const int mod=998244353;
const int mod=1e9+7;

int n,m;
int a[maxm];
int b[maxm];
vector<int>g[maxm];
void solve() {
  cin>>n>>m; 
  for(int i=1;i<=m;i++){
    int l,r;cin>>l>>r;
    g[l].push_back(r);
    g[r].push_back(l);
  }
  for(int i=1;i<=n;i++){
    cin>>a[i];
  }
  vector<int>ans;
  queue<int>q;
  for(int i=1;i<=n;i++){
    if(a[i]==b[i]){
      q.push(i);
    }
  }
  while(q.size()){
    int x=q.front();q.pop();
    if(b[x]!=a[x]){
      continue;
    }
    ans.push_back(x);
    for(int v:g[x]){
      b[v]++;
      if(b[v]==a[v]){
        q.push(v);
      }
    }
  }
  cout<<ans.size()<<endl;
  for(auto i:ans){
    cout<<i<<' ';
  }
  cout<<endl;
}
signed main() {
// #define MULTI_CASE
  ios::sync_with_stdio(0);
  cin.tie(0);
#ifndef ONLINE_JUDGE
  freopen("../in.txt", "r", stdin);
  freopen("../out.txt", "w", stdout);
#endif
#ifdef MULTI_CASE
  int T;
  cin >> T;
  while (T--)
#endif
    solve();
  return 0;
}