C-岗位分配_河南萌新联赛2024第(四)场:河南理工大学 (nowcoder.com)
题解:可以用排列组合,也可以用dp,dp[i][j]表示j个人分给i组有多少种方案
状态转移方程:f[i][j] = (f[i - 1][j] + f[i][j - 1]) % mod;
代码:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int mod = 998244353;
const int N = 2010;
int f[N][N];
//j个人分到i组的方案数
signed main()
{
int n,m;
cin >> n >> m;
for(int i = 1;i <= n;i ++)
{
int x;
cin >> x;
m -= x;
}
//多余m人闲置,方案数为1
for(int i = 0;i <= m;i ++) f[0][i] = 1;
for(int i = 1;i <= n;i ++)
for(int j = 0;j <= m;j ++)
{
f[i][j] = (f[i - 1][j] + f[i][j - 1]) % mod;
}
cout<<f[n][m]<<endl;
return 0;
}
E-AND_河南萌新联赛2024第(四)场:河南理工大学 (nowcoder.com)
题解:由于题目给了两秒,所以可以跑一个1e8的欧拉筛,然后将筛出来的数放到vector当中,由于“有0则0”的规则,筛出来的数只能是2和奇数,奇数二进制最后一位必定为1,所以要使区间按位为0,左区间必定为2,一共有sum-2个答案,其他答案均为0,看代码。
代码:
#include<bits/stdc++.h>
#define inf 0x3f3f3f
#define int long long
#define PII pair<int,int>
using namespace std;
int n;
const int N=1e8+10;
int x,y;
bool st[N];
vector<int >prime;
int cal(int v){
if(v<1) return 0;
return v-1;
}
void solve(){
cin>>x>>y;
int xi=lower_bound(prime.begin(),prime.end(),x)-prime.begin();
int yi=upper_bound(prime.begin(),prime.end(),y)-prime.begin();
int ans=(x==2)?cal(yi-1)-cal(xi-1):0;
cout<<yi-xi<<" "<<ans<<endl;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
prime.push_back(2);
for(int i=3;i<N/i;i+=2){
if(st[i]) continue;
for(int j=i*i;j<N;j+=2*i) st[j]=1;
}
for(int i=3;i<N;i+=2) {
if(!st[i]) prime.push_back(i);
}
// for(int i=0;i<10;i++) cout<<prime[i]<<" ";
int t=1;
cin>>t;
while(t--){
solve();
}
}
L-抓字符_河南萌新联赛2024第(四)场:河南理工大学 (nowcoder.com)
题解:详细见代码备注
代码:
#include<bits/stdc++.h>
#define inf 0x3f3f3f
#define int long long
#define PII pair<int,int>
#define ll long long
using namespace std;
int n;
const int mod=998244353;
const int N=1e6+10;
int g[2][28];//非严格递减的情况
int f[2][28];//非严格递增的情况
int d[28];i字母全选到的概率
string s;
int h,p,q;
int sum=1;
ll qm(ll a,ll b){//快速幂
ll res=1;
while(b){
if(b&1){
res=res*a%mod;
}
a=a*a%mod;
b>>=1;
}
return res;
}
void solve(){
cin>>n>>s;
s=' '+s;
f[0][0]=1;
g[0][0]=1;
for(int i=1;i<=n;i++) {
int x;
cin>>x;
h=i&1;//滚动数组,前i个的答案只与i-1相关
p = qm(x, mod - 2);
q = (1 - p + mod) % mod;
sum = sum * q % mod;
int k = s[i] - 'a';
if (d[k] == 0) {
d[k] = q;
}
else {
d[k] = d[k] * q % mod;
}
for(int j=0;j<27;j++){
f[h][j]+=f[h^1][j]*q%mod;
f[h][j]%=mod;
g[h][j]+=g[h^1][j]*q%mod;
g[h][j]%=mod;
}
k=s[i]-'a'+1;
for(int j=0;j<=k;j++){
f[h][k]+=f[h^1][j]*p%mod;
f[h][k]%=mod;
}
for(int j=k;j<27;j++){
g[h][k]+=g[h^1][j]*p%mod;
g[h][k]%=mod;
}
g[h][k]+=g[h^1][0]*p%mod;
g[h][k]%=mod;
for(int j=0;j<27;j++){//清空上一行
f[h ^ 1][j] = 0;
g[h ^ 1][j] = 0;
}
}
ll ans=0;
for(int i=1;i<27;i++){//前n个的所有答案
ans+=f[n&1][i];
ans%=mod;
ans+=g[n&1][i];
ans%=mod;
}
for(int i=0;i<26;i++){//减去字串中全为i字符的重叠情况
if(d[i]!=0){
int qt = sum * qm(d[i], mod - 2) % mod;
int cf = (1 - d[i] + mod) % mod * qt % mod;
ans -= cf;
ans += mod;
ans %= mod;
}
}
cout<<ans<<endl;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t=1;
while(t--){
solve();
}
}
J-尖塔第四强的高手_河南萌新联赛2024第(四)场:河南理工大学 (nowcoder.com)
题解:题意就是说查找这一个集合当中的最近公共祖先,直接套板子就好。
代码:
#include<bits/stdc++.h>
#define inf 0x3f3f3f
#define int long long
#define PII pair<int,int>
#define ll long long
using namespace std;
int n,r,q;
const int N=1e6+10;
int dev[N];
int f[N][20];
int F[30];
vector<int>a[N];
void dfs(int u,int fa){
dev[u]=dev[fa]+1;
f[u][0]=fa;
for(int i=1;i<=17;i++){
f[u][i]=f[f[u][i-1]][i-1];
}
for(auto v:a[u]){
if(v!=fa) dfs(v,u);
}
}
int ans=-1;
int lca(int x,int y){
if(dev[x]<dev[y]) swap(x,y);
for(int i=17;i>=0;i--){
if(dev[f[x][i]]>=dev[y]) {
x=f[x][i];
}
}
if(x==y) return x;
for(int i=17;i>=0;i--){
if(f[x][i]!=f[y][i]){
x=f[x][i];
y=f[y][i];
}
}
return f[x][0];
}
void solve(){
F[1]=1;
F[2]=2;
for(int i=3;i<30;i++){
F[i]=F[i-1]+F[i-2];
// cout<<F[i]<<endl;
}
cin>>n>>r>>q;
for(int i=0;i<n-1;i++){
int u,v;
cin>>u>>v;
a[u].push_back(v);
a[v].push_back(u);
}
dfs(r,0);
for(int i=0;i<q;i++){
int x,k;
cin>>x>>k;
if(x+F[k]>n||k>25) {
cout<<0<<endl;
continue;
}
else {
if(x+F[k+1]>n) cout<<x+F[k]<<endl;
else {
int res=lca(x+F[k],x+F[k+1]);
k+=2;
while(x+F[k]<=n){
res=lca(res,x+F[k++]);
}
cout<<res<<endl;
}
}
}
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t=1;
while(t--){
solve();
}
}