【ACM演算法競賽日常訓練】DAY5題解與分析【儲物點的距離】【糖糖別胡說,我真的不是簽到題目】| 字首和 | 思維

Eriktse0發表於2023-03-28

DAY5共2題:

  • 儲物點的距離(字首和)

  • 糖糖別胡說,我真的不是簽到題目(multiset,思維)

? 作者:Eriktse
? 簡介:19歲,211計算機在讀,現役ACM銀牌選手?力爭以通俗易懂的方式講解演演算法!❤️歡迎關注我,一起交流C++/Python演演算法。(優質好文持續更新中……)?
? 原文連結(閱讀原文獲得更好閱讀體驗):https://www.eriktse.com/algorithm/1096.html

儲物點的距離

題目連結:https://ac.nowcoder.com/acm/problem/14683

預處理出各點搬運到點1和點n的代價字首和,以及區間重量和。

假如我們要將區間[5, 7]的物品全部搬運到3,代價應該是區間[5, 7]的物品全部搬運到的1,然後減去多搬的代價:[5, 7]的重量和 * dist(1, 3)。

上面是x < l的情況,x > r的情況類似。

l <= x <= r只需將區間[l, r]分為兩部分求和即可。

記得取模,距離要取模,字首和也要取模。

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int maxn = 2e5 + 9, p = 1e9 + 7;
//sum_l[i]表示區間[1, i]的物品都運到點1的代價之和
//prefix_a[i]表示區間[1, i]的物品重量之和
//pos[i]是第i個點的位置,透過a[]作字首和得到
int a[maxn], pos[maxn], sum_l[maxn], sum_r[maxn], prefix_a[maxn];
int n, m;

//取模函式
int mo(int x){return (x % p + p) % p;}

int f(int x, int l, int r)
{
    if(l > r)return 0;
    int res = 0;
    if(x <= l)
    {
        res = mo(sum_l[r] - sum_l[l - 1]);
        res = mo(res - mo(pos[x] - pos[1]) * mo(prefix_a[r] - prefix_a[l - 1]) % p);
    }
    else if(x >= r)
    {
        res = mo(sum_r[r] - sum_r[l - 1]);
        res = mo(res - mo(pos[n] - pos[x]) * mo(prefix_a[r] - prefix_a[l - 1]) % p);
    }
    return res;
}

signed main()
{
    scanf("%lld %lld",&n, &m);
    pos[1] = 1;
    for(int i = 2, d;i <= n; ++ i)scanf("%lld", &d), pos[i] = pos[i - 1] + d;
    for(int i = 1;i <= n; ++ i)scanf("%lld", a + i);
    
    for(int i = 1;i <= n; ++ i)
    {
        sum_l[i] = mo(sum_l[i - 1] + a[i] * mo(pos[i] - pos[1]) % p);
        sum_r[i] = mo(sum_r[i - 1] + a[i] * mo(pos[n] - pos[i]) % p);
    }

    
    for(int i = 1;i <= n; ++ i)prefix_a[i] = mo(prefix_a[i - 1] + a[i]);
    
    while(m --)
    {
        int x, l, r;scanf("%lld %lld %lld", &x, &l, &r);
        int ans = 0;
        
        if(l <= x and x <= r)ans = mo(f(x, l, x - 1) + f(x, x + 1, r));
        else ans = f(x, l, r);
        
        printf("%lld\n", ans);
    }
    
    return 0;
}

糖糖別胡說,我真的不是簽到題目

題目連結:https://ac.nowcoder.com/acm/problem/14583

本題有兩種解法,第一種容易理解,第二種效率更優。

第一種解法:正向,multiset。

發功次數可以用一個桶來記錄,讓[1, i]的所有點的屬性值都+1,相當於把後面的都-1,用一個fix表示偏移量。

建立兩個multiset表示兩組中的糖糖,好處是可以快速找出能力值最小的,從而去除掉。

掃一遍,把第i只糖糖加入到屬於它的集合中,然後將另外一個集合中已有的能力值小的糖糖進行刪除,再檢查此時是否有發功。

最後集合中留下的糖糖個數即為答案。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e6 + 9, inf = 8e18;

struct Node
{
	int a, b;
}p[maxn];

int add[maxn];

void solve()
{
	int n, m;scanf("%lld %lld",&n, &m);
    memset(add, 0, sizeof(int) * (n + 2));
	for(int i = 1;i <= n; ++ i)
		scanf("%lld %lld", &p[i].a, &p[i].b);
    
	//注意同一時間可能施法多次
	for(int i = 1, x;i <= m; ++ i)scanf("%lld", &x), add[x] ++;
    
	int fix = 0, cnt = 0;
    multiset<int> st[2];
    
	for(int i = 1;i <= n; ++ i)
    {
        int a = p[i].a, b = p[i].b - fix;
        
        st[a].insert(b);
        
        while(!st[a ^ 1].empty() and *st[a ^ 1].begin() < b)
            st[a ^ 1].erase(st[a ^ 1].begin()), cnt ++;
        
        fix += add[i];
	}
	printf("%lld\n", n - cnt);
}

signed main()
{
	int _;scanf("%lld", &_);
	while(_ --)solve();
	return 0;
}

第二種解法:反向,思維。

我們想這麼一個問題,一個糖糖x是否會被刪除取決於x出現之後,在另外一個集合中,是否出現了能力值高於x的能力值的糖糖y

那麼我們逆向遍歷,維護兩個集合的最值,當前的新加入的糖糖x的能力值如果低於另外一個集合中已經存在的(即右邊的)糖糖能力值的最大值,說明他後面會被某個糖糖y刪除掉,直接打上標記,但是我們不用真的刪除。

糖糖x還可以用於刪除左邊的另一個集合的能力值較小的糖糖。注意此時的發功應該在迴圈開始時進行修改。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e6 + 9, inf = 8e18;

struct Node
{
	int a, b;
}p[maxn];

int add[maxn];

void solve()
{
	int n, m;scanf("%lld %lld",&n, &m);
    memset(add, 0, sizeof(int) * (n + 2));
	for(int i = 1;i <= n; ++ i)
		scanf("%lld %lld", &p[i].a, &p[i].b);
    
	//注意同一時間可能施法多次
	for(int i = 1, x;i <= m; ++ i)scanf("%lld", &x), add[x] ++;
    
	int fix = 0, cnt = 0;
    int mx[2] = {-inf, -inf};
    
	for(int i = n;i >= 1; -- i)
    {
        fix += add[i];
        int a = p[i].a, b = p[i].b + fix;
        mx[a] = max(mx[a], b);
        
        if(mx[a ^ 1] > b)cnt ++;
        
	}
	printf("%lld\n", n - cnt);
}

signed main()
{
	int _;scanf("%lld", &_);
	while(_ --)solve();
	return 0;
}

? 本文由eriktse原創,創作不易,如果對您有幫助,歡迎小夥伴們點贊?、收藏⭐、留言?

相關文章