Codeforces D. Task On The Board (思維 / 構造) (Round #650 Div.3)

傳送門

題意： 現有一個初始字符串a，可在其上刪除一些字符(也可以不刪)得到中間字符串t。而t含有m個字符，對應符合一個標準數組b；使得b[i]正好等於t[i]與t中其他比t[i]字典序大的字符索引j的 |i - j| 的和。先給出a字符串與b數組，需構造出中間字符串t。

思路：

• 因爲b[i] = 0的時候表示該字符是字符串中最大的字符，否則會有更大的字符對其產生影響(其值必不可能爲0)，所以我們就每次都以b[i] = 0爲特殊點開始處理。
• 首先找出足夠數目的最大字符來填充所以b[i] = 0的位置，並用vis[i] = 1來標記該位置已被填充。
• 那麼其餘b[i] ！= 0的位置就會被該字符影響，減去這個影響值即可(eg：b[i] = 0，那麼對於其他b[j] != 0，則有b[j] -= |i - j|)
• 於是就會出現新的b[i] = 0，再重複以上方法最後得到答案字符串即可。

代碼實現：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2000;

int T, m;
char a[N], t[N];
int b[N], vis[N], cnt[N];

signed main()
{
    IOS;

    cin >> T;
    while(T --){
        me(cnt); me(vis);
        cin >> (a + 1) >> m;
        for(int i = 1; i <= m; i ++)
            cin >> b[i];

        int la = strlen(a + 1);
        for(int i = 1; i <= la; i ++) //統計每個字符的數量
            cnt[a[i] - 'a'] ++;
        //因爲每次都得向下找到最大字符，所有得用bg來記錄每次尋找開始得位置
        int ok = 1, res = 0, bg = 25;
        while(ok){
            int num = 0, tmp = -1;
            //統計當前b[i] = 0的數量
            for(int i = 1; i <= m; i ++)
                if(!b[i] && !vis[i]) num ++;
            //找到最大的字符能夠填充所有b[i] = 0的位置
            for(int i = bg; ~i; i --){
                if(cnt[i] >= num){
                    tmp = i;
                    break;
                }
            }
            int tt = 0, hh = 0;//開始填充
            for(int i = 1; i <= m; i ++){
                if(!b[i] && !vis[i]){
                    vis[i] = 1;
                    //統計已構造字符串的長度
                    res ++;
                    //將找到的字符tmp複製給t[i]
                    t[i] = tmp + 'a';
                    //減去i位置對後面的影響
                    for(int j = 1; j < i; j ++)
                        if(b[j]) b[j] -= (i - j);
                    hh += i; tt ++;
                }
                //減去前面對i位置的影響
                else if(b[i]) b[i] -= (tt * i - hh);
            }
            //下一次字符的尋找從tmp - 1開始
            bg = tmp - 1;
            //m長度的字符串已構造完成
            if(res == m) break;
        }

        for(int i = 1; i <= m; i ++)
            cout << t[i];
        cout << endl;
    }

    return 0;
}