NUC_TeamTEST_B(贪心)
中间超int啊,不难的贪心。CodeForces,DIV2的A题
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | #include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int max_size = 100000+10; int main() { int n, k; char str[max_size]; long long arry[26]; while ( scanf ( "%d %d" , &n, &k) == 2) { getchar (); memset (arry, 0, sizeof (arry)); scanf ( "%s" , str); int len = strlen (str); int i; for (i = 0; i < len; ++i) { arry[str[i]- 'A' ]++; } i = 25; sort(arry, arry+26); //printf("%d\n", arry[25]); long long sum = 0; long long now = k; do { if (now > arry[i]) { sum += arry[i] * arry[i]; now -= arry[i]; //printf("%d\n", sum); i--; } else if (now <= arry[i]){ sum += now*now; ///这个比较坑,没考虑到超int break ; } } while (now != 0); cout << sum << endl; } return 0; } |
【推荐】还在用 ECharts 开发大屏?试试这款永久免费的开源 BI 工具!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步