牛客练习赛97

比赛链接

牛客练习赛97

C.哦~唔西迪西小姐~

题目描述

唔西迪西地在正处在一个冰火迷宫中，迷宫由 $n$ 个格子组成，每个格子要么是冰之格，要么是火之格，唔西迪西网开始可以选择从迷宫中任意一个扞始走，走到第 $i$ 个立置时会得到值为 $a_{i}$ 的积分。（注意：唔西迪西也可以选择一个格子都不走)
如果唔西迪西当前在㲻之格，那么她可以选择一个编号大于当前格子的冰之格，跳到氻里。如果唔西迪西当前在火之格，那么她可以选择一个编号大于当前格子的火之格，跳到氻哩。如果唔西迪西目前没有格子可以走，那么结束。同时，即使存在可以跳过去的格子，唔西迪西也可以选择在任意时刻结束。
$\mathrm{~ 唔西迪西想最大化她的得分，于是她学厺了一个超能力，她能在比赛开始的时候改变最 ~}$ 个格子从冰之格变成火之格或从火之格变成冰之格，改变第 $i$ 个格子的状态会让唔西迪西的得分減少 $p_{i}$ 。 (唔西迪西改变格子的状态后才开始挑选起点开始行动，也就是说，得分分成两部分，一部分是改变格子状态的得分，部分是走格子的得分)
你能告诉唔西迪西她最多得几分吗

输入描述:

第一行两个正整数 $n, m\left(1 \leq n, m \leq 10^{5}\right)$ 。
第二行 $n$ 个整数表示 $a_{i}$ 。
第三行 $n$ 个整数表示 $p_{i}\left(-10^{5} \leq a_{i}, p_{i} \leq 10^{5}\right)$ 。
第四行 $n$ 个整数 $b_{i}$ 为 0 或 1,0 表示这个格子是冰之格, 1 表示这个格子是火之格。

输出描述:

一个数表示答案。

示例1

输入

输出

示例2

输入

10 3
80 86 57 69 59 52 94 74 9 63
8 32 77 64 53 49 22 68 27 63
1 0 0 0 1 0 0 1 1 1

输出

解题思路

贪心

考虑冰之格的情况，火之格同理，不妨考虑不能改变格子状态的情况，即每个冰之格正数之和即为答案，当有状态改变的情况时，考虑对答案的影响，当改变冰之格时，如果 $a[i]>0$ ，对答案的影响为 $-p[i]-a[i]$ ，否则影响为 $-p[i]$ ，当改变火之格时，如果 $a[i]>0$ 时，对答案的影响为 $a[i]-p[i]$ ，否则影响为 $-p[i]$

时间复杂度： $O(nlogn)$

代码

// Problem: 哦~唔西迪西小姐~
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/11187/C
// Memory Limit: 524288 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

// %%%Skyqwq
#include <bits/stdc++.h>
 
// #define int long long
#define help {cin.tie(NULL); cout.tie(NULL);}
#define pb push_back
#define fi first
#define se second
#define mkp make_pair
using namespace std;
 
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
 
template <typename T> bool chkMax(T &x, T y) { return (y > x) ? x = y, 1 : 0; }
template <typename T> bool chkMin(T &x, T y) { return (y < x) ? x = y, 1 : 0; }
 
template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

const int N=1e5+5;
int n,m,a[N],p[N];
vector<PII> v1,v2;
vector<int> v;
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)cin>>p[i];
    for(int i=1;i<=n;i++)
    {
    	int x;
    	cin>>x;
    	if(x==0)v1.pb({a[i],i});
    	else
    		v2.pb({a[i],i});
    }
    LL res1=0;
    for(int i=0;i<v1.size();i++)res1+=max(0,v1[i].fi);
    for(int i=0;i<v1.size();i++)
    	if(v1[i].fi<=0)v.pb(-p[v1[i].se]);
    	else
    		v.pb(-p[v1[i].se]-v1[i].fi);
    for(int i=0;i<v2.size();i++)
    	if(v2[i].fi>0)v.pb(v2[i].fi-p[v2[i].se]);
    	else
    		v.pb(-p[v2[i].se]);
    	// v.pb(max(0,);
    sort(v.begin(),v.end());
    for(int i=v.size()-1;i>=0&&v.size()-1-i+1<=m;i--)res1+=max(0,v[i]);
    
    v.clear();
    LL res2=0;
    for(int i=0;i<v2.size();i++)res2+=max(0,v2[i].fi);
    for(int i=0;i<v2.size();i++)
    	if(v2[i].fi<=0)v.pb(-p[v2[i].se]);
    	else
    		v.pb(-p[v2[i].se]-v2[i].fi);
    for(int i=0;i<v1.size();i++)
    	if(v1[i].fi>0)v.pb(v1[i].fi-p[v1[i].se]);
    	else
    		v.pb(-p[v1[i].se]);
    sort(v.begin(),v.end());
    for(int i=v.size()-1;i>=0&&v.size()-1-i+1<=m;i--)res2+=max(0,v[i]);
    cout<<max(res1,res2);
	
    return 0;
}

