Codeforces 707D.Persistent Bookcase(离线算法,dfs树,好题!)

传送门http://codeforces.com/problemset/problem/707/D
题目大意:
有一个书架,书架有n层(分别编号为1…n),每层能放m本书(分别编号为1…m),有q次操作(分别编号为1…q):
1 i j:在第i层第j位放一本书(如果这个位置没有书)
2 i j :在第i层第j位取走一本书(如果这个位置有书)
3 i:翻转第i行(无书变有书,有书变无书)
4 k:回退到第k个操作后的状态,若k=0则为清空
输出每次操作后书架上有几本书。
题目分析:
一开始没什么思路,以为是线段树。去百度搜了一下大神的做法,想了大半个下午终于想明白了。
用布尔型二维数组bool book[1005][1005]模拟书架,用一棵树记录每个操作之后的状态,树的节点记录下操作号(op,i,j)和此时书架上书的数目(ans)。读入数据时记录树的邻接表,其中遇到1,2,3操作依次向子节点延伸,遇到4操作在回退的操作编号上增加一个子节点,然后dfs遍历这棵树,更新每个节点的数目域。
这样说可能有点不太明白,看下面的测试用例:

19 16 14
3 10
3 8
3 3
2 5 3
1 7 15
4 0
2 11 2
1 16 3
1 16 3
3 13
1 13 3
4 9
1 5 11
3 1

首先树根的数目域为0,接下来按输入的操作顺序建树如下:

测试用例对应的树型结构

然后dfs这个树,读每个节点的操作号,如果是1,则子节点的ans可能+1也可能不变,如果是2则-1或不变,如果是3则统计第i行中有书的个数,用m减这个个数。如果是4则直接向下搜索。(例如节点9和节点12的ans域是一样的,因为12的操作为4 9)。记得向下搜索完回溯,保证搜索到第i个节点时,书架的情况正好是第i次操作后的书架。

本题似乎还可以用二维线段树解决?不过操作4怎么办?

AC代码:

#include <bits/stdc++.h>
using namespace std;
struct node {
  int op,i,j;//每次操作的操作数
  int ans;//当前节点中书的数目
};
node p[100005];
bool book[1005][1005];
vector<int> g[100005];//邻接表
int n,m,q;
void dfs(int t,int cur) {
  p[t].ans=cur;
  for(int i=0;i<g[t].size();i++) {
    int next=g[t][i];
    if(p[next].op==1) {
      if(book[p[next].i][p[next].j]) {//操作1下如果有书则书架不变
        dfs(next,cur);
      }
      else {
        book[p[next].i][p[next].j]=true;
        dfs(next,cur+1);
        book[p[next].i][p[next].j]=false;//回溯,保持书架情况与当前节点一致
      }
    }
    else if(p[next].op==2) {
      if(!book[p[next].i][p[next].j]) {
        dfs(next,cur);
      }
      else {
        book[p[next].i][p[next].j]=false;
        dfs(next,cur-1);
        book[p[next].i][p[next].j]=true;
      }
    }
    else if(p[next].op==3) {
      int change=0;//要改变的书本数目
      for(int j=1;j<=m;j++) {
        if(book[p[next].i][j]) {
          book[p[next].i][j]=false;
          change--;
        }
        else {
          book[p[next].i][j]=true;
          change++;
        }
      }
      dfs(next,cur+change);
      for(int j=1;j<=m;j++)
        book[p[next].i][j]=!book[p[next].i][j];
    }
    else {
      dfs(next,cur); //跳转不改变书本数量
    }
  }
}
int main() {
  memset(book,false,sizeof(book));
  p[0].ans=0;
  scanf("%d %d %d", &n,&m,&q);
  for(int i=1;i<=q;i++) {
    scanf("%d", &p[i].op);
    if(p[i].op==1 || p[i].op==2) {
      g[i-1].push_back(i);//向下顺序建树
      scanf("%d %d", &p[i].i,&p[i].j);
    }
    else if(p[i].op==3) {
      g[i-1].push_back(i);
      scanf("%d", &p[i].i);
    }
    else {
      scanf("%d", &p[i].i);
      g[p[i].i].push_back(i);//要退回的节点增加一个孩子
    }
  }
  dfs(0,0);//0号节点为根节点,有0本书
  for(int i=1;i<=q;i++)
    printf("%d\n", p[i].ans);
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
题目描述 给定一棵 $n$ 个节点的,每个节点都有一个权值 $w_i$。你需要删去上的一些边,使得剩下的每个连通块中,所有节点的权值之和不超过 $k$。 求最多能删去多少条边。 输入格式 第一行包含两个整数 $n,k$。 第二行包含 $n$ 个整数 $w_1,w_2,\cdots,w_n$。 接下来 $n-1$ 行,每行包含两个整数 $a,b$,表示节点 $a$ 和节点 $b$ 之间有一条边。 输出格式 输出一个整数,表示最多能删去的边数。 数据范围 $1\le n \le 10^5$ $1 \le k,w_i \le 10^9$ 输入样例1: 5 6 2 3 1 5 4 1 2 1 3 2 4 2 5 输出样例1: 2 输入样例2: 5 3 2 3 1 5 4 1 2 1 3 2 4 2 5 输出样例2: 0 算法1 (dfs) $O(n)$ 首先我们可以想到暴力的做法,即对于每个连通块,暴力枚举删去哪些边,尝试得到最多的删边数。那么如何求解一个连通块内的所有节点权值之和呢?我们可以使用 DFS 遍历,对于每个节点,求出其子内所有节点的权值之和,那么以该节点为根的子内的所有节点权值之和就是该节点自身的权值加上其所有子节点的权值之和。 对于每个连通块,我们枚举该连通块内任意两个节点 $x,y$。如果 $x$ 与 $y$ 之间的路径上的所有边都被删去了,那么 $x$ 和 $y$ 就会分别成为两个新的连通块,这两个新的连通块内所有节点的权值之和都不超过 $k$。因此我们只需要枚举所有的 $x,y$ 对,对于每个 $x,y$ 对尝试删去它们之间的路径上的所有边,看是否能够让 $x$ 和 $y$ 成为两个新的连通块,进而更新答案即可。 时间复杂度 参考文献 python3 代码 算法2 (暴力枚举) $O(n^2)$ blablabla 时间复杂度 参考文献 C++ 代码 class Solution { public: const int N = 1e5+10; int n,k; int h[N],e[N<<1],ne[N<<1],idx; int w[N]; int sum[N]; bool st[N]; int res; void add(int a,int b) { e[idx] = b,ne[idx] = h[a],h[a] = idx++; } void dfs(int u,int father) { sum[u] = w[u]; for(int i=h[u];~i;i=ne[i]) { int j = e[i]; if(j == father) continue; dfs(j,u); sum[u] += sum[j]; } } void dfs2(int u,int father) { for(int i=h[u];~i;i=ne[i]) { int j = e[i]; if(j == father) continue; dfs2(j,u); if(sum[j] <= k) res++; else if(sum[u] - sum[j] <= k) res++; } } void solve() { cin >> n >> k; memset(h,-1,sizeof(h)); for(int i=1;i<=n;i++) cin >> w[i]; for(int i=1;i<n;i++) { int a,b; cin >> a >> b; add(a,b); add(b,a); } dfs(1,-1); dfs2(1,-1); cout << res << endl; } }; int main() { Solution().solve(); return 0; }

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值