月之暗面

题目描述

给出一棵 $n$ 个点的树，有 $x$ 种普通颜色， $y$ 种特殊颜色

现在要给树上的每个节点染色，普通颜色染色没有限制，但两个相邻的节点不能染相同颜色的特殊颜色

求染色方案数，答案对 $998244353$ 取模。

输入描述:

第一行三个整数 $n, x, y （ 1 \leq n \leq 10^{6}, 1 \leq x, y \leq 10^{9} ）$ , 分别表示树的节点数, 普通颜色的种数, 特殊颜色的种数
接下来 $n-1$ 行描述这棵树, 每行两个整数 $u, v(1 \leq u, v, \leq n)$ 表示从 $u$ 到 $v$ 有一条树边保证输入的树合法。

输出描述:

一行一个整数，表示答案。

示例1

输入

2 1 2
1 2

输出

示例2

输入

输出

示例3

输入

输出

188688550

解题思路

树形dp

状态表示：
- $f[u][0]$ 表示 $u$ 染色为普通颜色时其及其子树的方案数
- $f[u][1]$ 表示 $u$ 染色为特殊颜色时其及其子树的方案数
状态计算：
- $f[u][0]=\prod (f[v][0]+f[v][1])$ ，其中 $v$ 为 $u$ 的儿子节点
- $f[u][1]=\prod (f[v][0]+f[v][1]-f[v][1]/y)$

注意：不能出现相邻的特殊颜色，所以计算某个根节点特殊颜色的方案数时需要减去儿子特殊颜色等于父节点的情况

时间复杂度： $O(n)$

代码

// Problem: 月之暗面
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/11187/D
// Memory Limit: 524288 MB
// Time Limit: 4000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

// %%%Skyqwq
#include <bits/stdc++.h>
 
#define int long long
#define help {cin.tie(NULL); cout.tie(NULL);}
#define pb push_back
#define fi first
#define se second
#define mkp make_pair
using namespace std;
 
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
 
template <typename T> bool chkMax(T &x, T y) { return (y > x) ? x = y, 1 : 0; }
template <typename T> bool chkMin(T &x, T y) { return (y < x) ? x = y, 1 : 0; }
 
template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

const int N=1e6+5,mod=998244353;
int f[N][2],n,x,y;
vector<int> adj[N];
bool vis[N];
int ksm(int a,int b,int p)
{
	int res=1%p;
	for(;b;b>>=1)
	{
		if(b&1)res=res*a%p;
		a=a*a%p;
	}
	return res;
}
PII dfs(int u)
{
	int res1=x,res2=y;
	for(int v:adj[u])
	{
		auto t=dfs(v);
		int t1=t.fi,t2=t.se;
		res1=res1*(t1+t2)%mod;
		res2=(res2*(t1+t2-t2*ksm(y,mod-2,mod)%mod)%mod+mod)%mod;
	}
	return {res1,res2};
}
signed main()
{
    cin>>n>>x>>y;
    for(int i=1;i<n;i++)
    {
    	int u,v;
    	cin>>u>>v;
    	adj[u].pb(v);
    	vis[v]=true;
    }
    int root=1;
    while(vis[root])root++;
    cout<<(dfs(root).fi+dfs(root).se)%mod;
    return 0;
}

__EOF__

本文作者：acwing_zyy
本文链接：https://www.cnblogs.com/zyyun/p/15995098.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！

posted @ 2022-03-11 19:41 zyy2001 阅读(86) 评论(0) 编辑收藏举报

刷新页面返回顶部

登录后才能查看或发表评论，立即登录或者逛逛博客园首页

相关博文：

· 牛客小白月赛49

· 牛客练习赛96

· 牛客练习赛97

· 牛客周赛 Round 30

阅读排行：
· 阿里最新开源QwQ-32B，效果媲美deepseek-r1满血版，部署成本又又又降低了！
· 单线程的Redis速度为什么快？
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决：字节Trae VS Cursor，谁才是开发者新宠？
· 展开说说关于C#中ORM框架的用法！

公告

牛客练习赛97

发表于 2022-03-11 19:41阅读：86评论：0推荐：0

牛客练习赛树形dp 贪心

关注

跳至底部

别人躺着我站着，别人走着我跑着。

比赛链接

C.哦~唔西迪西小姐~

题目描述

输入描述:

输出描述:

示例1

输入

输出

示例2

输入

输出

解题思路

代码

月之暗面

题目描述

输入描述:

输出描述:

示例1

输入

输出

示例2

输入

输出

示例3

输入

输出

解题思路

代码

公告

acwing_zyy

牛客练习赛97

搜索

常用链接

最新随笔

随笔档案